Update app.py

app.py

@@ -2,12 +2,11 @@ import os
 import json
 import uuid
 from datetime import datetime
-from flask import Flask, request, Response, jsonify, copy_current_request_context
+from flask import Flask, request, Response, jsonify
 import socketio
 import requests
 import logging
 from threading import Event
-from functools import partial
 
 app = Flask(__name__)
 logging.basicConfig(level=logging.INFO)
@@ -123,7 +122,6 @@ 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"})
 
-            @copy_current_request_context
             def on_connect():
                 logging.info("Connected to Perplexity AI")
                 sio.emit('perplexity_ask', previous_messages, {
@@ -143,8 +141,8 @@ def messages():
                     "query_source": "home"
                 })
 
-            @copy_current_request_context
             def on_query_progress(data):
+                nonlocal response_text
                 if 'text' in data:
                     text = json.loads(data['text'])
                     chunk = text['chunks'][-1] if text['chunks'] else None
@@ -156,16 +154,13 @@ def messages():
                             "delta": {"type": "text_delta", "text": chunk},
                         })
 
-            @copy_current_request_context
             def on_query_complete(data):
                 response_event.set()
 
-            @copy_current_request_context
             def on_disconnect():
                 logging.info("Disconnected from Perplexity AI")
                 response_event.set()
 
-            @copy_current_request_context
             def on_connect_error(data):
                 logging.error(f"Connection error: {data}")
                 return create_event("content_block_delta", {
@@ -183,6 +178,18 @@ def messages():
             try:
                 sio.connect('wss://www.perplexity.ai/', **connect_opts, headers=sio_opts['extraHeaders'])
                 response_event.wait(timeout=30)  # 等待响应，最多30秒
+                
+                # 处理接收到的响应
+                while not response_event.is_set():
+                    sio.sleep(0.1)
+                    if response_text:
+                        chunk = response_text.pop(0)
+                        yield create_event("content_block_delta", {
+                            "type": "content_block_delta",
+                            "index": 0,
+                            "delta": {"type": "text_delta", "text": chunk},
+                        })
+                
             except Exception as e:
                 logging.error(f"Error during socket connection: {str(e)}")
                 yield create_event("content_block_delta", {