Update app.py

app.py

@@ -2,11 +2,12 @@ import os
 import json
 import uuid
 from datetime import datetime
-from flask import Flask, request, Response, jsonify, current_app
+from flask import Flask, request, Response, jsonify, copy_current_request_context
 import socketio
 import requests
 import logging
 from threading import Event
+from functools import partial
 
 app = Flask(__name__)
 logging.basicConfig(level=logging.INFO)
@@ -122,8 +123,8 @@ 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"})
 
-            @sio.event
-            def connect():
+            @copy_current_request_context
+            def on_connect():
                 logging.info("Connected to Perplexity AI")
                 sio.emit('perplexity_ask', previous_messages, {
                     "version": "2.9",
@@ -142,37 +143,42 @@ def messages():
                     "query_source": "home"
                 })
 
-            @sio.event
-            def query_progress(data):
+            @copy_current_request_context
+            def on_query_progress(data):
                 if 'text' in data:
                     text = json.loads(data['text'])
                     chunk = text['chunks'][-1] if text['chunks'] else None
                     if chunk:
                         response_text.append(chunk)
-                        yield create_event("content_block_delta", {
+                        return create_event("content_block_delta", {
                             "type": "content_block_delta",
                             "index": 0,
                             "delta": {"type": "text_delta", "text": chunk},
                         })
 
-            @sio.event
-            def query_complete(data):
+            @copy_current_request_context
+            def on_query_complete(data):
                 response_event.set()
 
-            @sio.event
-            def disconnect():
+            @copy_current_request_context
+            def on_disconnect():
                 logging.info("Disconnected from Perplexity AI")
                 response_event.set()
 
-            @sio.event
-            def connect_error(data):
+            @copy_current_request_context
+            def on_connect_error(data):
                 logging.error(f"Connection error: {data}")
-                yield create_event("content_block_delta", {
+                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_event.set()
+
+            sio.on('connect', on_connect)
+            sio.on('query_progress', on_query_progress)
+            sio.on('query_complete', on_query_complete)
+            sio.on('disconnect', on_disconnect)
+            sio.on('connect_error', on_connect_error)
 
             try:
                 sio.connect('wss://www.perplexity.ai/', **connect_opts, headers=sio_opts['extraHeaders'])