Update app.py

app.py

@@ -114,6 +114,43 @@ async def run_task(task_id: str, prompt: str):
         await task_manager.fail_task(task_id, str(e))
 
 
+@app.get("/tasks/{task_id}/events")
+async def task_events(task_id: str):
+    async def event_generator():
+        if task_id not in task_manager.queues:
+            yield f"event: error\ndata: {dumps({'message': 'Task not found'})}\n\n"
+            return
+
+        queue = task_manager.queues[task_id]
+        task = task_manager.tasks.get(task_id)
+        if task:
+            yield f"event: status\ndata: {dumps({'type': 'status', 'status': task.status, 'steps': task.steps})}\n\n"
+
+        while True:
+            try:
+                event = await queue.get()
+                formatted_event = dumps(event)
+                yield ": heartbeat\n\n"
+                yield f"event: {event['type']}\ndata: {formatted_event}\n\n"
+
+                if event["type"] in ["complete", "error"]:
+                    break
+
+            except asyncio.CancelledError:
+                break
+            except Exception as e:
+                yield f"event: error\ndata: {dumps({'message': str(e)})}\n\n"
+                break
+
+    return StreamingResponse(
+        event_generator(),
+        media_type="text/event-stream",
+        headers={
+            "Cache-Control": "no-cache",
+            "Connection": "keep-alive",
+            "X-Accel-Buffering": "no",
+        },
+    )