Update app.py

app.py

@@ -5,8 +5,10 @@ from datetime import datetime
 from flask import Flask, request, Response, jsonify
 import socketio
 import requests
+import logging
 
 app = Flask(__name__)
+logging.basicConfig(level=logging.INFO)
 
 # 从环境变量中获取API密钥
 API_KEY = os.environ.get('PPLX_KEY')
@@ -25,11 +27,11 @@ if proxy_url:
 else:
     transport = None
 
-sio = socketio.Client(http_session=transport)
+sio = socketio.Client(http_session=transport, logger=True, engineio_logger=True)
 
 # 连接选项
 connect_opts = {
-    'transports': ['websocket'],
+    'transports': ['websocket', 'polling'],  # 允许回退到轮询
 }
 
 # 其他选项
@@ -50,7 +52,7 @@ def log_request(req, status):
     timestamp = datetime.now().isoformat()
     ip = req.remote_addr
     route = req.path
-    print(f"{timestamp} - {ip} - {route} - {status}")
+    logging.info(f"{timestamp} - {ip} - {route} - {status}")
 
 def validate_api_key():
     api_key = request.headers.get('x-api-key')
@@ -122,8 +124,9 @@ def messages():
             yield create_event("content_block_start", {"type": "content_block_start", "index": 0, "content_block": {"type": "text", "text": ""}})
             yield create_event("ping", {"type": "ping"})
 
-            @sio.on('connect')
-            def on_connect():
+            @sio.event
+            def connect():
+                logging.info("Connected to Perplexity AI")
                 sio.emit('perplexity_ask', previous_messages, {
                     "version": "2.9",
                     "source": "default",
@@ -141,8 +144,8 @@ def messages():
                     "query_source": "home"
                 })
 
-            @sio.on('query_progress')
-            def on_query_progress(data):
+            @sio.event
+            def query_progress(data):
                 if 'text' in data:
                     text = json.loads(data['text'])
                     chunk = text['chunks'][-1] if text['chunks'] else None
@@ -153,30 +156,32 @@ def messages():
                             "delta": {"type": "text_delta", "text": chunk},
                         })
 
-            @sio.on('disconnect')
-            def on_disconnect():
-                print(" > [Disconnected]")
+            @sio.event
+            def disconnect():
+                logging.info("Disconnected from Perplexity AI")
 
-            @sio.on('error')
-            def on_error(error):
-                print(f"Socket error: {error}")
+            @sio.event
+            def connect_error(data):
+                logging.error(f"Connection error: {data}")
                 yield create_event("content_block_delta", {
                     "type": "content_block_delta",
                     "index": 0,
-                    "delta": {"type": "text_delta", "text": "Error occurred while fetching output 输出时出现错误\nPlease refer to the log for more information 请查看日志以获取更多信息"},
+                    "delta": {"type": "text_delta", "text": f"Error connecting to Perplexity AI: {data}"},
                 })
-                yield create_event("content_block_stop", {"type": "content_block_stop", "index": 0})
-                yield create_event("message_delta", {
-                    "type": "message_delta",
-                    "delta": {"stop_reason": "end_turn", "stop_sequence": None},
-                    "usage": {"output_tokens": 12},
-                })
-                yield create_event("message_stop", {"type": "message_stop"})
-                log_request(request, 500)
 
-            sio.connect('wss://www.perplexity.ai/', **connect_opts, headers=sio_opts['extraHeaders'])
-            sio.wait()
-            sio.disconnect()
+            try:
+                sio.connect('wss://www.perplexity.ai/', **connect_opts, headers=sio_opts['extraHeaders'])
+                sio.wait()
+            except Exception as e:
+                logging.error(f"Error during socket connection: {str(e)}")
+                yield create_event("content_block_delta", {
+                    "type": "content_block_delta",
+                    "index": 0,
+                    "delta": {"type": "text_delta", "text": f"Error during socket connection: {str(e)}"},
+                })
+            finally:
+                if sio.connected:
+                    sio.disconnect()
 
             yield create_event("content_block_stop", {"type": "content_block_stop", "index": 0})
             yield create_event("message_delta", {
@@ -190,7 +195,7 @@ def messages():
         return Response(generate(), content_type='text/event-stream')
 
     except Exception as e:
-        print(f"Request error: {str(e)}")
+        logging.error(f"Request error: {str(e)}")
         log_request(request, 400)
         return jsonify({"error": str(e)}), 400
 
@@ -201,7 +206,7 @@ def not_found(error):
 
 @app.errorhandler(500)
 def server_error(error):
-    print(f"Server error: {str(error)}")
+    logging.error(f"Server error: {str(error)}")
     log_request(request, 500)
     return "Something broke!", 500
 
@@ -212,7 +217,7 @@ def create_event(event, data):
 
 if __name__ == '__main__':
     port = int(os.environ.get('PORT', 8081))
-    print(f"Perplexity proxy listening on port {port}")
+    logging.info(f"Perplexity proxy listening on port {port}")
     if not API_KEY:
-        print("Warning: PPLX_KEY environment variable is not set. API key validation will fail.")
+        logging.warning("Warning: PPLX_KEY environment variable is not set. API key validation will fail.")
     app.run(host='0.0.0.0', port=port)