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)