Dub / app.py
imseldrith's picture
Update app.py
37df380
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)