Update app.py

app.py

@@ -5,7 +5,8 @@ import tempfile
 import subprocess
 import os
 
-# Extraire une frame pour prévisualisation
+selection_box = [0, 0, 0, 0]
+
 def extract_preview(video):
     temp_dir = tempfile.mkdtemp()
     frame_path = os.path.join(temp_dir, "preview.png")
@@ -21,33 +22,37 @@ def extract_preview(video):
     return frame_path
 
 
-# Traitement vidéo avec le rectangle sélectionné
-def process_video(video, selection):
-    x1, y1, x2, y2 = selection
+def on_select(evt: gr.SelectData):
+    global selection_box
+    (x1, y1), (x2, y2) = evt.index
+    selection_box = [x1, y1, x2, y2]
+    return f"Zone sélectionnée : {selection_box}"
+
+
+def process_video(video):
+    x1, y1, x2, y2 = selection_box
+
     temp_dir = tempfile.mkdtemp()
     frames_dir = os.path.join(temp_dir, "frames")
     os.makedirs(frames_dir, exist_ok=True)
 
-    # Extraire les frames (FPS réduit)
     subprocess.run([
         "ffmpeg", "-i", video,
         "-vf", "fps=10,scale=1280:-1",
         f"{frames_dir}/frame_%04d.png"
     ], check=True)
 
-    # Inpainting
     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[int(y1):int(y2), int(x1):int(x2)] = 255
+        mask[y1:y2, x1:x2] = 255
         mask = cv2.dilate(mask, np.ones((3,3), np.uint8), 1)
 
         result = cv2.inpaint(img, mask, 3, cv2.INPAINT_TELEA)
         cv2.imwrite(path, result)
 
-    # Réassembler la vidéo
     output = os.path.join(temp_dir, "output.mp4")
     subprocess.run([
         "ffmpeg", "-y",
@@ -62,26 +67,20 @@ def process_video(video, selection):
 
 
 with gr.Blocks() as demo:
-    gr.Markdown("## 🎥 Supprimer un texte sur une vidéo")
+    gr.Markdown("## 🎥 Suppression de texte sur vidéo")
 
     video = gr.Video(label="Upload vidéo (≤ 60s)")
-    preview = gr.Image(label="Sélectionne la zone du texte", tool="select")
+    preview = gr.Image(label="Clique-glisse pour sélectionner la zone", interactive=True)
+
+    status = gr.Textbox(label="Statut", interactive=False)
 
-    btn_preview = gr.Button("Afficher une image pour sélectionner")
+    btn_preview = gr.Button("Afficher une image")
     btn_process = gr.Button("Supprimer le texte")
 
     output = gr.Video(label="Vidéo finale")
 
-    btn_preview.click(
-        extract_preview,
-        inputs=video,
-        outputs=preview
-    )
-
-    btn_process.click(
-        process_video,
-        inputs=[video, preview.select],
-        outputs=output
-    )
+    btn_preview.click(extract_preview, video, preview)
+    preview.select(on_select, None, status)
+    btn_process.click(process_video, video, output)
 
 demo.launch()