from flask import Flask, request, send_file
import os
import subprocess
from werkzeug.utils import secure_filename

app = Flask(__name__)
UPLOAD_FOLDER = "uploads"
OUTPUT_FOLDER = "outputs"
os.makedirs(UPLOAD_FOLDER, exist_ok=True)
os.makedirs(OUTPUT_FOLDER, exist_ok=True)

@app.route("/sync", methods=["POST"])
def sync_lips():
    if "video" not in request.files or "audio" not in request.files:
        return {"error": "Missing video or audio file"}, 400
    
    video = request.files["video"]
    audio = request.files["audio"]
    
    video_filename = secure_filename(video.filename)
    audio_filename = secure_filename(audio.filename)
    
    video_path = os.path.join(UPLOAD_FOLDER, video_filename)
    audio_path = os.path.join(UPLOAD_FOLDER, audio_filename)
    output_path = os.path.join(OUTPUT_FOLDER, f"synced_{video_filename}")
    
    video.save(video_path)
    audio.save(audio_path)
    
    # Run LatentSync processing
    command = ["python", "scripts/inference.py", "--video", video_path, "--audio", audio_path, "--output", output_path]
    result = subprocess.run(command, capture_output=True, text=True)
    
    if result.returncode != 0:
        return {"error": "Lip sync processing failed", "details": result.stderr}, 500
    
    return send_file(output_path, mimetype="video/mp4", as_attachment=True)

if __name__ == "__main__":
    app.run(debug=True, host="0.0.0.0", port=5000)