Update app.py

app.py

@@ -1,22 +1,61 @@
 import gradio as gr
 from inference.wan_infer import generate_video
 
-def run(prompt):
-    return generate_video(prompt)
+def run(prompt, resolution, aspect, fps, seconds):
+    return generate_video(
+        prompt=prompt,
+        resolution=resolution,
+        aspect=aspect,
+        fps=fps,
+        seconds=seconds,
+    )
 
 with gr.Blocks() as demo:
-    gr.Markdown("## Wan2.2 TI2V-5B (HF Spaces Compatible)")
+    gr.Markdown("## Wan2.2 TI2V-5B · L40S Optimized")
 
     prompt = gr.Textbox(
         label="Prompt",
-        value="A cinematic shot of a futuristic city at sunset"
+        value="A cinematic shot of a futuristic city at sunset, dramatic lighting"
     )
 
+    with gr.Row():
+        resolution = gr.Radio(
+            ["480p", "720p"],
+            value="720p",
+            label="Resolution"
+        )
+
+        aspect = gr.Radio(
+            ["16:9", "9:16"],
+            value="16:9",
+            label="Aspect Ratio"
+        )
+
+    with gr.Row():
+        fps = gr.Slider(
+            minimum=12,
+            maximum=30,
+            step=1,
+            value=24,
+            label="FPS"
+        )
+
+        seconds = gr.Slider(
+            minimum=2,
+            maximum=10,
+            step=1,
+            value=3,
+            label="Duration (seconds)"
+        )
+
     output = gr.Video(label="Generated Video")
 
     btn = gr.Button("Generate")
-    btn.click(run, prompt, output)
+    btn.click(
+        run,
+        inputs=[prompt, resolution, aspect, fps, seconds],
+        outputs=output
+    )
 
-# ✅ Gradio 6.x compatible
 demo.queue(default_concurrency_limit=1)
 demo.launch()