import os
import uuid
import requests

from flask import Flask, render_template, request, redirect, url_for, flash
from gtts import gTTS
from moviepy.editor import VideoFileClip, AudioFileClip

app = Flask(__name__)
app.secret_key = "secret_key"

def generate_dubbed_video(video_file, language):
    # Extract audio from video
    video = VideoFileClip(video_file)
    audio = video.audio
    audio_file = f"{uuid.uuid4()}.mp3"
    audio.write_audiofile(audio_file)
    
    # Generate dubbed audio
    audio_data = audio.to_soundarray().tobytes()
    audio_data = audio_data.encode('utf-8', 'ignore')
    tts = gTTS(text=audio_data.decode(), lang=language)
    #tts = gTTS(text=audio.to_soundarray().tobytes().decode(), lang=language)
    dubbed_audio_file = f"{uuid.uuid4()}.mp3"
    tts.save(dubbed_audio_file)
    
    # Replace video audio with dubbed audio
    dubbed_audio = AudioFileClip(dubbed_audio_file)
    dubbed_video = video.set_audio(dubbed_audio)
    dubbed_video_file = f"{uuid.uuid4()}.mp4"
    dubbed_video.write_videofile(dubbed_video_file)
    
    # Clean up temporary audio and video files
    os.remove(audio_file)
    os.remove(dubbed_audio_file)
    
    return dubbed_video_file

@app.route("/", methods=["GET", "POST"])
def index():
    if request.method == "POST":
        if "video_url" in request.form:
            video_url = request.form["video_url"]
            video_file = f"{uuid.uuid4()}.mp4"
            response = requests.get(video_url)
            open(video_file, "wb").write(response.content)
        elif "video_file" in request.files:
            video_file = request.files["video_file"]
            video_file.save(f"{uuid.uuid4()}.mp4")
        
        language = request.form.get("language")
        dubbed_video_file = generate_dubbed_video(video_file, language)
        flash("Dubbed video generated successfully!")
        return redirect(url_for("download", dubbed_video_file=dubbed_video_file))
    
    return render_template("index.html")

@app.route("/download/<dubbed_video_file>")
def download(dubbed_video_file):
    return redirect(url_for("static", filename=dubbed_video_file))

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