Spaces:
Sleeping
Sleeping
File size: 4,906 Bytes
4e90819 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 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) |