Update app.py

app.py

@@ -146,6 +146,7 @@ class AIStateManager:
                     if self.heartbeat_count >= 20:
                         self.set_state("research")  # Restart research after sleep
                         asyncio.create_task(self.run_research())
+                print(self.state)
 
     async def run_research(self):
         """Runs research tasks in order."""
@@ -189,10 +190,11 @@ async def websocket_endpoint(websocket: WebSocket):
     ai_manager.clients.add(websocket)
     try:
         while True:
-            data = await websocket.receive_text()
-            print(data)
-            ai_manager.receive_message("human", data)
-    except Exception:
+            data = await websocket.receive_text()  # Receive messages from the user.
+            print(f"Received: {data}")  # Debug: Print the received data.
+            ai_manager.receive_message("human", data)  # Send it to the AI manager.
+    except Exception as e:
+        print(f"Connection closed with error: {e}")
         ai_manager.clients.remove(websocket)
 
 @app.get("/")