Update app.py

app.py

@@ -4,12 +4,18 @@ import json
 import time
 import random
 import logging
-from requests.exceptions import RequestException
+import sys
 
 app = Flask(__name__)
 
 # 配置日志
-logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+logging.basicConfig(
+    level=logging.DEBUG,
+    format='%(asctime)s [%(levelname)s] %(message)s',
+    handlers=[
+        logging.StreamHandler(sys.stdout)
+    ]
+)
 logger = logging.getLogger(__name__)
 
 SYSTEM_ASSISTANT = """作为 Stable Diffusion Prompt 提示词专家，您将从关键词中创建提示，通常来自 Danbooru 等数据库。
@@ -58,12 +64,20 @@ def translate_and_enhance_prompt(prompt, auth_token):
         'Content-Type': 'application/json',
         'Authorization': auth_token
     }
+    
+    logger.info(f"Sending request to {translate_url}")
+    logger.info(f"Request headers: {headers}")
+    logger.info(f"Request body: {json.dumps(translate_body, ensure_ascii=False)}")
+    
     try:
         response = requests.post(translate_url, headers=headers, json=translate_body, timeout=30)
+        logger.info(f"Response status code: {response.status_code}")
+        logger.info(f"Response content: {response.text}")
+        
         response.raise_for_status()
         result = response.json()
         return result['choices'][0]['message']['content']
-    except RequestException as e:
+    except requests.exceptions.RequestException as e:
         logger.error(f"Error in translate_and_enhance_prompt: {str(e)}")
         raise
 
@@ -108,12 +122,19 @@ def handle_request():
             'Authorization': random_token
         }
         
+        logger.info(f"Sending request to {new_url}")
+        logger.info(f"Request headers: {headers}")
+        logger.info(f"Request body: {json.dumps(new_request_body, ensure_ascii=False)}")
+        
         try:
             response = requests.post(new_url, headers=headers, json=new_request_body, timeout=60)
+            logger.info(f"Response status code: {response.status_code}")
+            logger.info(f"Response content: {response.text}")
+            
             response.raise_for_status()
             response_body = response.json()
             image_url = response_body['images'][0]['url']
-        except RequestException as e:
+        except requests.exceptions.RequestException as e:
             logger.error(f"Error in image generation request: {str(e)}")
             return jsonify({"error": "Failed to generate image"}), 500