Update app.py

app.py

@@ -111,6 +111,8 @@ async def run_task(task_id: str, prompt: str):
 
 @app.get("/tasks/{task_id}/events")
 async def task_events(task_id: str):
+    logger.info(f"Client subscribed to events for task: {task_id}")
+    
     async def event_generator():
         if task_id not in task_manager.queues:
             yield f"event: error\ndata: {dumps({'message': 'Task not found'})}\n\n"
@@ -119,6 +121,7 @@ async def task_events(task_id: str):
         queue = task_manager.queues[task_id]
         task = task_manager.tasks.get(task_id)
         if task:
+            logger.info(f"Sending initial task status for: {task_id}")
             yield f"event: status\ndata: {dumps({'type': 'status', 'status': task.status, 'steps': task.steps})}\n\n"
 
         while True:
@@ -132,8 +135,10 @@ async def task_events(task_id: str):
                     break
 
             except asyncio.CancelledError:
+                logger.warning("Client disconnected")
                 break
             except Exception as e:
+                logger.error(f"Event stream error: {e}")
                 yield f"event: error\ndata: {dumps({'message': str(e)})}\n\n"
                 break