Chris4K commited on
Commit
a8bf13e
·
verified ·
1 Parent(s): eaa1c15

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -7
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
- await websocket.send_json({"msg": "Hello WebSocket"})
209
-
210
  ai_manager.clients.add(websocket)
211
  try:
 
 
 
 
212
  while True:
213
- data = await websocket.receive_text() # Receive messages from the user.
214
- print(f"Received: {data}") # Debug: Print the received data.
215
- ai_manager.receive_message("human", data) # Send it to the AI manager.
 
 
 
 
 
216
  except Exception as e:
217
- print(f"Connection closed with error: {e}")
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("/")