Update app.py

app.py

@@ -154,11 +154,14 @@ def process_video(video_file, language_choice):
         model = TTS(language=language, device=device)
         speaker_ids = model.hps.data.spk2id
     
-        def generate_segment_audio(segment, speaker_id):
-            start, end, translated_text = segment
-            segment_path = os.path.join(output_dir, f'segment_{start}_{end}.wav')
-            model.tts_to_file(translated_text, speaker_id, segment_path, speed=speed)
-            return segment_path, start, end, translated_text
+        def generate_segment_audio_batch(translation_batch, speaker_id):
+            segment_files = []
+            for segment in translation_batch:
+                start, end, translated_text = segment
+                segment_path = os.path.join(output_dir, f'segment_{start}_{end}.wav')
+                model.tts_to_file(translated_text, speaker_id, segment_path, speed=speed)
+                segment_files.append((segment_path, start, end, translated_text))
+            return segment_files
     
         for speaker_key in speaker_ids.keys():
             speaker_id = speaker_ids[speaker_key]