Spaces:
Paused
Paused
| 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) |