Update app.py

app.py

@@ -5,17 +5,27 @@ import tempfile
 import subprocess
 import os
 
+# --------- UTILITAIRES ---------
+
 def extract_preview(video):
+    if video is None:
+        raise gr.Error("Aucune vidéo fournie")
+
+    video_path = video["name"]
+
     temp_dir = tempfile.mkdtemp()
     frame_path = os.path.join(temp_dir, "preview.png")
 
-    subprocess.run([
-        "ffmpeg", "-y",
-        "-i", video,
-        "-vf", "select=eq(n\\,10)",
-        "-vframes", "1",
-        frame_path
-    ], check=True)
+    subprocess.run(
+        [
+            "ffmpeg", "-y",
+            "-i", video_path,
+            "-vf", "select=eq(n\\,10)",
+            "-vframes", "1",
+            frame_path
+        ],
+        check=True
+    )
 
     return frame_path
 
@@ -26,49 +36,64 @@ def save_selection(evt: gr.SelectData):
 
 
 def process_video(video, box):
+    if video is None:
+        raise gr.Error("Aucune vidéo")
     if box is None:
         raise gr.Error("Aucune zone sélectionnée")
 
+    video_path = video["name"]
     x1, y1, x2, y2 = box
 
     temp_dir = tempfile.mkdtemp()
     frames_dir = os.path.join(temp_dir, "frames")
     os.makedirs(frames_dir, exist_ok=True)
 
-    subprocess.run([
-        "ffmpeg", "-i", video,
-        "-vf", "fps=10,scale=1280:-1",
-        f"{frames_dir}/frame_%04d.png"
-    ], check=True)
-
+    # Extraire les frames
+    subprocess.run(
+        [
+            "ffmpeg",
+            "-i", video_path,
+            "-vf", "fps=10,scale=1280:-1",
+            f"{frames_dir}/frame_%04d.png"
+        ],
+        check=True
+    )
+
+    # Inpainting sur chaque frame
     for f in sorted(os.listdir(frames_dir)):
         path = os.path.join(frames_dir, f)
         img = cv2.imread(path)
 
         mask = np.zeros(img.shape[:2], dtype=np.uint8)
         mask[y1:y2, x1:x2] = 255
-        mask = cv2.dilate(mask, np.ones((3,3), np.uint8), 1)
+        mask = cv2.dilate(mask, np.ones((3, 3), np.uint8), 1)
 
         result = cv2.inpaint(img, mask, 3, cv2.INPAINT_TELEA)
         cv2.imwrite(path, result)
 
-    output = os.path.join(temp_dir, "output.mp4")
-    subprocess.run([
-        "ffmpeg", "-y",
-        "-framerate", "10",
-        "-i", f"{frames_dir}/frame_%04d.png",
-        "-c:v", "libx264",
-        "-pix_fmt", "yuv420p",
-        output
-    ], check=True)
+    # Réassembler la vidéo
+    output_path = os.path.join(temp_dir, "output.mp4")
+    subprocess.run(
+        [
+            "ffmpeg", "-y",
+            "-framerate", "10",
+            "-i", f"{frames_dir}/frame_%04d.png",
+            "-c:v", "libx264",
+            "-pix_fmt", "yuv420p",
+            output_path
+        ],
+        check=True
+    )
+
+    return output_path
 
-    return output
 
+# --------- INTERFACE ---------
 
 with gr.Blocks() as demo:
-    gr.Markdown("## 🎥 Suppression de texte sur vidéo")
+    gr.Markdown("## 🎥 Supprimer un texte ou sous-titre d’une vidéo")
 
-    video = gr.Video(label="Upload vidéo (≤ 60s)")
+    video = gr.Video(label="Uploader une vidéo (≤ 60s)")
     preview = gr.Image(label="Sélectionne la zone à supprimer", interactive=True)
 
     selection_state = gr.State(None)