Update app.py

app.py

@@ -39,8 +39,6 @@ def compress_video(video_file, target_width, target_height, progress=gr.Progress
         raise gr.Error("Could not open video file for compression.")
     
     fps = cap.get(cv2.CAP_PROP_FPS)
-    
-    # Use the target resolution directly.
     new_width = int(target_width)
     new_height = int(target_height)
     
@@ -187,11 +185,14 @@ def read_motion_csv(csv_filename):
     print("[INFO] Completed reading motion CSV.")
     return motion_data
 
-def stabilize_video_using_csv(video_file, csv_file, zoom=1.0, vertical_only=False, progress=gr.Progress(), progress_offset=0.6, progress_scale=0.4, output_file=None):
+def stabilize_video_using_csv(video_file, csv_file, zoom=1.0, vertical_only=False, 
+                              progress=gr.Progress(), progress_offset=0.6, progress_scale=0.4, 
+                              output_file=None):
     """
     Stabilizes the video using motion data from the CSV.
     If vertical_only is True, only vertical motion is corrected.
     Updates progress from progress_offset to progress_offset+progress_scale.
+    Uses cv2.BORDER_REPLICATE to fill any gaps, preventing black borders.
     """
     start_time = time.time()
     print(f"[INFO] Starting stabilization using CSV: {csv_file}")
@@ -235,7 +236,8 @@ def stabilize_video_using_csv(video_file, csv_file, zoom=1.0, vertical_only=Fals
             dx = 0  # Only vertical stabilization.
         transform = np.array([[1, 0, dx],
                               [0, 1, dy]], dtype=np.float32)
-        stabilized_frame = cv2.warpAffine(frame, transform, (width, height))
+        # Use BORDER_REPLICATE to avoid black borders
+        stabilized_frame = cv2.warpAffine(frame, transform, (width, height), borderMode=cv2.BORDER_REPLICATE)
         
         out.write(stabilized_frame)
         if frame_idx % 10 == 0 or frame_idx == total_frames: