Update app.py

app.py

@@ -1,12 +1,8 @@
-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
-
+from gradio_client import Client, handle_file
 
 def extract_audio(input_video_name):
     # Define the input video file and output audio file
@@ -49,22 +45,30 @@ def main(url, clip_type, parameters, progress=gr.Progress()):
     if clip_type == "dub":
         progress(0, desc="پردازش شروع شد")
         yield "پردازش شروع شد", None
-
-        video = download_video(url)
+        time.sleep(2)
+        
         progress(5, desc="در حال دریافت ویدئو")
         yield "در حال دریافت ویدئو", None
+        video = download_video(url)
 
-        mp3_file, duration = extract_audio(video)
-        progress(10, desc="استخراج صوت")
-        yield "استخراج صوت", None
-
-        srt_list = transcribe(mp3_file)
+        #progress(10, desc="استخراج صوت")
+        #yield "استخراج صوت", None
+        #mp3_file, duration = extract_audio(video)
+        
         progress(35, desc="تبدیل متن به صوت")
         yield "تبدیل متن به صوت", None
+        client = Client("rayesh/transcribe")
+        srt_file = client.predict(
+        		mp3_file=handle_file(video),
+        		api_name="/transcribe"
+        )
+        client.close()
+        
 
-        subtitle_file = translate(srt_list)
         progress(55, desc="در حال ترجمه")
         yield "در حال ترجمه", None
+        subtitle_file = translate(srt_file)
+
 
         output_video_file = dub(subtitle_file, video)
         progress(100, desc="Dubbing complete")
@@ -77,27 +81,45 @@ def main(url, clip_type, parameters, progress=gr.Progress()):
         progress(0, desc="پردازش شروع شد")
         yield "پردازش شروع شد", None
 
-        video = download_video(url)
-        progress(5, desc="در حال دریافت ویدئو")
+        progress(0.05, desc="در حال دریافت ویدئو")
         yield "در حال دریافت ویدئو", None
+        video = download_video(url)
 
-        mp3_file, duration = extract_audio(video)
         progress(10, desc="استخراج صوت")
         yield "استخراج صوت", None
+        mp3_file, duration = extract_audio(video)
 
-        srt_list = transcribe(mp3_file)
-        progress(35, desc="تبدیل متن به صوت")
+        progress(0.35, desc="تبدیل متن به صوت")
         yield "تبدیل متن به صوت", None
+        client = Client("rayesh/transcribe")
+        srt_file = client.predict(
+        		mp3_file=handle_file(mp3_file),
+        		api_name="/transcribe"
+        )
+        client.close()
 
-        subtitle_file = translate(srt_list)
-        progress(55, desc="در حال ترجمه")
+        progress(0.55, desc="در حال ترجمه")
         yield "در حال ترجمه", None
+        Client("rayesh/translate")
+        subtitle_file = client.predict(
+        		file=handle_file(srt_file),
+                max_chars=3000,
+        		api_name="/translate"
+        )
+        client.close()
 
-        
-        output_video_file = video_edit(subtitle_file, video, color, font, input_audio="audio.mp3")
-        progress(100, desc="Video editing complete")
+        progress(1.0, desc="Video editing complete")
         yield "درحال پردازش ویدئو", output_video_file
-        os.remove(subtitle_file)
+        client = Client("SPACERUNNER99/video_edite")
+        output_video_file = client.predict(
+        		srt=handle_file(subtitle_file),
+        		input_video={"video":handle_file(video)},
+        		color=color,
+        		font=font,
+                audio_in=mp3_file,
+        		api_name="/video_edit"
+        )
+        client.close()
 
 
 with gr.Blocks() as demo: