Create app.py

app.py

@@ -0,0 +1,121 @@
+from flask import Flask, request, jsonify, send_from_directory
+from flask_cors import CORS  # Importe CORS
+from deepface import DeepFace
+import os
+import tempfile
+import shutil
+import uuid
+
+app = Flask(__name__, static_folder='static')
+CORS(app)  # Active CORS pour toutes les routes de l'application
+
+# Configuration pour l'upload des fichiers
+UPLOAD_FOLDER = 'static/uploads'  # Utilise le dossier 'static' pour servir les fichiers statiques
+ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg'}
+app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
+app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024  # Limite à 16MB
+
+# Fonction pour vérifier l'extension des fichiers
+def allowed_file(filename):
+    return '.' in filename and \
+           filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
+
+# Fonction pour générer un nom de fichier unique
+def unique_filename(filename):
+    name, ext = os.path.splitext(filename)
+    new_name = f"{name}-{uuid.uuid4().hex}{ext}"
+    return new_name
+
+# Route pour servir les fichiers statiques (y compris le HTML, CSS, JS et les images uploadées)
+@app.route('/')
+def index():
+    return send_from_directory(app.static_folder, 'index.html')
+
+# Route pour la comparaison de visages
+@app.route('/verify', methods=['POST'])
+def verify_faces():
+    if 'image1' not in request.files or 'image2' not in request.files:
+        return jsonify({'error': 'Deux images sont requises pour la comparaison.'}), 400
+
+    image1 = request.files['image1']
+    image2 = request.files['image2']
+
+    if image1.filename == '' or image2.filename == '':
+        return jsonify({'error': 'Les noms de fichiers ne peuvent pas être vides.'}), 400
+
+    if image1 and allowed_file(image1.filename) and image2 and allowed_file(image2.filename):
+        temp_dir = tempfile.mkdtemp(dir=app.config['UPLOAD_FOLDER'])
+        
+        # Sauvegarder les images temporairement avec des noms de fichiers uniques
+        image1_filename = unique_filename(image1.filename)
+        image2_filename = unique_filename(image2.filename)
+        image1_path = os.path.join(temp_dir, image1_filename)
+        image2_path = os.path.join(temp_dir, image2_filename)
+        image1.save(image1_path)
+        image2.save(image2_path)
+
+        try:
+            result = DeepFace.verify(img1_path=image1_path, img2_path=image2_path)
+
+            # Renvoyer les résultats de la comparaison et les URLs des images
+            result['image1_url'] = os.path.join(app.config['UPLOAD_FOLDER'], image1_filename)
+            result['image2_url'] = os.path.join(app.config['UPLOAD_FOLDER'], image2_filename)
+
+            # Déplacer les images du dossier temporaire vers le dossier d'upload
+            shutil.move(image1_path, os.path.join(app.config['UPLOAD_FOLDER'], image1_filename))
+            shutil.move(image2_path, os.path.join(app.config['UPLOAD_FOLDER'], image2_filename))
+
+            # Nettoyer le dossier temporaire
+            os.rmdir(temp_dir)
+
+            return jsonify(result)
+        except Exception as e:
+            # Nettoyer en cas d'erreur
+            if os.path.exists(temp_dir):
+                shutil.rmtree(temp_dir, ignore_errors=True)
+            return jsonify({'error': str(e)}), 500
+    else:
+        return jsonify({'error': 'Extensions de fichiers non autorisées.'}), 400
+
+# Route pour l'analyse d'un visage
+@app.route('/analyze', methods=['POST'])
+def analyze_face():
+    if 'image' not in request.files:
+        return jsonify({'error': 'Aucune image fournie.'}), 400
+
+    image = request.files['image']
+
+    if image.filename == '':
+        return jsonify({'error': 'Le nom de fichier ne peut pas être vide.'}), 400
+
+    if image and allowed_file(image.filename):
+        temp_dir = tempfile.mkdtemp(dir=app.config['UPLOAD_FOLDER'])
+        image_filename = unique_filename(image.filename)
+        image_path = os.path.join(temp_dir, image_filename)
+        image.save(image_path)
+
+        try:
+            result = DeepFace.analyze(img_path=image_path, actions=['age', 'gender', 'race', 'emotion'])
+
+            # Renvoyer les résultats de l'analyse et l'URL de l'image
+            result['image_url'] = os.path.join(app.config['UPLOAD_FOLDER'], image_filename)
+
+            # Déplacer l'image du dossier temporaire vers le dossier d'upload
+            shutil.move(image_path, os.path.join(app.config['UPLOAD_FOLDER'], image_filename))
+
+            # Nettoyer le dossier temporaire
+            os.rmdir(temp_dir)
+
+            return jsonify(result)
+        except Exception as e:
+            # Nettoyer en cas d'erreur
+            if os.path.exists(temp_dir):
+                shutil.rmtree(temp_dir, ignore_errors=True)
+            return jsonify({'error': str(e)}), 500
+    else:
+        return jsonify({'error': 'Extension de fichier non autorisée.'}), 400
+
+if __name__ == '__main__':
+    if not os.path.exists(UPLOAD_FOLDER):
+        os.makedirs(UPLOAD_FOLDER)
+    app.run(debug=True)