fix: gradio output handling

app.py

@@ -44,7 +44,7 @@ def generate_video(
     second_pass_flow_shift,
     second_pass_cfg,
     show_both_outputs
-) -> Union[str, List[str]]:
+) -> tuple:
     # Get model ID from selection
     model_id = MODEL_OPTIONS[model_choice]
     
@@ -194,11 +194,16 @@ def generate_video(
     
     # Return the appropriate video output(s)
     if enable_second_pass and show_both_outputs and len(output_files) > 1:
-        return output_files
+        return output_files[0], output_files[1]  # Return both first and second pass
     elif len(output_files) > 0:
-        return output_files[-1]  # Return the last generated output (either first or second pass)
+        if enable_second_pass:
+            # Return only second pass (and None for first output if showing both)
+            return None if show_both_outputs else output_files[0], output_files[0]
+        else:
+            # Return first pass only
+            return output_files[0], None
     else:
-        return "No video was generated. Please check the logs for errors."
+        return None, None
 
     # Create the Gradio interface
 with gr.Blocks() as demo:
@@ -394,14 +399,7 @@ with gr.Blocks() as demo:
                     outputs=[second_output_video]
                 )
             
-    # Define a visibility update function separately 
-    def update_second_video_visibility(enable_pass, show_both):
-        if enable_pass and show_both:
-            return gr.update(visible=True)
-        else:
-            return gr.update(visible=False)
-            
-    # Process generation without trying to update visibility in the same function
+    # Fixed output handling for Gradio
     generate_btn.click(
         fn=generate_video,
         inputs=[
@@ -427,22 +425,7 @@ with gr.Blocks() as demo:
             second_pass_cfg,
             show_both_outputs
         ],
-        outputs=[
-            output_video if not show_both_outputs else [output_video, second_output_video]
-        ]
-    )
-    
-    # Update visibility when options change
-    enable_second_pass.change(
-        fn=update_second_video_visibility,
-        inputs=[enable_second_pass, show_both_outputs],
-        outputs=[second_output_video]
-    )
-    
-    show_both_outputs.change(
-        fn=update_second_video_visibility,
-        inputs=[enable_second_pass, show_both_outputs],
-        outputs=[second_output_video]
+        outputs=[output_video, second_output_video]
     )
     
     gr.Markdown("""