Update app.py

app.py

@@ -1,21 +1,26 @@
+from http.server import HTTPServer, BaseHTTPRequestHandler
 import json
 from datetime import datetime
-import aiohttp
-from aiohttp import web
+import urllib.request
+import ssl
 
-@app.route('/')
-def index():
-    return "flux2api with siliconflow", 200
+class ChatCompletionsHandler(BaseHTTPRequestHandler):
+    def do_POST(self):
+        if self.path != '/ai/v1/chat/completions':
+            self.send_error(404, "Not Found")
+            return
+
+        content_length = int(self.headers['Content-Length'])
+        post_data = self.rfile.read(content_length)
+        body = json.loads(post_data.decode('utf-8'))
 
-async def handle_chat_completions(request):
-    try:
-        body = await request.json()
         model = body.get('model')
         messages = body.get('messages')
         stream = body.get('stream', False)
 
         if not model or not messages or len(messages) == 0:
-            return web.json_response({'error': 'Bad Request: Missing required fields'}, status=400)
+            self.send_error(400, "Bad Request: Missing required fields")
+            return
 
         prompt = messages[-1]['content']
         new_url = f"https://api.siliconflow.cn/v1/{model}/text-to-image"
@@ -27,20 +32,27 @@ async def handle_chat_completions(request):
             "guidance_scale": 1
         }
 
-        async with aiohttp.ClientSession() as session:
-            async with session.post(new_url, json=new_request_body, headers={
-                'accept': 'application/json',
-                'content-type': 'application/json',
-                'Authorization': request.headers.get('Authorization')
-            }) as response:
-                response_body = await response.json()
+        req = urllib.request.Request(new_url, 
+                                     data=json.dumps(new_request_body).encode('utf-8'), 
+                                     headers={
+                                         'accept': 'application/json',
+                                         'content-type': 'application/json',
+                                         'Authorization': self.headers.get('Authorization')
+                                     },
+                                     method='POST')
+
+        ctx = ssl.create_default_context()
+        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",
@@ -56,12 +68,11 @@ async def handle_chat_completions(request):
                     }
                 ]
             }
-            data_string = json.dumps(response_payload)
-            return web.Response(text=f"data: {data_string}\n\n", 
-                                status=200, 
-                                content_type='text/event-stream')
+            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",
@@ -84,14 +95,16 @@ async def handle_chat_completions(request):
                     "total_tokens": len(prompt) + len(image_url)
                 }
             }
-            return web.json_response(response_payload)
-
-    except Exception as error:
-        return web.json_response({'error': f"Internal Server Error: {str(error)}"}, status=500)
-
-app = web.Application()
-app.router.add_post('/ai/v1/chat/completions', handle_chat_completions)
+            self.send_response(200)
+            self.send_header('Content-Type', 'application/json')
+            self.end_headers()
+            self.wfile.write(json.dumps(response_payload).encode('utf-8'))
 
+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}")
+    httpd.serve_forever()
 
 if __name__ == '__main__':
-    web.run_app(app, port=8000)
+    run()