Update app.py

app.py

@@ -100,7 +100,9 @@ def _generate_video_segment(input_image_path: str, output_image_path: str, promp
     )
     return result[0]["video"]
 
-def build_relight_prompt(light_type, light_direction, light_intensity, illumination_env, prompt):
+def build_relight_prompt(light_type, light_type_custom, light_direction, light_direction_custom, 
+                         light_intensity, light_intensity_custom, illumination_env, 
+                         illumination_env_custom, prompt):
     """Build the relighting prompt based on user selections - Qwen style."""
     
     # Priority 1: User's prompt (translated to Chinese if needed)
@@ -231,16 +233,28 @@ def build_relight_prompt(light_type, light_direction, light_intensity, illuminat
     }
     
     # Build the prompt - Qwen style (comma-separated, Chinese)
-    if light_type != "none":
+    # Handle custom light type
+    if light_type == "custom" and light_type_custom and light_type_custom.strip():
+        prompt_parts.append(translate_to_chinese(light_type_custom))
+    elif light_type != "none":
         prompt_parts.append(light_descriptions.get(light_type, ""))
     
-    if illumination_env != "none":
+    # Handle custom illumination environment
+    if illumination_env == "custom" and illumination_env_custom and illumination_env_custom.strip():
+        prompt_parts.append(translate_to_chinese(illumination_env_custom))
+    elif illumination_env != "none":
         prompt_parts.append(illumination_envs.get(illumination_env, ""))
     
-    if light_direction != "none":
+    # Handle custom light direction
+    if light_direction == "custom" and light_direction_custom and light_direction_custom.strip():
+        prompt_parts.append(translate_to_chinese(light_direction_custom))
+    elif light_direction != "none":
         prompt_parts.append(direction_descriptions.get(light_direction, ""))
     
-    if light_intensity != "none":
+    # Handle custom light intensity
+    if light_intensity == "custom" and light_intensity_custom and light_intensity_custom.strip():
+        prompt_parts.append(translate_to_chinese(light_intensity_custom))
+    elif light_intensity != "none":
         prompt_parts.append(intensity_descriptions.get(light_intensity, ""))
     
     final_prompt = ",".join([p for p in prompt_parts if p])
