Update app.py
Browse files
app.py
CHANGED
@@ -4,12 +4,18 @@ import json
|
|
4 |
import time
|
5 |
import random
|
6 |
import logging
|
7 |
-
|
8 |
|
9 |
app = Flask(__name__)
|
10 |
|
11 |
# 配置日志
|
12 |
-
logging.basicConfig(
|
|
|
|
|
|
|
|
|
|
|
|
|
13 |
logger = logging.getLogger(__name__)
|
14 |
|
15 |
SYSTEM_ASSISTANT = """作为 Stable Diffusion Prompt 提示词专家,您将从关键词中创建提示,通常来自 Danbooru 等数据库。
|
@@ -58,12 +64,20 @@ def translate_and_enhance_prompt(prompt, auth_token):
|
|
58 |
'Content-Type': 'application/json',
|
59 |
'Authorization': auth_token
|
60 |
}
|
|
|
|
|
|
|
|
|
|
|
61 |
try:
|
62 |
response = requests.post(translate_url, headers=headers, json=translate_body, timeout=30)
|
|
|
|
|
|
|
63 |
response.raise_for_status()
|
64 |
result = response.json()
|
65 |
return result['choices'][0]['message']['content']
|
66 |
-
except RequestException as e:
|
67 |
logger.error(f"Error in translate_and_enhance_prompt: {str(e)}")
|
68 |
raise
|
69 |
|
@@ -108,12 +122,19 @@ def handle_request():
|
|
108 |
'Authorization': random_token
|
109 |
}
|
110 |
|
|
|
|
|
|
|
|
|
111 |
try:
|
112 |
response = requests.post(new_url, headers=headers, json=new_request_body, timeout=60)
|
|
|
|
|
|
|
113 |
response.raise_for_status()
|
114 |
response_body = response.json()
|
115 |
image_url = response_body['images'][0]['url']
|
116 |
-
except RequestException as e:
|
117 |
logger.error(f"Error in image generation request: {str(e)}")
|
118 |
return jsonify({"error": "Failed to generate image"}), 500
|
119 |
|
|
|
4 |
import time
|
5 |
import random
|
6 |
import logging
|
7 |
+
import sys
|
8 |
|
9 |
app = Flask(__name__)
|
10 |
|
11 |
# 配置日志
|
12 |
+
logging.basicConfig(
|
13 |
+
level=logging.DEBUG,
|
14 |
+
format='%(asctime)s [%(levelname)s] %(message)s',
|
15 |
+
handlers=[
|
16 |
+
logging.StreamHandler(sys.stdout)
|
17 |
+
]
|
18 |
+
)
|
19 |
logger = logging.getLogger(__name__)
|
20 |
|
21 |
SYSTEM_ASSISTANT = """作为 Stable Diffusion Prompt 提示词专家,您将从关键词中创建提示,通常来自 Danbooru 等数据库。
|
|
|
64 |
'Content-Type': 'application/json',
|
65 |
'Authorization': auth_token
|
66 |
}
|
67 |
+
|
68 |
+
logger.info(f"Sending request to {translate_url}")
|
69 |
+
logger.info(f"Request headers: {headers}")
|
70 |
+
logger.info(f"Request body: {json.dumps(translate_body, ensure_ascii=False)}")
|
71 |
+
|
72 |
try:
|
73 |
response = requests.post(translate_url, headers=headers, json=translate_body, timeout=30)
|
74 |
+
logger.info(f"Response status code: {response.status_code}")
|
75 |
+
logger.info(f"Response content: {response.text}")
|
76 |
+
|
77 |
response.raise_for_status()
|
78 |
result = response.json()
|
79 |
return result['choices'][0]['message']['content']
|
80 |
+
except requests.exceptions.RequestException as e:
|
81 |
logger.error(f"Error in translate_and_enhance_prompt: {str(e)}")
|
82 |
raise
|
83 |
|
|
|
122 |
'Authorization': random_token
|
123 |
}
|
124 |
|
125 |
+
logger.info(f"Sending request to {new_url}")
|
126 |
+
logger.info(f"Request headers: {headers}")
|
127 |
+
logger.info(f"Request body: {json.dumps(new_request_body, ensure_ascii=False)}")
|
128 |
+
|
129 |
try:
|
130 |
response = requests.post(new_url, headers=headers, json=new_request_body, timeout=60)
|
131 |
+
logger.info(f"Response status code: {response.status_code}")
|
132 |
+
logger.info(f"Response content: {response.text}")
|
133 |
+
|
134 |
response.raise_for_status()
|
135 |
response_body = response.json()
|
136 |
image_url = response_body['images'][0]['url']
|
137 |
+
except requests.exceptions.RequestException as e:
|
138 |
logger.error(f"Error in image generation request: {str(e)}")
|
139 |
return jsonify({"error": "Failed to generate image"}), 500
|
140 |
|