Update app.py

app.py

@@ -105,7 +105,6 @@ def messages():
             msg_id = str(uuid.uuid4())
             response_event = Event()
             response_text = []
-            final_received = False
 
             yield create_event("message_start", {
                 "type": "message_start",
@@ -144,10 +143,9 @@ def messages():
                 sio.emit('perplexity_ask', (previous_messages, emit_data))
 
             def on_query_progress(data):
-                nonlocal response_text, final_received
+                nonlocal response_text
                 if 'text' in data:
                     text = json.loads(data['text'])
-                    final_received = text.get('final', False)
                     chunks = text.get('chunks', [])
                     for chunk in chunks:
                         decoded_chunk = chunk.encode('utf-8').decode('unicode_escape')
@@ -159,6 +157,8 @@ def messages():
                         })
 
             def on_query_complete(data):
+                nonlocal response_event
+                logging.info("Query complete")
                 response_event.set()
 
             def on_disconnect():
@@ -182,10 +182,7 @@ def messages():
 
             try:
                 sio.connect('wss://www.perplexity.ai/', **connect_opts, headers=sio_opts['extraHeaders'])
-
-                while not final_received:
-                    sio.sleep(0.1)
-
+                response_event.wait(timeout=30)  # 等待响应完成，最多30秒
             except Exception as e:
                 logging.error(f"Error during socket connection: {str(e)}")
                 yield create_event("content_block_delta", {