Update app.py

app.py

@@ -33,18 +33,22 @@ def is_ffmpeg_available():
     except (subprocess.CalledProcessError, FileNotFoundError):
         return False
 
-def remove_watermark_from_frame(frame, mask_coords=None):
-    mask = np.zeros(frame.shape[:2], np.uint8)
-    if mask_coords:
-        x1, y1 = mask_coords[0]
-        x2, y2 = mask_coords[1]
-        x1, x2 = sorted([x1, x2])
-        y1, y2 = sorted([y1, y2])
-        mask[y1:y2, x1:x2] = 255
+
+def remove_watermark_from_frame(frame, mask_coords):
+    if mask_coords is None:
+        return frame
+
+    # בדיקה אם נשלח tuple רגיל (x1, y1, x2, y2)
+    if isinstance(mask_coords[0], int):
+        x1, y1, x2, y2 = mask_coords
     else:
-        h, w = frame.shape[:2]
-        mask[0:50, 0:w] = 255
-    return cv2.inpaint(frame, mask, 10, cv2.INPAINT_NS)
+        (x1, y1), (x2, y2) = mask_coords
+
+    mask = np.zeros(frame.shape[:2], dtype=np.uint8)
+    mask[y1:y2, x1:x2] = 255
+    frame = cv2.inpaint(frame, mask, 3, cv2.INPAINT_TELEA)
+    return frame
+
 
 def extract_first_frame(video_file_path):
     cap = cv2.VideoCapture(video_file_path)
@@ -61,6 +65,7 @@ def frames_to_video(frames, output_path, fps, frame_size):
         out.write(frame)
     out.release()
     return True
+    
 
 def remove_watermark_process(video_path, coords_input=None):
     if not is_ffmpeg_available():