Update app.py

app.py

@@ -5,8 +5,6 @@ import tempfile
 import subprocess
 import os
 
-selection_box = [0, 0, 0, 0]
-
 def extract_preview(video):
     temp_dir = tempfile.mkdtemp()
     frame_path = os.path.join(temp_dir, "preview.png")
@@ -22,15 +20,16 @@ def extract_preview(video):
     return frame_path
 
 
-def on_select(evt: gr.SelectData):
-    global selection_box
+def save_selection(evt: gr.SelectData):
     (x1, y1), (x2, y2) = evt.index
-    selection_box = [x1, y1, x2, y2]
-    return f"Zone sélectionnée : {selection_box}"
+    return [int(x1), int(y1), int(x2), int(y2)]
+
 
+def process_video(video, box):
+    if box is None:
+        raise gr.Error("Aucune zone sélectionnée")
 
-def process_video(video):
-    x1, y1, x2, y2 = selection_box
+    x1, y1, x2, y2 = box
 
     temp_dir = tempfile.mkdtemp()
     frames_dir = os.path.join(temp_dir, "frames")
@@ -70,9 +69,9 @@ with gr.Blocks() as demo:
     gr.Markdown("## 🎥 Suppression de texte sur vidéo")
 
     video = gr.Video(label="Upload vidéo (≤ 60s)")
-    preview = gr.Image(label="Clique-glisse pour sélectionner la zone", interactive=True)
+    preview = gr.Image(label="Sélectionne la zone à supprimer", interactive=True)
 
-    status = gr.Textbox(label="Statut", interactive=False)
+    selection_state = gr.State(None)
 
     btn_preview = gr.Button("Afficher une image")
     btn_process = gr.Button("Supprimer le texte")
@@ -80,7 +79,7 @@ with gr.Blocks() as demo:
     output = gr.Video(label="Vidéo finale")
 
     btn_preview.click(extract_preview, video, preview)
-    preview.select(on_select, None, status)
-    btn_process.click(process_video, video, output)
+    preview.select(save_selection, None, selection_state)
+    btn_process.click(process_video, [video, selection_state], output)
 
 demo.launch()