Update app.py

app.py

@@ -8,22 +8,19 @@ import gradio as gr
 def main(url, parameters, progress=gr.Progress()):
     try:
         font_type, font_size, font_color, service, target, style, subject = parameters.split(',')
-        
         progress(0, desc="پردازش شروع شد")
         yield "پردازش شروع شد", None
 
-        # Transcription Stage - Updated API endpoint
+        # Transcription Stage
         progress(0.35, desc="تبدیل صوت به متن")
         yield "تبدیل صوت به متن", None
         
-        transcribe_client = Client("rayesh/transcribe")
+        transcribe_client = Client("kavehtaheri/transcribe")
         try:
-            # Changed to use default API name
-            job = transcribe_client.submit(url)
+            job = transcribe_client.submit(url, api_name="/transcribe")
             while not job.done():
                 time.sleep(0.5)
-                progress(0.35, desc="تبدیل صوت به متن")
-            
+                
             results = job.outputs()[0]
             print(results)
             if len(results) != 3:
@@ -33,41 +30,41 @@ def main(url, parameters, progress=gr.Progress()):
         finally:
             transcribe_client.close()
 
-        # Translation Stage - Updated API endpoint
+        # Translation Stage
         progress(0.55, desc="در حال ترجمه")
         yield "در حال ترجمه", None
         
-        translate_client = Client("rayesh/translate")
+        translate_client = Client("kavehtaheri/translate")
         try:
-            # Changed to use default API name
-            job = translate_client.submit(results[0], target)
+            job = translate_client.submit(
+                handle_file(srt_file),
+                3000,
+                api_name="/translate"
+            )
             while not job.done():
-                time.sleep(0.5)
-                progress(0.55, desc="در حال ترجمه")
-            
-            translated_text = job.outputs()[0]
+                time.sleep(0.3)
+            subtitle_file = job.outputs()[0]
         finally:
             translate_client.close()
 
-        # Video Processing Stage - Updated API endpoint
+        # Video Processing Stage
         progress(0.75, desc="پردازش ویدئو")
         yield "درحال پردازش ویدئو", None
         
-        video_client = Client("SPACERUNNER99/video_edite")
+        video_client = Client("kavehtaheri/video_edite")
         try:
-            # Changed API name to match typical video processing endpoints
             job = video_client.submit(
-                url,
-                translated_text,
+                handle_file(subtitle_file),
+                {"video": handle_file(video["video"])},
+                font_color,
                 font_type,
                 font_size,
-                font_color,
-                api_name="/process"  # Changed to common endpoint name
+                handle_file(mp3_file),
+                api_name="/video_edit"
             )
             while not job.done():
                 time.sleep(0.5)
-                progress(0.75, desc="پردازش ویدئو")
-            
+                
             output_video = job.outputs()[0]['video']