Update app.py

app.py

@@ -32,7 +32,7 @@ import imageio # Added for final video rendering
 
 # FastRTC imports
 from fastrtc import WebRTC, get_turn_credentials
-from fastrtc.utils import AdditionalOutputs, CloseStream
+from fastrtc.utils import AdditionalOutputs #, CloseStream
 
 # Original project imports
 from pipeline import CausalInferencePipeline
@@ -297,7 +297,7 @@ def video_generation_handler(prompt, seed, enable_torch_compile, enable_fp8, use
                     print(f"⚠️ Could not save final video: {e}")
 
                 yield frame_bgr, AdditionalOutputs(status_html, video_update, gr.update(visible=False))
-                yield CloseStream("🎉 Video generation completed successfully!")
+                # yield CloseStream("🎉 Video generation completed successfully!")
                 return
             else:  # Regular frames - simpler status
                 status_html = (
@@ -321,7 +321,7 @@ def video_generation_handler(prompt, seed, enable_torch_compile, enable_fp8, use
     print(f"✅ Video generation completed! Total frames yielded: {total_frames_yielded}")
     
     # Signal completion
-    yield CloseStream("🎉 Video generation completed successfully!")
+    # yield CloseStream("🎉 Video generation completed successfully!")
 
 # --- Gradio UI Layout ---
 with gr.Blocks(theme=gr.themes.Soft(), title="Self-Forcing FastRTC Demo") as demo: