Upload app.py

app.py

@@ -403,24 +403,50 @@ Follow this example structure:
         if progress:
             progress.progress(0.5, "Converting text to speech...")
         
-        # Process TTS in batches to prevent overwhelming the system
+        # Process TTS in batches for concurrent processing
         audio_files = []
         total_lines = len(podcast_json['podcast'])
         
-        for i, item in enumerate(podcast_json['podcast']):
-            if progress:
-                current_progress = 0.5 + (0.4 * (i / total_lines))
-                progress.progress(current_progress, f"Processing speech {i+1}/{total_lines}...")
-                
+        # Define batch size to control concurrency
+        batch_size = 10  # Adjust based on system resources
+        
+        # Process in batches
+        for batch_start in range(0, total_lines, batch_size):
+            batch_end = min(batch_start + batch_size, total_lines)
+            batch = podcast_json['podcast'][batch_start:batch_end]
+            
+            # Create tasks for concurrent processing
+            tts_tasks = []
+            for item in batch:
+                tts_task = self.tts_generate(item['line'], item['speaker'], speaker1, speaker2)
+                tts_tasks.append(tts_task)
+            
             try:
-                audio_file = await self.tts_generate(item['line'], item['speaker'], speaker1, speaker2)
-                audio_files.append(audio_file)
+                # Process batch concurrently
+                batch_results = await asyncio.gather(*tts_tasks, return_exceptions=True)
+                
+                # Check for exceptions and handle results
+                for i, result in enumerate(batch_results):
+                    if isinstance(result, Exception):
+                        # Clean up any files already created
+                        for file in audio_files:
+                            if os.path.exists(file):
+                                os.remove(file)
+                        raise Exception(f"Error generating speech: {str(result)}")
+                    else:
+                        audio_files.append(result)
+                        
+                # Update progress
+                if progress:
+                    current_progress = 0.5 + (0.4 * (batch_end / total_lines))
+                    progress.progress(current_progress, f"Processed {batch_end}/{total_lines} speech segments...")
+            
             except Exception as e:
                 # Clean up any files already created
                 for file in audio_files:
                     if os.path.exists(file):
                         os.remove(file)
-                raise Exception(f"Error generating speech for line {i+1}: {str(e)}")
+                raise Exception(f"Error in batch TTS generation: {str(e)}")
         
         combined_audio = await self.combine_audio_files(audio_files, progress)
         return combined_audio