Update app.py

app.py

@@ -44,7 +44,7 @@ def fn(vid, fps, color):
         pil_image = Image.fromarray(frame)
         processed_image = process(pil_image, color)
         processed_frames.append(np.array(processed_image))
-        yield processed_image, gr.update(visible=False)
+        yield processed_image, None
 
     # Create a new video from the processed frames
     processed_video = mp.ImageSequenceClip(processed_frames, fps=fps)
@@ -58,9 +58,10 @@ def fn(vid, fps, color):
     unique_filename = str(uuid.uuid4()) + ".mp4"
     temp_filepath = os.path.join(temp_dir, unique_filename)
     processed_video.write_videofile(temp_filepath, codec="libx264")
-
+    
+    yield gr.update(visible=False), gr.update(visible=True)
     # Return the path to the temporary file
-    yield gr.update(visible=False), temp_filepath
+    yield None, temp_filepath
 
 
 def process(image, color_hex):
@@ -97,7 +98,7 @@ def process_file(f, color="#00FF00"):
 with gr.Blocks() as demo:
     with gr.Row():
         in_video = gr.Video(label="Input Video")
-        stream_image = gr.Image(label="Streaming Output")
+        stream_image = gr.Image(label="Streaming Output", visible=False)
         out_video = gr.Video(label="Final Output Video")  
     submit_button = gr.Button("Change Background")
     with gr.Row():