Update app.py

app.py

@@ -12,39 +12,55 @@ def main(url, parameters, progress=gr.Progress()):
         font_type, font_size, font_color, service, target, style, subject = parameters.split(',')
         progress(0, desc="پردازش شروع شد")
         yield "پردازش شروع شد", None
-        
-        # Transcription
+
+        # Transcription Stage
         progress(0.35, desc="تبدیل صوت به متن")
         yield "تبدیل متن به صوت", None
-        with Client("rayesh/transcribe") as client:
-            job = client.submit(url, api_name="/transcribe")
+        
+        transcribe_client = Client("rayesh/transcribe")
+        try:
+            job = transcribe_client.submit(url, api_name="/transcribe")
             while not job.done():
                 time.sleep(0.5)
+                progress(0.35 + (job.status().progress * 0.15))
+            
             if job.failed():
-                raise gr.Error("Transcription failed")
+                raise gr.Error("Transcription API failed!")
+                
             results = job.outputs()
             if len(results) != 3:
                 raise ValueError(f"Expected 3 outputs, got {len(results)}")
+                
             srt_file, video, mp3_file = results
+        finally:
+            transcribe_client.close()
 
-        # Translation
+        # Translation Stage
         progress(0.55, desc="در حال ترجمه")
         yield "در حال ترجمه", None
-        with Client("rayesh/translate") as client:
-            job = client.submit(
+        
+        translate_client = Client("rayesh/translate")
+        try:
+            job = translate_client.submit(
                 handle_file(srt_file),
                 3000,
                 api_name="/translate"
             )
             while not job.done():
-                time.sleep(0.5)
+                time.sleep(0.3)
+                progress(0.55 + (job.status().progress * 0.15))
+                
             subtitle_file = job.outputs()[0]
+        finally:
+            translate_client.close()
 
-        # Video Processing
+        # Video Processing Stage
         progress(0.75, desc="پردازش ویدئو")
         yield "درحال پردازش ویدئو", None
-        with Client("SPACERUNNER99/video_edite") as client:
-            job = client.submit(
+        
+        video_client = Client("SPACERUNNER99/video_edite")
+        try:
+            job = video_client.submit(
                 handle_file(subtitle_file),
                 {"video": handle_file(video["video"])},
                 font_color,
@@ -55,7 +71,11 @@ def main(url, parameters, progress=gr.Progress()):
             )
             while not job.done():
                 time.sleep(0.5)
+                progress(0.75 + (job.status().progress * 0.25))
+                
             output_video = job.outputs()[0]['video']
+        finally:
+            video_client.close()
 
         yield "پردازش کامل شد", output_video