Spaces:
Build error
Build error
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 | |
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") | |
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) |