Update app.py

app.py

@@ -14,12 +14,21 @@ API_KEY = os.environ.get('PPLX_KEY')
 # 代理设置
 proxy_url = os.environ.get('PROXY_URL')
 
+# 连接选项
+connect_opts = {
+    'transports': ['websocket'],
+}
+
+# 如果设置了代理，添加到连接选项中
+if proxy_url:
+    connect_opts['http'] = proxy_url
+    connect_opts['https'] = proxy_url
+
+# 其他选项
 sio_opts = {
     'auth': {
         'jwt': 'anonymous-ask-user',
     },
-    'reconnection': False,
-    'transports': ['websocket'],
     'extraHeaders': {
         'Cookie': os.environ.get('PPLX_COOKIE'),
         'User-Agent': os.environ.get('USER_AGENT'),
@@ -29,11 +38,6 @@ sio_opts = {
     }
 }
 
-# 如果设置了代理，添加到 socketio 选项中
-if proxy_url:
-    sio_opts['http'] = proxy_url
-    sio_opts['https'] = proxy_url
-
 def log_request(req, status):
     timestamp = datetime.now().isoformat()
     ip = req.remote_addr
@@ -162,7 +166,7 @@ def messages():
                 yield create_event("message_stop", {"type": "message_stop"})
                 log_request(request, 500)
 
-            sio.connect('wss://www.perplexity.ai/', **sio_opts)
+            sio.connect('wss://www.perplexity.ai/', **connect_opts, headers=sio_opts['extraHeaders'])
             sio.wait()
             sio.disconnect()