Rename app.file to app.py

app.file → app.py

@@ -1,19 +1,26 @@
+from flask import Flask, render_template, request, send_from_directory, redirect, url_for
+import os
 from pytube import YouTube
 from moviepy.editor import VideoFileClip, AudioFileClip
 from transformers import pipeline, WhisperProcessor, WhisperForConditionalGeneration
-import librosa, scipy
+import librosa
+import scipy
+
+app = Flask(__name__)
+app.config['UPLOAD_FOLDER'] = 'static'
 
 def download_video(url):
     yt = YouTube(url)
     stream = yt.streams.filter(only_audio=True).first()
-    stream.download(filename="video_audio.mp4")
-    return "video_audio.mp4"
+    stream.download(filename=os.path.join(app.config['UPLOAD_FOLDER'], "video_audio.mp4"))
+    return os.path.join(app.config['UPLOAD_FOLDER'], "video_audio.mp4")
 
 def extract_audio(video_file):
     video = VideoFileClip(video_file)
     audio = video.audio
-    audio.write_audiofile("output_audio.wav")
-    return "output_audio.wav"
+    audio_file = os.path.join(app.config['UPLOAD_FOLDER'], "output_audio.wav")
+    audio.write_audiofile(audio_file)
+    return audio_file
 
 def speech_to_text(audio_file):
     processor = WhisperProcessor.from_pretrained("openai/whisper-small")
@@ -31,30 +38,40 @@ def translate_to_hindi(text):
 def text_to_speech(text, output_file="hindi_output.wav"):
     synthesizer = pipeline("text-to-speech", model="facebook/mms-tts-hin")
     speech = synthesizer(text)
-    scipy.io.wavfile.write(output_file, rate=speech["sampling_rate"], data=speech["audio"])
-    return output_file
+    output_path = os.path.join(app.config['UPLOAD_FOLDER'], output_file)
+    scipy.io.wavfile.write(output_path, rate=speech["sampling_rate"], data=speech["audio"])
+    return output_path
 
 def merge_audio_to_video(video_file, audio_file, output_video="hindi_dubbed.mp4"):
     video = VideoFileClip(video_file)
     audio = AudioFileClip(audio_file)
+    output_path = os.path.join(app.config['UPLOAD_FOLDER'], output_video)
     video = video.set_audio(audio)
-    video.write_videofile(output_video)
-    return output_video
-
-if __name__ == "__main__":
-    # YouTube video link input
-    video_url = input("Enter YouTube video URL: ")
-    print("Downloading video...")
-    video_file = download_video(video_url)
-    print("Extracting audio...")
-    audio_file = extract_audio(video_file)
-    print("Transcribing audio...")
-    transcription = speech_to_text(audio_file)
-    print("Translating to Hindi...")
-    hindi_text = translate_to_hindi(transcription)
-    print("Generating Hindi audio...")
-    hindi_audio = text_to_speech(hindi_text)
-    print("Merging audio to video...")
-    output_video = merge_audio_to_video(video_file, hindi_audio)
-    print(f"Hindi dubbed video saved as: {output_video}")
+    video.write_videofile(output_path)
+    return output_path
+
+@app.route('/', methods=['GET', 'POST'])
+def index():
+    if request.method == 'POST':
+        video_url = request.form['video_url']
+        video_file = download_video(video_url)
+        audio_file = extract_audio(video_file)
+        transcription = speech_to_text(audio_file)
+        hindi_text = translate_to_hindi(transcription)
+        hindi_audio = text_to_speech(hindi_text)
+        output_video = merge_audio_to_video(video_file, hindi_audio)
+        return redirect(url_for('result', filename=os.path.basename(output_video)))
+    return render_template('index.html')
+
+@app.route('/result/<filename>')
+def result(filename):
+    return render_template('result.html', filename=filename)
+
+@app.route('/static/<filename>')
+def serve_file(filename):
+    return send_from_directory(app.config['UPLOAD_FOLDER'], filename)
+
+if __name__ == '__main__':
+    os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
+    app.run(host='0.0.0.0', port=5000)