Update app.py

app.py

@@ -1,21 +1,14 @@
-from flask import Flask, request, redirect, url_for, send_from_directory, render_template
-from werkzeug.utils import secure_filename
-import os
+import gradio as gr
 import cv2
 import dlib
 import numpy as np
+import os
 
 UPLOAD_FOLDER = 'uploads'
 ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif', 'mp4', 'avi'}
 
-app = Flask(__name__)
-app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
-
 detector = dlib.get_frontal_face_detector()
 
-def allowed_file(filename):
-    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
-
 def detect_faces(image_path):
     img = cv2.imread(image_path)
     gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
@@ -38,44 +31,34 @@ def swap_faces(img, face1, face2):
 
     return img
 
-@app.route('/')
-def index():
-    return render_template('upload.html')
-
-@app.route('/upload', methods=['POST'])
-def upload_file():
-    if 'file' not in request.files:
-        return redirect(request.url)
-    file = request.files['file']
-    if file.filename == '':
-        return redirect(request.url)
-    if file and allowed_file(file.filename):
-        filename = secure_filename(file.filename)
-        file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
-        return redirect(url_for('process_file', filename=filename))
-    return redirect(request.url)
-
-@app.route('/uploads/<filename>')
-def uploaded_file(filename):
-    return send_from_directory(app.config['UPLOAD_FOLDER'], filename)
-
-@app.route('/process/<filename>')
-def process_file(filename):
-    image_path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
-    faces, img = detect_faces(image_path)
+def process_file(image):
+    if image is None:
+        return "Please upload an image."
+    
+    filename = os.path.join(UPLOAD_FOLDER, 'uploaded_image.jpg')
+    cv2.imwrite(filename, image)
 
+    faces, img = detect_faces(filename)
     if len(faces) < 2:
-        return "Need at least two faces to swap", 400
+        return "Need at least two faces to swap"
 
     swapped_img = swap_faces(img, faces[0], faces[1])
     
-    result_filename = 'result_' + filename
-    result_path = os.path.join(app.config['UPLOAD_FOLDER'], result_filename)
-    cv2.imwrite(result_path, swapped_img)
+    result_filename = os.path.join(UPLOAD_FOLDER, 'result_image.jpg')
+    cv2.imwrite(result_filename, swapped_img)
+    return swapped_img
+
+if not os.path.exists(UPLOAD_FOLDER):
+    os.makedirs(UPLOAD_FOLDER)
 
-    return redirect(url_for('uploaded_file', filename=result_filename))
+iface = gr.Interface(
+    fn=process_file,
+    inputs=gr.Image(type="numpy", label="Upload an Image"),
+    outputs=gr.Image(type="numpy", label="Swapped Faces"),
+    title="Face Swap",
+    description="Upload an image with at least two faces, and this tool will swap the faces."
+)
 
 if __name__ == "__main__":
-    if not os.path.exists(UPLOAD_FOLDER):
-        os.makedirs(UPLOAD_FOLDER)
-    app.run(debug=True)
+    iface.launch()
+