Update app.py

app.py

@@ -46,7 +46,7 @@ async def generate_accurate_srt(batch_text, batch_num, start_offset):
 
     return srt_content, audio_file, start_time
 
-# Batch processing function with concurrent processing and progress indicator
+# Batch processing function with concurrent processing and manual progress update
 async def batch_process_srt_and_audio(script_text, progress=gr.Progress()):
     batches = [script_text[i:i+500] for i in range(0, len(script_text), 500)]
     all_srt_content = ""
@@ -59,9 +59,11 @@ async def batch_process_srt_and_audio(script_text, progress=gr.Progress()):
         for batch_num, batch_text in enumerate(batches)
     ]
     
-    # Execute tasks concurrently with progress tracking
-    for result in progress.track(asyncio.as_completed(tasks), total=len(tasks), description="Processing batches..."):
-        srt_content, audio_file, end_offset = await result
+    # Execute tasks concurrently and update progress after each completion
+    completed_tasks = 0
+    total_tasks = len(tasks)
+    for task in asyncio.as_completed(tasks):
+        srt_content, audio_file, end_offset = await task
         all_srt_content += srt_content
 
         # Append the audio of each batch to the combined audio
@@ -72,6 +74,10 @@ async def batch_process_srt_and_audio(script_text, progress=gr.Progress()):
         # Clean up the individual batch audio file
         os.remove(audio_file)
 
+        # Update progress
+        completed_tasks += 1
+        progress(completed_tasks / total_tasks)
+
     # Export combined audio and SRT
     combined_audio.export("final_audio.wav", format="wav")
     with open("final_subtitles.srt", "w") as srt_file: