Update app.py

app.py

@@ -3,6 +3,11 @@ import json
 from datetime import datetime
 import urllib.request
 import ssl
+import logging
+
+# 设置日志
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 class ChatCompletionsHandler(BaseHTTPRequestHandler):
     def do_POST(self):
@@ -45,65 +50,78 @@ class ChatCompletionsHandler(BaseHTTPRequestHandler):
         ctx.check_hostname = False
         ctx.verify_mode = ssl.CERT_NONE
 
-        with urllib.request.urlopen(req, context=ctx) as response:
-            response_body = json.loads(response.read().decode('utf-8'))
-
-        image_url = response_body['images'][0]['url']
-        unique_id = int(datetime.now().timestamp() * 1000)
-        current_timestamp = int(unique_id / 1000)
-
-        if stream:
-            response_payload = {
-                "id": unique_id,
-                "object": "chat.completion.chunk",
-                "created": current_timestamp,
-                "model": model,
-                "choices": [
-                    {
-                        "index": 0,
-                        "delta": {
-                            "content": f"![]({image_url})"
-                        },
-                        "finish_reason": "stop"
-                    }
-                ]
-            }
-            self.send_response(200)
-            self.send_header('Content-Type', 'text/event-stream')
-            self.end_headers()
-            self.wfile.write(f"data: {json.dumps(response_payload)}\n\n".encode('utf-8'))
-        else:
-            response_payload = {
-                "id": unique_id,
-                "object": "chat.completion",
-                "created": current_timestamp,
-                "model": model,
-                "choices": [
-                    {
-                        "index": 0,
-                        "message": {
-                            "role": "assistant",
-                            "content": f"![]({image_url})"
-                        },
-                        "logprobs": None,
-                        "finish_reason": "length"
+        try:
+            with urllib.request.urlopen(req, context=ctx) as response:
+                response_body = json.loads(response.read().decode('utf-8'))
+            
+            logger.info(f"API Response: {response_body}")  # 记录 API 响应
+
+            if 'images' not in response_body or not response_body['images']:
+                raise ValueError("No images in the response")
+
+            image_url = response_body['images'][0].get('url')
+            if not image_url:
+                raise ValueError("No URL in the image response")
+
+            unique_id = int(datetime.now().timestamp() * 1000)
+            current_timestamp = int(unique_id / 1000)
+
+            if stream:
+                response_payload = {
+                    "id": unique_id,
+                    "object": "chat.completion.chunk",
+                    "created": current_timestamp,
+                    "model": model,
+                    "choices": [
+                        {
+                            "index": 0,
+                            "delta": {
+                                "content": f"![]({image_url})"
+                            },
+                            "finish_reason": "stop"
+                        }
+                    ]
+                }
+                self.send_response(200)
+                self.send_header('Content-Type', 'text/event-stream')
+                self.end_headers()
+                self.wfile.write(f"data: {json.dumps(response_payload)}\n\n".encode('utf-8'))
+            else:
+                response_payload = {
+                    "id": unique_id,
+                    "object": "chat.completion",
+                    "created": current_timestamp,
+                    "model": model,
+                    "choices": [
+                        {
+                            "index": 0,
+                            "message": {
+                                "role": "assistant",
+                                "content": f"![]({image_url})"
+                            },
+                            "logprobs": None,
+                            "finish_reason": "length"
+                        }
+                    ],
+                    "usage": {
+                        "prompt_tokens": len(prompt),
+                        "completion_tokens": len(image_url),
+                        "total_tokens": len(prompt) + len(image_url)
                     }
-                ],
-                "usage": {
-                    "prompt_tokens": len(prompt),
-                    "completion_tokens": len(image_url),
-                    "total_tokens": len(prompt) + len(image_url)
                 }
-            }
-            self.send_response(200)
-            self.send_header('Content-Type', 'application/json')
-            self.end_headers()
-            self.wfile.write(json.dumps(response_payload).encode('utf-8'))
+                self.send_response(200)
+                self.send_header('Content-Type', 'application/json')
+                self.end_headers()
+                self.wfile.write(json.dumps(response_payload).encode('utf-8'))
+
+        except Exception as e:
+            logger.error(f"Error occurred: {str(e)}")
+            self.send_error(500, f"Internal Server Error: {str(e)}")
 
 def run(server_class=HTTPServer, handler_class=ChatCompletionsHandler, port=8000):
     server_address = ('', port)
     httpd = server_class(server_address, handler_class)
-    print(f"Starting server on port {port}")
+    logger.info(f"Starting server on port {port}")
     httpd.serve_forever()
 
 if __name__ == '__main__':