Update app.py

app.py

@@ -522,9 +522,27 @@ with gr.Blocks(title="Self-Forcing Streaming Demo") as demo:
             download_btn = gr.Button("💾 Download Video", variant="secondary")
             download_output = gr.File(label="Download")
 
+    # Define a wrapper function to ensure we always return the expected outputs
+    def safe_video_generator(p, s, f):
+        # Clear frames from previous generation
+        APP_STATE["current_frames"] = []
+        
+        # Call the generator function and ensure it returns both video and status
+        try:
+            for video_path, status_html in video_generation_handler_streaming(p, s, f):
+                if video_path is not None:
+                    yield video_path, status_html
+                else:
+                    yield None, status_html
+        except Exception as e:
+            import traceback
+            print(f"Error in generator: {e}")
+            traceback.print_exc()
+            yield None, f"<div style='color: red; padding: 10px; border: 1px solid #ffcccc; border-radius: 5px;'>Error: {e}</div>"
+    
     # Connect the generator to the streaming video
     start_btn.click(
-        fn=lambda p, s, f: (APP_STATE.update({"current_frames": []}) or video_generation_handler_streaming(p, s, f)),
+        fn=safe_video_generator,
         inputs=[prompt, seed, fps],
         outputs=[streaming_video, status_display]
     )