Update app.py

app.py

@@ -144,13 +144,8 @@ css = '''#col-container { max-width: 800px; margin: 0 auto; }
 .dark .progress-text{color: white !important}
 #examples{max-width: 800px; margin: 0 auto; }'''
 
-is_reset = gr.State(value=False)
-
 def reset_all():
-    return [0, 0, 0, 0, False, True]
-
-def end_reset():
-    return False
+    return [0, 0, 0, 0]
 
 def update_dimensions_on_upload(image):
     if image is None:
@@ -193,7 +188,7 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
                     rotate_deg = gr.Slider(label="Rotate Left–Right (degrees °)", minimum=-90, maximum=90, step=45, value=0)
                     move_forward = gr.Slider(label="Move Forward → Close-Up", minimum=0, maximum=10, step=5, value=0)
                     vertical_tilt = gr.Slider(label="Vertical Angle (Bird ↔ Worm)", minimum=-1, maximum=1, step=1, value=0)
-                    wideangle = gr.Checkbox(label="Wide-Angle Lens", value=False)
+                    wideangle = gr.Slider(label="Wide-Angle Lens", minimum=0, maximum=1, step=1, value=0)
                 with gr.Row():
                         reset_btn = gr.Button("Reset")
                         run_btn = gr.Button("Generate", variant="primary")
@@ -225,9 +220,9 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
     reset_btn.click(
         fn=reset_all,
         inputs=None,
-        outputs=[rotate_deg, move_forward, vertical_tilt, wideangle, is_reset],
+        outputs=[rotate_deg, move_forward, vertical_tilt, wideangle],
         queue=False
-    ).then(fn=end_reset, inputs=None, outputs=[is_reset], queue=False)
+    )
 
     # Manual generation
     run_event = run_btn.click(fn=infer_camera_edit, inputs=inputs, outputs=outputs)
@@ -258,34 +253,21 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
     ).then(
         fn=reset_all,
         inputs=None,
-        outputs=[rotate_deg, move_forward, vertical_tilt, wideangle, is_reset],
-        queue=False
-    ).then(
-        fn=end_reset, 
-        inputs=None, 
-        outputs=[is_reset], 
+        outputs=[rotate_deg, move_forward, vertical_tilt, wideangle],
         queue=False
     )
 
 
     # Live updates
-    def maybe_infer(is_reset, progress=gr.Progress(track_tqdm=True), *args):
-        if is_reset:
-            return gr.update(), gr.update(), gr.update()
-        else:
-            return infer_camera_edit(*args)
-
     control_inputs = [
         image, rotate_deg, move_forward,
         vertical_tilt, wideangle,
         seed, randomize_seed, true_guidance_scale, num_inference_steps, height, width, prev_output
     ]
-    control_inputs_with_flag = [is_reset] + control_inputs
 
-    for control in [rotate_deg, move_forward, vertical_tilt]:
-        control.release(fn=maybe_infer, inputs=control_inputs_with_flag, outputs=outputs)
-    
-    wideangle.change(fn=maybe_infer, inputs=control_inputs_with_flag, outputs=outputs)
+    # All sliders use .release() to avoid triggering during reset
+    for control in [rotate_deg, move_forward, vertical_tilt, wideangle]:
+        control.release(fn=infer_camera_edit, inputs=control_inputs, outputs=outputs)
     
     run_event.then(lambda img, *_: img, inputs=[result], outputs=[prev_output])