Update app.py

app.py

@@ -1,37 +1,67 @@
-import os
-import subprocess
+import gradio as gr
 import cv2
 import numpy as np
-from flask import Flask, request, send_file, jsonify
-from flask_cors import CORS
-
-app = Flask(__name__)
-CORS(app)
-
-def inpaint_video(input_path, output_path, x, y, w, h):
-    # Utilise FFmpeg delogo (très rapide)
-    cmd = [
-        'ffmpeg', '-y', '-i', input_path,
-        '-vf', f'delogo=x={x}:y={y}:w={w}:h={h}',
-        '-c:a', 'copy',
-        '-c:v', 'libx264', '-preset', 'ultrafast',
-        output_path
-    ]
-    subprocess.run(cmd)
-
-@app.route('/process', methods=['POST'])
-def process():
-    if 'video' not in request.files:
-        return jsonify({"error": "Aucun fichier"}), 400
-    file = request.files['video']
-    x, y, w, h = int(request.form['x']), int(request.form['y']), int(request.form['w']), int(request.form['h'])
-    file.save("in.mp4")
-    inpaint_video("in.mp4", "out.mp4", x, y, w, h)
-    return send_file("out.mp4", mimetype='video/mp4')
-
-@app.route('/', methods=['GET'])
-def health():
-    return "Backend Viewify Ready!"
-
-if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)
+import os
+import tempfile
+import subprocess
+
+def process_video(video, x1, y1, x2, y2):
+    temp_dir = tempfile.mkdtemp()
+
+    input_video = video
+    frames_dir = os.path.join(temp_dir, "frames")
+    os.makedirs(frames_dir, exist_ok=True)
+
+    # 1) Extraire les frames (FPS réduit)
+    subprocess.run([
+        "ffmpeg", "-i", input_video,
+        "-vf", "fps=10",
+        f"{frames_dir}/frame_%04d.png"
+    ], check=True)
+
+    # 2) Inpainting image par image
+    for frame_name in sorted(os.listdir(frames_dir)):
+        frame_path = os.path.join(frames_dir, frame_name)
+        img = cv2.imread(frame_path)
+
+        mask = np.zeros(img.shape[:2], dtype=np.uint8)
+        mask[int(y1):int(y2), int(x1):int(x2)] = 255
+        mask = cv2.dilate(mask, np.ones((3,3), np.uint8), iterations=1)
+
+        result = cv2.inpaint(img, mask, 3, cv2.INPAINT_TELEA)
+        cv2.imwrite(frame_path, result)
+
+    # 3) Réassembler la vidéo (sans audio)
+    output_video = 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_video
+    ], check=True)
+
+    return output_video
+
+
+with gr.Blocks() as demo:
+    gr.Markdown("## 🎥 Suppression de texte sur vidéo (100 % gratuit)")
+
+    video = gr.Video(label="Vidéo (max 30–60 s)")
+    with gr.Row():
+        x1 = gr.Number(label="X1")
+        y1 = gr.Number(label="Y1")
+        x2 = gr.Number(label="X2")
+        y2 = gr.Number(label="Y2")
+
+    btn = gr.Button("Supprimer le texte")
+    output = gr.Video(label="Vidéo finale")
+
+    btn.click(
+        process_video,
+        inputs=[video, x1, y1, x2, y2],
+        outputs=output
+    )
+
+demo.launch()