Update app.py

app.py

@@ -110,25 +110,31 @@ def run_inference(prompt, image, pose_type, speed, use_flow_integration, cam_pos
 # 3. Gradio App Interface
 # =========================================
 
-with gr.Blocks() as demo:
-    gr.Markdown("## 🎥 FloVD: Optical Flow + CogVideoX Video Generation")
-    with gr.Row():
-        with gr.Column():
-            prompt = gr.Textbox(label="Prompt", value="A girl riding a bicycle through a park.")
-            image = gr.Image(type="pil", label="Input Image")
-            pose_type = gr.Radio(choices=["manual", "re10k"], value="manual", label="Camera Pose Type")
-            cam_pose_name = gr.Textbox(label="Camera Trajectory Name", placeholder="e.g. zoom_in, tilt_up")
-            speed = gr.Slider(minimum=0.1, maximum=2.0, step=0.1, value=0.5, label="Speed")
-            use_flow_integration = gr.Checkbox(label="Use Flow Integration", value=False)
-            submit = gr.Button("Generate Video")
-        with gr.Column():
-            output_video = gr.Video(label="Generated Video")
-            output_logs = gr.Textbox(label="Logs", lines=20, interactive=False)
-
-    submit.click(
-        fn=run_inference,
-        inputs=[prompt, image, pose_type, speed, use_flow_integration, cam_pose_name],
-        outputs=[output_video, output_logs]
-    )
-
-demo.launch(show_error=True)
+demo = gr.Interface(
+    fn=run_inference,
+    inputs=[
+        gr.Textbox(label="Prompt", value="A girl riding a bicycle through a park."),
+        gr.Image(type="pil", 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.Dropdown(
+            choices=["zoom_in", "zoom_out", "tilt_up", "tilt_down", "circle", ""],
+            label="Camera Trajectory",
+            value="zoom_in",
+            allow_custom_value=True
+        )
+    ],
+    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."
+)
+
+# -----------------------------
+# Launch the App
+# -----------------------------
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=7860)