from flask_cors import CORS
import os
from flask import Flask, render_template, request, jsonify
from app_rvc import SoniTranslate  # Importuj SoniTranslate z app_rvc.py


app = Flask(__name__, template_folder="templates", static_folder="static")
CORS(app)  # CORS(app, resources={r"/*": {"origins": "http://frontend-server.com"}})

# Ujisti se, že adresář existuje
UPLOAD_FOLDER = "uploads"
if not os.path.exists(UPLOAD_FOLDER):
    os.makedirs(UPLOAD_FOLDER)

@app.route("/")
def index():
    return render_template("index.html")

@app.route("/process_video", methods=["POST"])
def process_video():
    file = request.files["video"]
    target_language = request.form["target_language"]

    # Ulož dočasný soubor
    file_path = os.path.join(UPLOAD_FOLDER, file.filename)
    file.save(file_path)

    try:
        # Inicializuj SoniTranslate a spusť zpracování
        translator = SoniTranslate(cpu_mode=False)
        result = translator.multilingual_media_conversion(
            media_file=file_path,
            target_language=target_language,
            is_gui=False,
        )
        return jsonify({"status": "success", "result": result})
    except Exception as e:
        return jsonify({"status": "error", "message": str(e)})
    finally:
        os.remove(file_path)

if __name__ == "__main__":
    app.run(debug=True)