Update app.py

app.py

@@ -6,98 +6,66 @@ import time
 from gradio_client import Client, handle_file
 
 
-def main(url, clip_type, parameters, progress=gr.Progress()):
+def main(url, 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
-        time.sleep(2)
-        
-        progress(5, desc="در حال دریافت ویدئو")
-        yield "در حال دریافت ویدئو", None
-        video = download_video(url)
-
-        #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()
-        
-
-        progress(55, desc="در حال ترجمه")
-        yield "در حال ترجمه", None
-        subtitle_file = translate(srt_file)
-
-
-        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"
+    # Assume parameters is a comma-separated string: "color,font"
+    if parameters:
         color, font = parameters.split(",")
-        progress(0, desc="پردازش شروع شد")
-        yield "پردازش شروع شد", None
-        time.sleep(2)
-        
-        progress(0.35, desc="تبدیل متن به صوت")
-        yield "تبدیل متن به صوت", None
-        client = Client("rayesh/transcribe")
-        srt_file, video, mp3_file = client.predict(
-        		url=url,
-        		api_name="/transcribe"
-        )
-        client.close()
-        print(mp3_file)
-        progress(0.55, desc="در حال ترجمه")
-        yield "در حال ترجمه", None
-        client = Client("rayesh/translate")
-        subtitle_file = client.predict(
-        		file=handle_file(srt_file),
-                max_chars=3000,
-        		api_name="/translate"
-        )
-        client.close()
+    else:
+        color, font = "yellow", "arial"
+    progress(0, desc="پردازش شروع شد")
+    yield "پردازش شروع شد", None
+    time.sleep(2)
+    
+    progress(0.35, desc="تبدیل متن به صوت")
+    yield "تبدیل متن به صوت", None
+    client = Client("rayesh/transcribe")
+    srt_file, video, mp3_file = client.predict(
+            url=url,
+            api_name="/transcribe"
+    )
+    client.close()
+    print(mp3_file)
+    progress(0.55, desc="در حال ترجمه")
+    yield "در حال ترجمه", None
+    client = Client("rayesh/translate")
+    subtitle_file = client.predict(
+            file=handle_file(srt_file),
+            max_chars=3000,
+            api_name="/translate"
+    )
+    client.close()
 
-        progress(1.0, desc="Video editing complete")
-        yield "درحال پردازش ویدئو", None
-        client = Client("SPACERUNNER99/video_edite")
-        output_video_file = client.predict(
-        		srt=handle_file(subtitle_file),
-        		input_video={"video":handle_file(video["video"])},
-        		color=color,
-        		font=font,
-                input_audio=handle_file(mp3_file),
-        		api_name="/video_edit"
-        )
-        client.close()
-        output_video_file = output_video_file['video']
-        yield "درحال پردازش ویدئو", output_video_file
+    progress(1.0, desc="Video editing complete")
+    yield "درحال پردازش ویدئو", None
+    client = Client("SPACERUNNER99/video_edite")
+    output_video_file = client.predict(
+            srt=handle_file(subtitle_file),
+            input_video={"video":handle_file(video["video"])},
+            color=color,
+            font=font,
+            input_audio=handle_file(mp3_file),
+            api_name="/video_edit"
+    )
+    client.close()
+    output_video_file = output_video_file['video']
+    yield "درحال پردازش ویدئو", output_video_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)")
+        video_file_input = gr.Video(label="Upload Video")
+        parameters = gr.Text(label="pick font and font color(font = arial, yekan, nazanin/ colors = black, yellow, white) and input it int the following format: font,font_color")
         btn = gr.Button("Create")
         video_file_output = gr.Video(label="Result Video")
         btn.click(
             fn=main,
-            inputs=[video_file_input, clip_type, parameters],
+            inputs=[video_file_input, parameters],
             outputs=[progress_output, video_file_output],
             concurrency_limit=4,
         )