Update app.py

app.py

@@ -12,30 +12,32 @@ def main(url, parameters, progress=gr.Progress()):
         progress(0, desc="پردازش شروع شد")
         yield "پردازش شروع شد", None
 
-        # Transcription Stage
+        # Transcription Stage - Updated API endpoint
         progress(0.35, desc="تبدیل صوت به متن")
         yield "تبدیل صوت به متن", None
         
         transcribe_client = Client("rayesh/transcribe")
         try:
-            job = transcribe_client.submit(url, api_name="/predict")
+            # Changed to use default API name
+            job = transcribe_client.submit(url)
             while not job.done():
                 time.sleep(0.5)
                 progress(0.35 + job.progress*0.2, desc="تبدیل صوت به متن")
             
-            results = job.outputs()[0]
-            if len(results) != 3:
-                raise ValueError(f"Expected 3 outputs, got {len(results)}")
+            results = job.outputs()
+            if len(results) < 1:
+                raise ValueError("No transcription results returned")
         finally:
             transcribe_client.close()
 
-        # Translation Stage
+        # Translation Stage - Updated API endpoint
         progress(0.55, desc="در حال ترجمه")
         yield "در حال ترجمه", None
         
         translate_client = Client("rayesh/translate")
         try:
-            job = translate_client.submit(results[0], target, api_name="/predict")
+            # Changed to use default API name
+            job = translate_client.submit(results[0], target)
             while not job.done():
                 time.sleep(0.5)
                 progress(0.55 + job.progress*0.2, desc="در حال ترجمه")
@@ -44,34 +46,38 @@ def main(url, parameters, progress=gr.Progress()):
         finally:
             translate_client.close()
 
-        # Video Processing Stage
+        # Video Processing Stage - Updated API endpoint
         progress(0.75, desc="پردازش ویدئو")
         yield "درحال پردازش ویدئو", None
         
         video_client = Client("SPACERUNNER99/video_edite")
         try:
+            # Changed API name to match typical video processing endpoints
             job = video_client.submit(
                 url,
                 translated_text,
                 font_type,
                 font_size,
                 font_color,
-                api_name="/predict"
+                api_name="/process"  # Changed to common endpoint name
             )
             while not job.done():
                 time.sleep(0.5)
                 progress(0.75 + job.progress*0.25, desc="پردازش ویدئو")
             
-            output_video = job.outputs()[0]['video']
+            output_video = job.outputs()[0]
+            
+            # Handle different output formats
+            if isinstance(output_video, dict):
+                output_video = output_video.get('video', output_video.get('output'))
             
             # Create temporary file
             with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as tmp_file:
-                # Download the video content
                 response = requests.get(output_video)
                 tmp_file.write(response.content)
                 tmp_file_path = tmp_file.name
             
-            # Create downloadable URL
+            # Generate direct download link
             base_url = os.environ.get('GRADIO_SERVER_NAME', 'http://localhost:7860')
             download_url = f"{base_url}/file/{tmp_file_path}"