Update app.py

app.py

@@ -270,7 +270,15 @@ async def run_task(task_id: str, prompt: str):
                 await task_manager.update_task_step(
                     self.task_id, 0, cleaned_message, event_type
                 )
+        
+        def make_sse_handler(task_id):
+            handler = SSELogHandler(task_id)
+
+            def sink(message):
+                asyncio.create_task(handler(str(message)))
 
+            return sink
+            
         logger.add(make_sse_handler(task_id))
 
         import re