Create app.py

app.py

@@ -0,0 +1,118 @@
+from transcribe import transcribe
+from moviepy import VideoFileClip
+from translate import translate  
+from edite_video import video_edit
+import gradio as gr
+import requests
+import os
+from dub import dub
+
+
+def extract_audio(input_video_name):
+    # Define the input video file and output audio file
+    mp3_file = "audio.mp3"
+    # Load the video clip
+    video_clip = VideoFileClip(input_video_name)
+
+    # Extract the audio from the video clip
+    audio_clip = video_clip.audio
+    duration = audio_clip.duration
+    print(f"Audio duration: {duration}")
+    # Write the audio to a separate file
+    audio_clip.write_audiofile(mp3_file)
+
+    # Close the video and audio clips
+    audio_clip.close()
+    video_clip.close()
+
+    print("Audio extraction successful!")
+    return mp3_file, duration
+
+
+def download_video(url):
+    response = requests.get(url, stream=True)
+    response.raise_for_status()
+    video_file = "video.mp4"
+    with open(video_file, 'wb') as file:
+        for chunk in response.iter_content(chunk_size=8192):
+            if chunk:
+                file.write(chunk)
+    print("Video downloaded successfully!")
+    return video_file
+
+
+def main(url, clip_type, parameters, progress=gr.Progress()):
+    """
+    This function reports progress by yielding intermediate updates. 
+    Gradio watches these yields to update the progress bar.
+    """
+    if clip_type == "dub":
+        progress(0, desc="پردازش شروع شد")
+        yield "پردازش شروع شد", None
+
+        video = download_video(url)
+        progress(5, desc="در حال دریافت ویدئو")
+        yield "در حال دریافت ویدئو", None
+
+        mp3_file, duration = extract_audio(video)
+        progress(10, desc="استخراج صوت")
+        yield "استخراج صوت", None
+
+        srt_list = transcribe(mp3_file)
+        progress(35, desc="تبدیل متن به صوت")
+        yield "تبدیل متن به صوت", None
+
+        subtitle_file = translate(srt_list)
+        progress(55, desc="در حال ترجمه")
+        yield "در حال ترجمه", None
+
+        output_video_file = dub(subtitle_file, video)
+        progress(100, desc="Dubbing complete")
+        yield "درحال پردازش ویدئو", output_video_file
+        os.remove(subtitle_file)
+
+    else:
+        # Assume parameters is a comma-separated string: "color,font"
+        color, font = parameters.split(",")
+        progress(0, desc="پردازش شروع شد")
+        yield "پردازش شروع شد", None
+
+        video = download_video(url)
+        progress(5, desc="در حال دریافت ویدئو")
+        yield "در حال دریافت ویدئو", None
+
+        mp3_file, duration = extract_audio(video)
+        progress(10, desc="استخراج صوت")
+        yield "استخراج صوت", None
+
+        srt_list = transcribe(mp3_file)
+        progress(35, desc="تبدیل متن به صوت")
+        yield "تبدیل متن به صوت", None
+
+        subtitle_file = translate(srt_list)
+        progress(55, desc="در حال ترجمه")
+        yield "در حال ترجمه", None
+
+        
+        output_video_file = video_edit(subtitle_file, video, color, font, input_audio="audio.mp3")
+        progress(100, desc="Video editing complete")
+        yield "درحال پردازش ویدئو", output_video_file
+        os.remove(subtitle_file)
+
+
+with gr.Blocks() as demo:
+    gr.Markdown("Start typing below and then click **Run** to see the progress and final output.")
+    with gr.Column():
+        progress_output = gr.Textbox(label="Progress", visible=False)
+        video_file_input = gr.Text(label="Video URL")
+        clip_type = gr.Dropdown(["dub", "sub"], label="Clip Type")
+        parameters = gr.Text(label="Additional Parameters (for subtitles: color,font)")
+        btn = gr.Button("Create")
+        video_file_output = gr.Video(label="Result Video")
+        btn.click(
+            fn=main,
+            inputs=[video_file_input, clip_type, parameters],
+            outputs=[progress_output, video_file_output],
+        )
+
+demo.launch(debug=True)