Update app.py

app.py

@@ -119,28 +119,27 @@ def run_inference(prompt, image, pose_type, speed, use_flow_integration, cam_pos
 # =========================================
 # 3. Define FloVD Gradio Interface
 # =========================================
+with gr.Blocks() as video_tab:
+    gr.Markdown("## 🎥 FloVD: Optical Flow + CogVideoX Video Generation")
+
+    prompt = gr.Textbox(label="Prompt", value="A girl riding a bicycle through a park.")
+    image = gr.Image(label="Input Image")
+    pose_type = gr.Radio(choices=["manual", "re10k"], value="manual", label="Camera Pose Type")
+    speed = gr.Slider(minimum=0.1, maximum=2.0, step=0.1, value=0.5, label="Camera Speed")
+    use_flow_integration = gr.Checkbox(label="Use Flow Integration", value=False)
+    cam_pose_name = gr.Textbox(label="Camera Trajectory", placeholder="e.g., zoom_in, custom_motion, etc.", lines=1)
+
+    generate_btn = gr.Button("🎬 Generate Video")
+
+    video_output = gr.Video(label="Generated Video")
+    log_output = gr.Textbox(label="Logs", lines=20, interactive=False)
+
+    generate_btn.click(
+        fn=run_inference,
+        inputs=[prompt, image, pose_type, speed, use_flow_integration, cam_pose_name],
+        outputs=[video_output, log_output]
+    )
 
-video_interface = gr.Interface(
-    fn=run_inference,
-    inputs=[
-        gr.Textbox(label="Prompt", value="A girl riding a bicycle through a park."),
-        gr.Image(label="Input Image"),
-        gr.Radio(choices=["manual", "re10k"], value="manual", label="Camera Pose Type"),
-        gr.Slider(minimum=0.1, maximum=2.0, step=0.1, value=0.5, label="Camera Speed"),
-        gr.Checkbox(label="Use Flow Integration", value=False),
-        gr.Textbox(
-            label="Camera Trajectory",
-            placeholder="e.g., zoom_in, custom_motion, etc.",
-            lines=1
-        )
-    ],
-    outputs=[
-        gr.Video(label="Generated Video"),
-        gr.Textbox(label="Logs", lines=20, interactive=False),
-    ],
-    title="🎥 FloVD: Optical Flow + CogVideoX Video Generation",
-    description="Upload an image and prompt to generate motion-controlled video using FloVD and CogVideoX."
-)
 # =========================================
 # 4. Live System Monitor (Fixed)
 # =========================================