from flask import Flask, render_template, request, redirect, url_for
import subprocess
import os
os.system("cd roop")
app = Flask(__name__)

@app.route('/', methods=['GET', 'POST'])
def index():
    if request.method == 'POST':
        # Get form inputs
        target_path = request.form['target_path']
        output_quality = request.form['output_quality']
        source_path = request.form['source_path']
        output_path = request.form['output_path']
        execution_provider = request.form['execution_provider']
        frame_processors = request.form.getlist('frame_processors')

        # Construct command
        command = [
            "python", "run.py",
            "--target", target_path,
            "--output-video-quality", output_quality,
            "--source", source_path,
            "-o", output_path,
            "--execution-provider", execution_provider,
        ]

        # Add frame processors if provided
        if frame_processors:
            command.extend(["--frame-processor"] + frame_processors)

        # Run the command
        subprocess.run(command)

        return redirect(url_for('index'))

    return render_template('./index.html')

if __name__ == '__main__':
    import uvicorn
    uvicorn.run(app, host="0.0.0.0", port=7860, debug=True)
    print(" Its not workin")