Update app.py

app.py

@@ -257,4 +257,54 @@ def process_video_ai(
         gr.Info("Motion estimated successfully.")
 
         if auto_zoom:
-            z = compute_auto_zoom(offsets, out
+            z = compute_auto_zoom(offsets, out_w, out_h)
+            gr.Info(f"Auto zoom factor computed: {z:.2f}")
+            zoom = z
+
+        stabilized_path = stabilize_stream(
+            video_file,
+            offsets,
+            zoom=float(zoom),
+            vertical_only=bool(vertical_only),
+            out_w=out_w,
+            out_h=out_h,
+            progress=progress,
+            progress_offset=0.55,
+            progress_scale=0.45,
+        )
+        gr.Info("Video stabilization complete.")
+        print("[INFO] Video processing complete.")
+
+    logs = log_buffer.getvalue()
+    return video_file, stabilized_path, logs
+
+with gr.Blocks() as demo:
+    gr.Markdown("# AI-Powered Video Stabilization")
+    gr.Markdown(
+        "Upload a video, select a zoom factor (or use Auto Zoom Mode), choose whether to apply only vertical stabilization, and optionally compress the output resolution. "
+        "The system estimates motion using RAFT if available (otherwise Farneback) and stabilizes the video with progress updates."
+    )
+
+    with gr.Row():
+        with gr.Column():
+            video_input = gr.Video(label="Input Video")
+            zoom_slider = gr.Slider(minimum=1.0, maximum=3.0, step=0.1, value=1.0, label="Zoom Factor (ignored if Auto Zoom enabled)")
+            auto_zoom_checkbox = gr.Checkbox(label="Auto Zoom Mode", value=False)
+            vertical_checkbox = gr.Checkbox(label="Vertical Stabilization Only", value=False)
+            compress_checkbox = gr.Checkbox(label="Compress Output Resolution", value=False)
+            target_width = gr.Number(label="Target Width (px)", value=640)
+            target_height = gr.Number(label="Target Height (px)", value=360)
+            process_button = gr.Button("Process Video")
+        with gr.Column():
+            original_video = gr.Video(label="Original Video")
+            stabilized_video = gr.Video(label="Stabilized Video")
+            logs_output = gr.Textbox(label="Logs", lines=10)
+
+    process_button.click(
+        fn=process_video_ai,
+        inputs=[video_input, zoom_slider, vertical_checkbox, compress_checkbox, target_width, target_height, auto_zoom_checkbox],
+        outputs=[original_video, stabilized_video, logs_output],
+    )
+
+if __name__ == "__main__":
+    demo.launch()