Update app.py

app.py

@@ -160,18 +160,22 @@ def process_video(video_file, youtube_url, language_choice):
     
             segment_files = []
             subtitle_entries = []
-            for segment in translation_segments:
-                segment_file, start, end, translated_text = generate_segment_audio(segment, speaker_id)
-    
+            for i in range(0, len(translation_segments), batch_size):
+            batch = translation_segments[i:i + batch_size]
+            with ThreadPoolExecutor(max_workers=5) as executor:
+                batch_segment_files = list(executor.map(generate_segment_audio_batch, [batch] * len(speaker_ids), [speaker_id] * len(speaker_ids)))
+                batch_segment_files = [item for sublist in batch_segment_files for item in sublist]  # Flatten the list
+
+            for segment_file, start, end, translated_text in batch_segment_files:
                 # Run the tone color converter
                 encode_message = "@MyShell"
                 tone_color_converter.convert(
-                audio_src_path=segment_file,
-                src_se=source_se,
-                tgt_se=target_se,
-                output_path=segment_file,
-                message=encode_message)
-                
+                    audio_src_path=segment_file,
+                    src_se=source_se,
+                    tgt_se=target_se,
+                    output_path=segment_file,
+                    message=encode_message
+                )
                 segment_files.append((segment_file, start, end, translated_text))
     
             # Combine the audio segments