Update app.py

app.py

@@ -3,9 +3,15 @@ import requests
 import json
 import time
 import random
+import logging
+from requests.exceptions import RequestException
 
 app = Flask(__name__)
 
+# 配置日志
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+logger = logging.getLogger(__name__)
+
 SYSTEM_ASSISTANT = """作为 Stable Diffusion Prompt 提示词专家，您将从关键词中创建提示，通常来自 Danbooru 等数据库。
 提示通常描述图像，使用常见词汇，按重要性排列，并用逗号分隔。避免使用"-"或"."，但可以接受空格和自然语言。避免词汇重复。
 
@@ -52,11 +58,14 @@ def translate_and_enhance_prompt(prompt, auth_token):
         'Content-Type': 'application/json',
         'Authorization': auth_token
     }
-    response = requests.post(translate_url, headers=headers, json=translate_body)
-    if response.status_code != 200:
-        raise Exception(f'Translation error: {response.status_code}')
-    result = response.json()
-    return result['choices'][0]['message']['content']
+    try:
+        response = requests.post(translate_url, headers=headers, json=translate_body, timeout=30)
+        response.raise_for_status()
+        result = response.json()
+        return result['choices'][0]['message']['content']
+    except RequestException as e:
+        logger.error(f"Error in translate_and_enhance_prompt: {str(e)}")
+        raise
 
 @app.route('/')
 def index():
@@ -74,13 +83,15 @@ def handle_request():
         
         prompt = messages[-1]['content']
         
-        # 获取随机token
         random_token = get_random_token(request.headers.get('Authorization'))
         if not random_token:
             return jsonify({"error": "Unauthorized: Invalid or missing Authorization header"}), 401
         
-        # 翻译并增强prompt
-        enhanced_prompt = translate_and_enhance_prompt(prompt, random_token)
+        try:
+            enhanced_prompt = translate_and_enhance_prompt(prompt, random_token)
+        except Exception as e:
+            logger.error(f"Error in translate_and_enhance_prompt: {str(e)}")
+            return jsonify({"error": "Failed to enhance prompt"}), 500
         
         new_url = f'https://api.siliconflow.cn/v1/{model}/text-to-image'
         new_request_body = {
@@ -97,9 +108,15 @@ def handle_request():
             'Authorization': random_token
         }
         
-        response = requests.post(new_url, headers=headers, json=new_request_body)
-        response_body = response.json()
-        image_url = response_body['images'][0]['url']
+        try:
+            response = requests.post(new_url, headers=headers, json=new_request_body, timeout=60)
+            response.raise_for_status()
+            response_body = response.json()
+            image_url = response_body['images'][0]['url']
+        except RequestException as e:
+            logger.error(f"Error in image generation request: {str(e)}")
+            return jsonify({"error": "Failed to generate image"}), 500
+        
         unique_id = int(time.time() * 1000)
         current_timestamp = unique_id // 1000
         
@@ -147,6 +164,7 @@ def handle_request():
             data_string = json.dumps(response_payload)
             return Response(f"{data_string}\n\n", content_type='text/event-stream')
     except Exception as e:
+        logger.error(f"Unexpected error in handle_request: {str(e)}")
         return jsonify({"error": f"Internal Server Error: {str(e)}"}), 500
 
 if __name__ == '__main__':