Update app.py

app.py

@@ -148,11 +148,6 @@ def messages():
                     chunk = text['chunks'][-1] if text['chunks'] else None
                     if chunk:
                         response_text.append(chunk)
-                        return create_event("content_block_delta", {
-                            "type": "content_block_delta",
-                            "index": 0,
-                            "delta": {"type": "text_delta", "text": chunk},
-                        })
 
             def on_query_complete(data):
                 response_event.set()
@@ -163,11 +158,8 @@ def messages():
 
             def on_connect_error(data):
                 logging.error(f"Connection error: {data}")
-                return 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)
             sio.on('query_progress', on_query_progress)
@@ -177,12 +169,10 @@ 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:
+                    while response_text:
                         chunk = response_text.pop(0)
                         yield create_event("content_block_delta", {
                             "type": "content_block_delta",