Update app.py

app.py

@@ -173,6 +173,7 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
     with gr.Column(elem_classes=["main-container"]):
         gr.Markdown("# ✨ Fast 4 steps Wan 2.1 I2V (14B) with CausVid LoRA")
 
+        # Badges
         gr.HTML("""
         <div class="badge-container">
             <a href="https://huggingface.co/spaces/Heartsync/WAN2-1-fast-T2V-FusioniX" target="_blank">
@@ -186,8 +187,9 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
             </a>
         </div>
         """)
-        
+
         with gr.Row():
+            # Input Column
             with gr.Column(elem_classes=["input-container"]):
                 input_image_component = gr.Image(
                     type="pil", 
@@ -207,7 +209,7 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
                     label="⏱️ Duration (seconds)", 
                     info=f"Clamped to model's {MIN_FRAMES_MODEL}-{MAX_FRAMES_MODEL} frames at {FIXED_FPS}fps."
                 )
-                
+
                 with gr.Accordion("⚙️ Advanced Settings", open=False):
                     negative_prompt_input = gr.Textbox(
                         label="❌ Negative Prompt", 
@@ -219,13 +221,11 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
                         minimum=0, 
                         maximum=MAX_SEED, 
                         step=1, 
-                        value=42, 
-                        interactive=True
+                        value=42
                     )
                     randomize_seed_checkbox = gr.Checkbox(
                         label="🔀 Randomize seed", 
-                        value=True, 
-                        interactive=True
+                        value=True
                     )
                     with gr.Row():
                         height_input = gr.Slider(
@@ -248,7 +248,7 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
                         step=1, 
                         value=4, 
                         label="🚀 Inference Steps"
-                    ) 
+                    )
                     guidance_scale_input = gr.Slider(
                         minimum=0.0, 
                         maximum=20.0, 
@@ -258,67 +258,66 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
                         visible=False
                     )
 
-                # --- Optional Local LoRA Controls ---
-with gr.Accordion("🎛️ Optional Local LoRA", open=False):
-    optional_lora_dropdown = gr.Dropdown(
-        label="Select Local LoRA",
-        choices=["None", "GEN", "INS"],
-        value="None"
-    )
-    lora_strength_slider = gr.Slider(
-        label="Local LoRA Strength",
-        minimum=0.0,
-        maximum=2.0,
-        value=1.0,
-        step=0.05
-    )
-
-# Insert before button
-generate_button = gr.Button(
+                    # LoRA controls
+                    optional_lora_dropdown = gr.Dropdown(
+                        choices=["None", "GEN", "INS"],
+                        value="None",
+                        label="🧩 Optional LoRA"
+                    )
+                    lora_strength_slider = gr.Slider(
+                        minimum=0.0,
+                        maximum=1.0,
+                        step=0.05,
+                        value=1.0,
+                        label="⚡ LoRA Strength"
+                    )
+
+                # Generate button
+                generate_button = gr.Button(
                     "🎬 Generate Video", 
                     variant="primary",
                     elem_classes=["generate-btn"]
                 )
-                
-        with gr.Column(elem_classes=["output-container"]):
-            video_output = gr.Video(
-                label="🎥 Generated Video", 
-                autoplay=True, 
-                interactive=False
-        )
 
+            # Output Column
+            with gr.Column(elem_classes=["output-container"]):
+                video_output = gr.Video(
+                    label="🎥 Generated Video", 
+                    autoplay=True, 
+                    interactive=False
+                )
+
+        # Image upload handlers
         input_image_component.upload(
             fn=handle_image_upload_for_dims_wan,
             inputs=[input_image_component, height_input, width_input],
             outputs=[height_input, width_input]
         )
-        
-        input_image_component.clear( 
+        input_image_component.clear(
             fn=handle_image_upload_for_dims_wan,
             inputs=[input_image_component, height_input, width_input],
             outputs=[height_input, width_input]
         )
-        
+
+        # Inputs list for generation
         ui_inputs = [
             input_image_component, prompt_input, height_input, width_input,
             negative_prompt_input, duration_seconds_input,
             guidance_scale_input, steps_slider, seed_input, randomize_seed_checkbox,
             optional_lora_dropdown, lora_strength_slider
-            input_image_component, prompt_input, height_input, width_input,
-            negative_prompt_input, duration_seconds_input,
-            guidance_scale_input, steps_slider, seed_input, randomize_seed_checkbox
         ]
         generate_button.click(fn=generate_video, inputs=ui_inputs, outputs=[video_output, seed_input])
 
+        # Example gallery
         with gr.Column():
             gr.Examples(
-                examples=[ 
+                examples=[
                     ["peng.png", "a penguin playfully dancing in the snow, Antarctica", 896, 896],
                     ["forg.jpg", "the frog jumps around", 832, 832],
                 ],
-                inputs=[input_image_component, prompt_input, height_input, width_input], 
-                outputs=[video_output, seed_input], 
-                fn=generate_video, 
+                inputs=[input_image_component, prompt_input, height_input, width_input],
+                outputs=[video_output, seed_input],
+                fn=generate_video,
                 cache_examples="lazy",
                 label="🌟 Example Gallery"
             )