Update app.py

app.py

@@ -90,6 +90,13 @@ class TaskManager:
 
 task_manager = TaskManager()
 
+def make_sse_handler(task_id):
+        handler = SSELogHandler(task_id)
+
+        def sink(message):
+            asyncio.create_task(handler(str(message)))
+
+        return sink
 
 def get_available_themes():
     """扫描themes目录获取所有可用主题"""
@@ -264,15 +271,7 @@ async def run_task(task_id: str, prompt: str):
                     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))
+        logger.add(make_sse_handler(task_id))
 
         import re
         def has_log_prefix(message):