Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 from utils import generate_script, generate_audio, truncate_text, extract_text_from_url
 from prompts import SYSTEM_PROMPT
-from pydub import AudioSegment
 import pypdf
 import os
 import tempfile
@@ -30,20 +29,21 @@ def generate_podcast(file, url, tone, length):
         
         script = generate_script(SYSTEM_PROMPT, truncated_text, tone, length)
         
-        audio_segments = []
+        audio_files = []
         transcript = ""
         for item in script.dialogue:
             audio_file = generate_audio(item.text, item.speaker)
-            audio_segment = AudioSegment.from_wav(audio_file)
-            audio_segments.append(audio_segment)
+            audio_files.append(audio_file)
             transcript += f"**{item.speaker}**: {item.text}\n\n"
-            os.remove(audio_file)  # Clean up temporary audio file
         
-        combined_audio = sum(audio_segments)
+        # Combine audio files
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as combined_audio:
+            for audio_file in audio_files:
+                with open(audio_file, 'rb') as f:
+                    combined_audio.write(f.read())
+                os.remove(audio_file)  # Clean up individual audio files
         
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as temp_audio:
-            combined_audio.export(temp_audio.name, format="wav")
-            return temp_audio.name, transcript
+        return combined_audio.name, transcript
     
     except Exception as e:
         return None, f"An error occurred: {str(e)}"