Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -203,18 +203,26 @@ threading.Thread(target=ai_manager.modify_research_tasks, daemon=True).start()
|
|
203 |
|
204 |
@app.websocket("/ws")
|
205 |
async def websocket_endpoint(websocket: WebSocket):
|
206 |
-
"""WebSocket connection handler."""
|
207 |
await websocket.accept()
|
208 |
-
|
209 |
-
|
210 |
ai_manager.clients.add(websocket)
|
211 |
try:
|
|
|
|
|
|
|
|
|
212 |
while True:
|
213 |
-
data = await websocket.receive_text()
|
214 |
-
print(f"Received: {data}") #
|
215 |
-
|
|
|
|
|
|
|
|
|
|
|
216 |
except Exception as e:
|
217 |
-
print(f"
|
218 |
ai_manager.clients.remove(websocket)
|
219 |
|
220 |
@app.get("/")
|
|
|
203 |
|
204 |
@app.websocket("/ws")
|
205 |
async def websocket_endpoint(websocket: WebSocket):
|
|
|
206 |
await websocket.accept()
|
207 |
+
print("WebSocket connection accepted") # Add explicit logging
|
208 |
+
|
209 |
ai_manager.clients.add(websocket)
|
210 |
try:
|
211 |
+
# Send initial welcome message
|
212 |
+
await websocket.send_text("Connected to AI State Machine")
|
213 |
+
print("Sent initial welcome message") # Debug logging
|
214 |
+
|
215 |
while True:
|
216 |
+
data = await websocket.receive_text()
|
217 |
+
print(f"Received WebSocket message: {data}") # Explicit receive logging
|
218 |
+
|
219 |
+
# Process message through AI manager
|
220 |
+
ai_manager.receive_message("human", data)
|
221 |
+
|
222 |
+
# Ensure broadcast is working
|
223 |
+
await ai_manager.broadcast(f"Received: {data}")
|
224 |
except Exception as e:
|
225 |
+
print(f"WebSocket connection error: {e}")
|
226 |
ai_manager.clients.remove(websocket)
|
227 |
|
228 |
@app.get("/")
|