Update app.py

app.py

@@ -128,8 +128,11 @@ class AIStateManager:
 
     async def broadcast(self, message):
         """Sends message to all connected WebSocket clients."""
-        for ws in self.clients:
-            await ws.send_text(message)
+        print("sending broadcast message")
+        for client in ai_manager.clients:
+            await client.send_text(f"Received: {data}")
+        #for ws in self.clients:
+        #    await ws.send_text(message)
 
     async def heartbeat(self):
         """Basic heartbeat loop."""
@@ -142,6 +145,7 @@ class AIStateManager:
     async def consciousness(self):
         """Controls research/sleep cycle."""
         while True:
+            self.broadcast("one, two three...")
             await asyncio.sleep(2)  # Consciousness checks every 2 sec
             with self.lock:
                 if self.state == "awake":
@@ -197,25 +201,33 @@ threading.Thread(target=ai_manager.modify_research_tasks, daemon=True).start()
 
 @app.websocket("/ws")
 async def websocket_endpoint(websocket: WebSocket):
+    print("WebSocket endpoint called")
     await websocket.accept()
-    print("WebSocket connection accepted")
-    ai_manager.clients.add(websocket)
+    print(f"WebSocket connection accepted from {websocket.client}")
     
     try:
-        # Initial welcome message
+        # Send initial message explicitly
         await websocket.send_text("Connected to AI State Machine")
+        print("Initial message sent to WebSocket") 
+        
+            # Directly broadcast the message
+            #for client in ai_manager.clients:
+            #    await client.send_text(f"Received: {data}")
+            
         
         while True:
             data = await websocket.receive_text()
-            print(f"Received message: {data}")
-            
-            # Directly broadcast the message
-            for client in ai_manager.clients:
-                await client.send_text(f"Received: {data}")
+            print(f"BACKEND RECEIVED: {data}")
             
+            # Echo the message back
+            response = f"Echo: {data}"
+            await websocket.send_text(response)
+            print(f"BACKEND SENT: {response}")
     except Exception as e:
-        print(f"WebSocket error: {e}")
-        ai_manager.clients.remove(websocket)
+        print(f"WebSocket ERROR: {e}")
+        # Log full traceback
+        import traceback
+        traceback.print_exc()
 
 @app.get("/")
 async def get():