Update app.py

app.py

@@ -146,19 +146,11 @@ def messages():
                 nonlocal response_text
                 if 'text' in data:
                     text = json.loads(data['text'])
-                    chunks = text.get('chunks', [])
-                    for chunk in chunks:
-                        decoded_chunk = chunk.encode('utf-8').decode('unicode_escape')
-                        response_text.append(decoded_chunk)
-                        yield create_event("content_block_delta", {
-                            "type": "content_block_delta",
-                            "index": 0,
-                            "delta": {"type": "text_delta", "text": decoded_chunk},
-                        })
+                    chunk = text['chunks'][-1] if text['chunks'] else None
+                    if chunk:
+                        response_text.append(chunk)
 
             def on_query_complete(data):
-                nonlocal response_event
-                logging.info("Query complete")
                 response_event.set()
 
             def on_disconnect():
@@ -167,11 +159,7 @@ def messages():
 
             def on_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": f"Error connecting to Perplexity AI: {data}"},
-                })
+                response_text.append(f"Error connecting to Perplexity AI: {data}")
                 response_event.set()
 
             sio.on('connect', on_connect)
@@ -182,7 +170,17 @@ 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)
+                    while 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", {