Update app.py

app.py

@@ -682,9 +682,9 @@ with gr.Blocks(title="Self-Forcing Streaming Demo") as demo:
     
     # 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_frame_generator,
         inputs=[prompt, seed, fps],
-        outputs=[streaming_video, status_display]
+        outputs=[streaming_video, status_display, final_video]
     )
     
     # Function to handle download button click