Update app.py

app.py

@@ -1,40 +1,60 @@
-import gradio as gr
-import moviepy.editor as mp
-import numpy as np
-import IPython.display as ipd
-import librosa
-import soundfile as sf
+import os
+import uuid
+import requests
+
+from flask import Flask, render_template, request, redirect, url_for, flash
 from gtts import gTTS
-from io import BytesIO
+from moviepy.editor import VideoFileClip, AudioFileClip
 
-# function to convert English audio to Hindi
-def convert_to_hindi(audio):
-    tts = gTTS(audio, lang='hi')
-    audio = tts.get_wav_data()
-    return audio
+app = Flask(__name__)
+app.secret_key = "secret_key"
 
-# function to extract audio from video and dub it to Hindi
-def dub_video(video_file):
-    # extract audio from video
-    video = mp.VideoFileClip(video_file)
+def generate_dubbed_video(video_file, language):
+    # Extract audio from video
+    video = VideoFileClip(video_file)
     audio = video.audio
-    audio_array = audio.to_soundarray()
+    audio_file = f"{uuid.uuid4()}.mp3"
+    audio.write_audiofile(audio_file)
+    
+    # Generate dubbed audio
+    tts = gTTS(text=audio.to_soundarray().tobytes().decode(), lang=language)
+    dubbed_audio_file = f"{uuid.uuid4()}.mp3"
+    tts.save(dubbed_audio_file)
     
-    # convert audio to Hindi
-    audio_text = librosa.core.to_text(audio_array)
-    audio_hindi = convert_to_hindi(audio_text)
+    # 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)
     
-    # add Hindi audio to video
-    audio_hindi = mp.AudioFileClip(BytesIO(audio_hindi))
-    video_dubbed = video.set_audio(audio_hindi)
+    # 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 video_dubbed
+    return render_template("index.html")
 
-# Gradio interface
-def dub_video_interface(inputs):
-    video_file = inputs[0]
-    video_dubbed = dub_video(video_file)
-    return video_dubbed
+@app.route("/download/<dubbed_video_file>")
+def download(dubbed_video_file):
+    return redirect(url_for("static", filename=dubbed_video_file))
 
-# run the Gradio interface
-gr.Interface(fn=dub_video_interface, inputs=gr.Video(label="Upload a video"), outputs=gr.Video(label="Dubbed Video"),title="Auto Dubbing").launch()
+if __name__ == "__main__":
+    app.run(host="0.0.0.0",port=7860,debug=True)