@@ -256,9 +270,13 @@ def build_relight_prompt(light_type, light_direction, light_intensity, illuminat
 def infer_relight(
     image,
     light_type,
+    light_type_custom,
     light_direction,
+    light_direction_custom,
     light_intensity,
+    light_intensity_custom,
     illumination_env,
+    illumination_env_custom,
     prompt,
     seed,
     randomize_seed,
@@ -269,7 +287,10 @@ def infer_relight(
     prev_output = None,
     progress=gr.Progress(track_tqdm=True)
 ):
-    final_prompt = build_relight_prompt(light_type, light_direction, light_intensity, illumination_env, prompt)
+    final_prompt = build_relight_prompt(light_type, light_type_custom, light_direction, 
+                                       light_direction_custom, light_intensity, 
+                                       light_intensity_custom, illumination_env, 
+                                       illumination_env_custom, prompt)
     print(f"Generated Prompt: {final_prompt}")
 
     if randomize_seed:
@@ -339,7 +360,7 @@ css = '''
 '''
 
 def reset_all():
-    return ["none", "none", "none", "none", "", False, True]
+    return ["none", "", "none", "", "none", "", "none", "", "", False, True]
 
 def end_reset():
     return False
@@ -365,6 +386,10 @@ def update_dimensions_on_upload(image):
     
     return new_width, new_height
 
+def toggle_custom_textbox(choice):
+    """Show textbox when Custom is selected"""
+    return gr.update(visible=(choice == "custom"))
+
 
 with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
     with gr.Column(elem_id="col-container"):
@@ -380,143 +405,169 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
                 prev_output = gr.Image(value=None, visible=False)
                 is_reset = gr.Checkbox(value=False, visible=False)
 
-                with gr.Accordion("💡 Light Type", open=True):
-                    light_type = gr.Radio(
-                        choices=[
-                            ("None", "none"),
-                            ("Soft Window Light", "soft_window"),
-                            ("Golden Hour", "golden_hour"),
-                            ("Studio Lighting", "studio"),
-                            ("Dramatic", "dramatic"),
-                            ("Natural Daylight", "natural"),
-                            ("Neon", "neon"),
-                            ("Candlelight", "candlelight"),
-                            ("Moonlight", "moonlight"),
-                            ("Sunrise", "sunrise"),
-                            ("Sunset over Sea", "sunset_sea"),
-                            ("Overcast", "overcast"),
-                            ("Harsh Sunlight", "harsh_sun"),
-                            ("Twilight", "twilight"),
-                            ("Aurora", "aurora"),
-                            ("Firelight", "firelight"),
-                            ("Lightning", "lightning"),
-                            ("Underwater", "underwater"),
-                            ("Foggy", "foggy"),
-                            ("Magic Light", "magic"),
-                            ("Cyberpunk", "cyberpunk"),
-                            ("Warm Home", "warm_home"),
-                            ("Cold Industrial", "cold_industrial"),
-                            ("Spotlight", "spotlight"),
-                            ("Rim Light", "rim_light"),
-                        ],
-                        value="none",
-                        elem_classes="radio-group"
-                    )
-                
-                with gr.Accordion("🧭 Light Direction", open=True):
-                    light_direction = gr.Radio(
-                        choices=[
-                            ("None", "none"),
-                            ("Front", "front"),
-                            ("Side", "side"),
-                            ("Left", "left"),
-                            ("Right", "right"),
-                            ("Back (Backlight)", "back"),
-                            ("Top", "top"),
-                            ("Bottom", "bottom"),
-                            ("Diagonal", "diagonal"),
-                        ],
-                        value="none",
-                        elem_classes="radio-group"
-                    )
-                
-                with gr.Accordion("⚡ Light Intensity", open=True):
-                    light_intensity = gr.Radio(
-                        choices=[
-                            ("None", "none"),
-                            ("Soft", "soft"),
-                            ("Medium", "medium"),
-                            ("Strong", "strong"),
-                        ],
-                        value="none",
-                        elem_classes="radio-group"
-                    )
-                
-                with gr.Accordion("🌍 Illumination Environment", open=False):
-                    illumination_env = gr.Radio(
-                        choices=[
-                            ("None", "none"),
-                            ("Sunshine from Window", "sunshine_window"),
-                            ("Neon Night, City", "neon_city"),
-                            ("Sci-Fi RGB Glowing, Cyberpunk", "sci_fi_rgb"),
-                            ("Warm Atmosphere, at Home, Bedroom", "warm_bedroom"),
-                            ("Magic Lit", "magic_lit"),
-                            ("Evil, Gothic, in a Cave", "gothic_cave"),
-                            ("Light and Shadow", "light_shadow"),
-                            ("Shadow from Window", "window_shadow"),
-                            ("Soft Studio Lighting", "soft_studio"),
-                            ("Home Atmosphere, Cozy Bedroom", "cozy_bedroom"),
-                            ("Neon, Wong Kar-wai, Warm", "wong_kar_wai"),
-                            ("Moonlight through Curtains", "moonlight_curtains"),
-                            ("Stormy Sky Lighting", "stormy_sky"),
-                            ("Underwater Glow, Deep Sea", "underwater_glow"),
-                            ("Foggy Forest at Dawn", "foggy_forest"),
-                            ("Golden Hour in a Meadow", "meadow_golden"),
-                            ("Rainbow Reflections, Neon", "rainbow_neon"),
-                            ("Apocalyptic, Smoky Atmosphere", "apocalyptic"),
-                            ("Red Glow, Emergency Lights", "emergency_red"),
-                            ("Mystical Glow, Enchanted Forest", "mystical_forest"),
-                            ("Campfire Light", "campfire"),
-                            ("Harsh, Industrial Lighting", "industrial_harsh"),
-                            ("Sunrise in the Mountains", "mountain_sunrise"),
-                            ("Evening Glow in the Desert", "desert_evening"),
-                            ("Moonlight in a Dark Alley", "dark_alley"),
-                            ("Golden Glow at a Fairground", "fairground"),
-                            ("Midnight in the Forest", "forest_midnight"),
-                            ("Purple and Pink Hues at Twilight", "twilight_purple"),
-                            ("Foggy Morning, Muted Light", "foggy_morning"),
-                            ("Candle-lit Room, Rustic Vibe", "rustic_candle"),
-                            ("Fluorescent Office Lighting", "office_fluorescent"),
-                            ("Lightning Flash in Storm", "storm_lightning"),
-                            ("Night, Cozy Warm Light from Fireplace", "fireplace_night"),
-                            ("Ethereal Glow, Magical Forest", "ethereal_magic"),
-                            ("Dusky Evening on a Beach", "beach_dusky"),
-                            ("Afternoon Light Filtering through Trees", "trees_afternoon"),
-                            ("Blue Neon Light, Urban Street", "urban_blue_neon"),
-                            ("Red and Blue Police Lights in Rain", "rain_police"),
-                            ("Aurora Borealis Glow, Arctic Landscape", "aurora_arctic"),
-                            ("Sunrise through Foggy Mountains", "foggy_mountains"),
-                            ("Golden Hour on a City Skyline", "city_skyline"),
-                            ("Mysterious Twilight, Heavy Mist", "twilight_mist"),
-                            ("Early Morning Rays, Forest Clearing", "forest_rays"),
-                            ("Colorful Lantern Light at Festival", "festival_lantern"),
-                            ("Soft Glow through Stained Glass", "stained_glass"),
-                            ("Harsh Spotlight in Dark Room", "dark_spotlight"),
-                            ("Mellow Evening Glow on a Lake", "lake_evening"),
-                            ("Crystal Reflections in a Cave", "cave_crystal"),
-                            ("Vibrant Autumn Lighting in a Forest", "autumn_forest"),
-                            ("Gentle Snowfall at Dusk", "snowfall_dusk"),
-                            ("Hazy Light of a Winter Morning", "winter_hazy"),
-                            ("Rain-soaked Reflections in City Lights", "rain_city"),
-                            ("Golden Sunlight Streaming through Trees", "trees_golden_sun"),
-                            ("Fireflies Lighting up a Summer Night", "fireflies_summer"),
-                            ("Glowing Embers from a Forge", "forge_embers"),
-                            ("Dim Candlelight in a Gothic Castle", "gothic_castle"),
-                            ("Midnight Sky with Bright Starlight", "starlight_midnight"),
-                            ("Warm Sunset in a Rural Village", "rural_sunset"),
-                            ("Flickering Light in a Haunted House", "haunted_flicker"),
-                            ("Desert Sunset with Mirage-like Glow", "desert_mirage"),
-                            ("Golden Beams Piercing through Storm Clouds", "storm_beams"),
-                        ],
-                        value="none",
-                        elem_classes="radio-group"
-                    )
-                
-                with gr.Accordion("✍️ Custom Prompt", open=False):
-                    prompt = gr.Textbox(
-                        placeholder="Example: Add warm sunset lighting from the right",
-                        lines=3
-                    )
+                with gr.Tab("Compose Prompt"):
+                    with gr.Accordion("💡 Light Type", open=True):
+                        light_type = gr.Radio(
+                            choices=[
+                                ("None", "none"),
+                                ("Soft Window Light", "soft_window"),
+                                ("Golden Hour", "golden_hour"),
+                                ("Studio Lighting", "studio"),
+                                ("Dramatic", "dramatic"),
+                                ("Natural Daylight", "natural"),
+                                ("Neon", "neon"),
+                                ("Candlelight", "candlelight"),
+                                ("Moonlight", "moonlight"),
+                                ("Sunrise", "sunrise"),
+                                ("Sunset over Sea", "sunset_sea"),
+                                ("Overcast", "overcast"),
+                                ("Harsh Sunlight", "harsh_sun"),
+                                ("Twilight", "twilight"),
+                                ("Aurora", "aurora"),
+                                ("Firelight", "firelight"),
+                                ("Lightning", "lightning"),
+                                ("Underwater", "underwater"),
+                                ("Foggy", "foggy"),
+                                ("Magic Light", "magic"),
+                                ("Cyberpunk", "cyberpunk"),
+                                ("Warm Home", "warm_home"),
+                                ("Cold Industrial", "cold_industrial"),
+                                ("Spotlight", "spotlight"),
+                                ("Rim Light", "rim_light"),
+                                ("Custom", "custom"),
+                            ],
+                            value="none",
+                            elem_classes="radio-group"
+                        )
+                        light_type_custom = gr.Textbox(
+                            label="Custom Light Type",
+                            placeholder="e.g., Bioluminescent glow, Laser light show, etc.",
+                            visible=False
+                        )
+                    
+                    with gr.Accordion("🧭 Light Direction", open=True):
+                        light_direction = gr.Radio(
+                            choices=[
+                                ("None", "none"),
+                                ("Front", "front"),
+                                ("Side", "side"),
+                                ("Left", "left"),
+                                ("Right", "right"),
+                                ("Back (Backlight)", "back"),
+                                ("Top", "top"),
+                                ("Bottom", "bottom"),
+                                ("Diagonal", "diagonal"),
+                                ("Custom", "custom"),
+                            ],
+                            value="none",
+                            elem_classes="radio-group"
+                        )
+                        light_direction_custom = gr.Textbox(
+                            label="Custom Light Direction",
+                            placeholder="e.g., From 45 degrees above left, Rotating around subject, etc.",
+                            visible=False
+                        )
+                    
+                    with gr.Accordion("⚡ Light Intensity", open=True):
+                        light_intensity = gr.Radio(
+                            choices=[
+                                ("None", "none"),
+                                ("Soft", "soft"),
+                                ("Medium", "medium"),
+                                ("Strong", "strong"),
+                                ("Custom", "custom"),
+                            ],
+                            value="none",
+                            elem_classes="radio-group"
+                        )
+                        light_intensity_custom = gr.Textbox(
+                            label="Custom Light Intensity",
+                            placeholder="e.g., Very dim, Blinding bright, Pulsating, etc.",
+                            visible=False
+                        )
+                    
+                    with gr.Accordion("🌍 Illumination Environment", open=False):
+                        illumination_env = gr.Radio(
+                            choices=[
+                                ("None", "none"),
+                                ("Sunshine from Window", "sunshine_window"),
+                                ("Neon Night, City", "neon_city"),
+                                ("Sci-Fi RGB Glowing, Cyberpunk", "sci_fi_rgb"),
+                                ("Warm Atmosphere, at Home, Bedroom", "warm_bedroom"),
+                                ("Magic Lit", "magic_lit"),
+                                ("Evil, Gothic, in a Cave", "gothic_cave"),
+                                ("Light and Shadow", "light_shadow"),
+                                ("Shadow from Window", "window_shadow"),
+                                ("Soft Studio Lighting", "soft_studio"),
+                                ("Home Atmosphere, Cozy Bedroom", "cozy_bedroom"),
+                                ("Neon, Wong Kar-wai, Warm", "wong_kar_wai"),
+                                ("Moonlight through Curtains", "moonlight_curtains"),
+                                ("Stormy Sky Lighting", "stormy_sky"),
+                                ("Underwater Glow, Deep Sea", "underwater_glow"),
+                                ("Foggy Forest at Dawn", "foggy_forest"),
+                                ("Golden Hour in a Meadow", "meadow_golden"),
+                                ("Rainbow Reflections, Neon", "rainbow_neon"),
+                                ("Apocalyptic, Smoky Atmosphere", "apocalyptic"),
+                                ("Red Glow, Emergency Lights", "emergency_red"),
+                                ("Mystical Glow, Enchanted Forest", "mystical_forest"),
+                                ("Campfire Light", "campfire"),
+                                ("Harsh, Industrial Lighting", "industrial_harsh"),
+                                ("Sunrise in the Mountains", "mountain_sunrise"),
+                                ("Evening Glow in the Desert", "desert_evening"),
+                                ("Moonlight in a Dark Alley", "dark_alley"),
+                                ("Golden Glow at a Fairground", "fairground"),
+                                ("Midnight in the Forest", "forest_midnight"),
+                                ("Purple and Pink Hues at Twilight", "twilight_purple"),
+                                ("Foggy Morning, Muted Light", "foggy_morning"),
+                                ("Candle-lit Room, Rustic Vibe", "rustic_candle"),
+                                ("Fluorescent Office Lighting", "office_fluorescent"),
+                                ("Lightning Flash in Storm", "storm_lightning"),
+                                ("Night, Cozy Warm Light from Fireplace", "fireplace_night"),
+                                ("Ethereal Glow, Magical Forest", "ethereal_magic"),
+                                ("Dusky Evening on a Beach", "beach_dusky"),
+                                ("Afternoon Light Filtering through Trees", "trees_afternoon"),
+                                ("Blue Neon Light, Urban Street", "urban_blue_neon"),
+                                ("Red and Blue Police Lights in Rain", "rain_police"),
+                                ("Aurora Borealis Glow, Arctic Landscape", "aurora_arctic"),
+                                ("Sunrise through Foggy Mountains", "foggy_mountains"),
+                                ("Golden Hour on a City Skyline", "city_skyline"),
+                                ("Mysterious Twilight, Heavy Mist", "twilight_mist"),
+                                ("Early Morning Rays, Forest Clearing", "forest_rays"),
+                                ("Colorful Lantern Light at Festival", "festival_lantern"),
+                                ("Soft Glow through Stained Glass", "stained_glass"),
+                                ("Harsh Spotlight in Dark Room", "dark_spotlight"),
+                                ("Mellow Evening Glow on a Lake", "lake_evening"),
+                                ("Crystal Reflections in a Cave", "cave_crystal"),
+                                ("Vibrant Autumn Lighting in a Forest", "autumn_forest"),
+                                ("Gentle Snowfall at Dusk", "snowfall_dusk"),
+                                ("Hazy Light of a Winter Morning", "winter_hazy"),
+                                ("Rain-soaked Reflections in City Lights", "rain_city"),
+                                ("Golden Sunlight Streaming through Trees", "trees_golden_sun"),
+                                ("Fireflies Lighting up a Summer Night", "fireflies_summer"),
+                                ("Glowing Embers from a Forge", "forge_embers"),
+                                ("Dim Candlelight in a Gothic Castle", "gothic_castle"),
+                                ("Midnight Sky with Bright Starlight", "starlight_midnight"),
+                                ("Warm Sunset in a Rural Village", "rural_sunset"),
+                                ("Flickering Light in a Haunted House", "haunted_flicker"),
+                                ("Desert Sunset with Mirage-like Glow", "desert_mirage"),
+                                ("Golden Beams Piercing through Storm Clouds", "storm_beams"),
+                                ("Custom", "custom"),
+                            ],
+                            value="none",
+                            elem_classes="radio-group"
+                        )
+                        illumination_env_custom = gr.Textbox(
+                            label="Custom Illumination Environment",
+                            placeholder="e.g., Inside a crystal palace, Underwater volcano, etc.",
+                            visible=False
+                        )
+                        
+                with gr.Tab("Custom Prompt"):
+                    with gr.Accordion("✍️ Custom Prompt (in any language)", open=False):
+                        prompt = gr.Textbox(
+                            placeholder="Example: Add warm sunset lighting from the right",
+                            lines=3
+                        )
                 
                 with gr.Row():
                     reset_btn = gr.Button("🔄 Reset")
@@ -532,22 +583,31 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
 
             with gr.Column(scale=1):
                 result = gr.Image(label="Output Image", interactive=False)
-                prompt_preview = gr.Textbox(label="Processed Prompt", interactive=False)
+                prompt_preview = gr.Textbox(label="Processed Prompt (in Chinese)", interactive=False)
                 create_video_button = gr.Button("🎥 Create Video Between Images", variant="secondary", visible=False)
                 with gr.Group(visible=False) as video_group:
                     video_output = gr.Video(label="Generated Video", show_download_button=True, autoplay=True)
                     
     inputs = [
-        image, light_type, light_direction, light_intensity, illumination_env, prompt,
-        seed, randomize_seed, true_guidance_scale, num_inference_steps, height, width, prev_output
+        image, light_type, light_type_custom, light_direction, light_direction_custom, 
+        light_intensity, light_intensity_custom, illumination_env, illumination_env_custom, 
+        prompt, seed, randomize_seed, true_guidance_scale, num_inference_steps, height, width, prev_output
     ]
     outputs = [result, seed, prompt_preview]
 
+    # Toggle custom textboxes visibility
+    light_type.change(fn=toggle_custom_textbox, inputs=[light_type], outputs=[light_type_custom], queue=False)
+    light_direction.change(fn=toggle_custom_textbox, inputs=[light_direction], outputs=[light_direction_custom], queue=False)
+    light_intensity.change(fn=toggle_custom_textbox, inputs=[light_intensity], outputs=[light_intensity_custom], queue=False)
+    illumination_env.change(fn=toggle_custom_textbox, inputs=[illumination_env], outputs=[illumination_env_custom], queue=False)
+
     # Reset behavior
     reset_btn.click(
         fn=reset_all,
         inputs=None,
-        outputs=[light_type, light_direction, light_intensity, illumination_env, prompt, is_reset],
+        outputs=[light_type, light_type_custom, light_direction, light_direction_custom, 
+                light_intensity, light_intensity_custom, illumination_env, illumination_env_custom, 
+                prompt, is_reset],
         queue=False
     ).then(fn=end_reset, inputs=None, outputs=[is_reset], queue=False)
 
@@ -579,12 +639,13 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
     # Examples
     gr.Examples(
         examples=[
-            ["harold.png", "dramatic", "side", "soft", "none", "", 0, True, 1.0, 4, 672, 1024],
-            ["distracted.png", "golden_hour", "side", "strong", "none", "", 0, True, 1.0, 4, 640, 1024],
-            ["disaster.jpg", "moonlight", "front", "medium", "neon_city", "", 0, True, 1.0, 4, 640, 1024],
+            ["harold.png", "dramatic", "", "side", "", "soft", "", "none", "", "", 0, True, 1.0, 4, 672, 1024],
+            ["distracted.png", "golden_hour", "", "side", "", "strong", "", "none", "", "", 0, True, 1.0, 4, 640, 1024],
+            ["disaster.jpg", "moonlight", "", "front", "", "medium", "", "neon_city", "", "", 0, True, 1.0, 4, 640, 1024],
         ],
-        inputs=[image, light_type, light_direction, light_intensity, illumination_env, prompt,
-                seed, randomize_seed, true_guidance_scale, num_inference_steps, height, width],
+        inputs=[image, light_type, light_type_custom, light_direction, light_direction_custom, 
+                light_intensity, light_intensity_custom, illumination_env, illumination_env_custom, 
+                prompt, seed, randomize_seed, true_guidance_scale, num_inference_steps, height, width],
         outputs=outputs,
         fn=infer_relight,
         cache_examples="lazy",
@@ -599,7 +660,9 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
     ).then(
         fn=reset_all,
         inputs=None,
-        outputs=[light_type, light_direction, light_intensity, illumination_env, prompt, is_reset],
+        outputs=[light_type, light_type_custom, light_direction, light_direction_custom, 
+                light_intensity, light_intensity_custom, illumination_env, illumination_env_custom, 
+                prompt, is_reset],
         queue=False
     ).then(
         fn=end_reset, 
@@ -609,7 +672,7 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
     )
 
 
-    # Live updates
+    # Live updates - only trigger on non-custom radio selections
     def maybe_infer(is_reset, progress=gr.Progress(track_tqdm=True), *args):
         if is_reset:
             return gr.update(), gr.update(), gr.update(), gr.update()
@@ -620,13 +683,22 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
             return result_img, result_seed, result_prompt, gr.update(visible=show_button)
 
     control_inputs = [
-        image, light_type, light_direction, light_intensity, illumination_env, prompt,
-        seed, randomize_seed, true_guidance_scale, num_inference_steps, height, width, prev_output
+        image, light_type, light_type_custom, light_direction, light_direction_custom, 
+        light_intensity, light_intensity_custom, illumination_env, illumination_env_custom, 
+        prompt, seed, randomize_seed, true_guidance_scale, num_inference_steps, height, width, prev_output
     ]
     control_inputs_with_flag = [is_reset] + control_inputs
 
+    # Only trigger live updates when selecting non-custom options
+    def should_trigger_infer(choice):
+        return choice != "custom"
+    
     for control in [light_type, light_direction, light_intensity, illumination_env]:
-        control.input(fn=maybe_infer, inputs=control_inputs_with_flag, outputs=outputs + [create_video_button])
+        control.input(
+            fn=lambda choice, *args: maybe_infer(*args) if should_trigger_infer(choice) else (gr.update(), gr.update(), gr.update(), gr.update()),
+            inputs=[control] + control_inputs_with_flag,
+            outputs=outputs + [create_video_button]
+        )
     
     run_event.then(lambda img, *_: img, inputs=[result], outputs=[prev_output])