Update app.py

app.py

@@ -1,59 +1,64 @@
-from flask import Flask, render_template, request, redirect, url_for, send_file
+from flask import Flask, request, render_template, send_file
 import moviepy.editor as mp
+import os
 import cv2
 from gtts import gTTS
-import os
-import progress
-import uuid
+import numpy as np
+from progress.bar import Bar
 import requests
-from bs4 import BeautifulSoup
 
 app = Flask(__name__)
+app.config["TEMP_FOLDER"] = "temp"
 
-@app.route('/')
+@app.route("/")
 def index():
-    return render_template('index.html')
+    return render_template("index.html")
 
-@app.route('/dub', methods=['POST'])
+@app.route("/dub", methods=["POST"])
 def dub():
-    # Get the video source from the user
-    source = request.form.get('source')
-    if source == 'url':
-        url = request.form.get('url')
-        # Download the video from the URL
+    # Check if a video was uploaded
+    if "video" in request.files:
+        video = request.files["video"]
+        filename = video.filename
+        video.save(os.path.join(app.config["TEMP_FOLDER"], filename))
+    else:
+        # Get video from URL
+        url = request.form["url"]
+        filename = url.split("/")[-1]
         response = requests.get(url)
-        file = "video.mp4"
-        with open(file, "wb") as f:
+        with open(os.path.join(app.config["TEMP_FOLDER"], filename), "wb") as f:
             f.write(response.content)
-    else:
-        file = request.files['file']
-        filename = str(uuid.uuid4().hex) + ".mp4"
-        file.save(filename)
-    
-    # Load the video using moviepy
-    video = mp.VideoFileClip(file)
 
-    # Create the audio in Hindi using gTTS
-    audio = gTTS(video.audio.fps, 'hi')
-    audio_filename = str(uuid.uuid4().hex) + ".mp3"
-    audio.save(audio_filename)
+    # Load video
+    clip = mp.VideoFileClip(os.path.join(app.config["TEMP_FOLDER"], filename))
+
+    # Get audio
+    audio = clip.audio
+    n = len(audio.fps)
 
-    # Replace the audio of the video with the Hindi audio
-    video.audio = mp.AudioFileClip(audio_filename)
+    bar = Bar("Dubbing", max=n)
+    text = ""
+    for i in range(n):
+        bar.next()
+        frame = audio.get_frame(i/audio.fps)
+        gray = cv2.cvtColor(frame, cv2.COLOR_RGB2GRAY)
+        text += " ".join([str(x) for x in gray.flatten()]) + " "
+    bar.finish()
 
-    # Save the dubbed video
-    dubbed_filename = str(uuid.uuid4().hex) + ".mp4"
-    video.write_videofile(dubbed_filename, progress_bar=progress.bar.IncrementalBar)
+    # Convert audio to text-to-speech in Hindi
+    tts = gTTS(text, lang="hi")
+    tts.save("temp/dubbed_audio.mp3")
 
-    # Delete the temporary audio and video files
-    os.remove(file)
-    os.remove(audio_filename)
+    # Add dubbed audio to video
+    dubbed_audio = mp.AudioFileClip("temp/dubbed_audio.mp3")
+    dubbed_video = clip.set_audio(dubbed_audio)
+    dubbed_video.write_videofile(os.path.join(app.config["TEMP_FOLDER"], "dubbed_" + filename))
 
-    return redirect(url_for('download', filename=dubbed_filename))
+    return render_template("dub.html", filename="dubbed_" + filename)
 
-@app.route('/download/<filename>')
+@app.route("/download/<filename>")
 def download(filename):
-    return send_file(filename, as_attachment=True)
+    return send_file(os.path.join(app.config["TEMP_FOLDER"], filename), as_attachment=True)
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     app.run(host="0.0.0.0",port=7860,debug=True)