update processing

app.py

@@ -10,6 +10,7 @@ Architecture:
 """
 
 import os
+import threading
 
 import gradio as gr
 from openai import OpenAI
@@ -134,18 +135,32 @@ def run_generation(image, prompt, seed, seconds):
         yield None, "", str(e)
         return
 
-    # Yield so the user sees the enhanced prompt; Gradio's timer runs during Step 2
+    # Yield so the user sees the enhanced prompt
     yield None, enhanced, "Generating video ..."
 
-    # Step 2: Send to router
+    # Step 2: Send to router in background thread, yield periodically to keep
+    # Gradio's processing indicator alive (SSE connection drops if idle too long)
     print(f"[Generate] Sending to router ...")
-    result = generate(
-        image=image,
-        video_prompt=enhanced,
-        seed=int(seed),
-        output_dir=OUTPUT_DIR,
-        seconds=int(seconds),
-    )
+    gen_result = [None]
+    gen_done = threading.Event()
+
+    def _do_generate():
+        gen_result[0] = generate(
+            image=image,
+            video_prompt=enhanced,
+            seed=int(seed),
+            output_dir=OUTPUT_DIR,
+            seconds=int(seconds),
+        )
+        gen_done.set()
+
+    threading.Thread(target=_do_generate, daemon=True).start()
+
+    # Keep yielding same values every 10s to maintain the connection
+    while not gen_done.wait(timeout=10):
+        yield None, enhanced, "Generating video ..."
+
+    result = gen_result[0]
 
     if result["error"]:
         print(f"[Generate] Error from router: {result['error']}")