Auto-Dubbing / app.py
imseldrith's picture
Update app.py
588e1a0
import os
import uuid
from flask import Flask, request, render_template, redirect, url_for
from gtts import gTTS
from moviepy.editor import VideoFileClip, AudioFileClip
from tqdm import tqdm
import requests
app = Flask(__name__)
app.secret_key = os.environ.get('SECRET_KEY') or str(uuid.uuid4())
@app.route('/', methods=['GET', 'POST'])
def index():
if request.method == 'POST':
if 'video_file' in request.files:
video_file = request.files['video_file']
elif 'video_url' in request.form:
video_url = request.form['video_url']
response = requests.get(video_url)
video_file = response.content
language = request.form['language']
dubbed_video_file = generate_dubbed_video(video_file, language)
return redirect(url_for('download', video_id=dubbed_video_file))
return render_template('index.html')
@app.route('/download/<video_id>')
def download(video_id):
return render_template('download.html', video_id=video_id)
@app.route("/generate_dubbed_video", methods=["GET", "POST"])
def generate_dubbed_video(video_file, language):
video_id = str(uuid.uuid4())
video = VideoFileClip(video_file)
audio = video.audio
text = audio.to_soundarray().tobytes()
try:
tts = gTTS(text=text.decode(), lang=language)
except UnicodeDecodeError:
tts = gTTS(text=text.decode(errors='ignore'), lang=language)
tts.save(f'{video_id}.mp3')
audio_clip = AudioFileClip(f'{video_id}.mp3')
with tqdm(total=100, desc="Dubbing Progress", unit="%") as progress_bar:
for i, frame in enumerate(video.iter_frames()):
video.set_frame(i)
audio_frame = audio_clip.get_frame(i / video.fps)
video.audio = audio_frame
progress_bar.update(i / len(video.iter_frames()) * 100)
video.write_videofile(f'{video_id}.mp4', fps=video.fps)
return video_id
if __name__ == '__main__':
app.run(host="0.0.0.0",port=7860,debug=True)