Update app.py

app.py

@@ -8,11 +8,17 @@ from huggingface_hub import snapshot_download, login
 import logging
 import os
 import spaces
+import warnings
 
 # Set up logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
 
+# Suppress specific warnings
+warnings.filterwarnings("ignore", category=UserWarning, message="Trying to convert audio automatically from float32 to 16-bit int format.")
+warnings.filterwarnings("ignore", category=RuntimeWarning, message="invalid value encountered in divide")
+warnings.filterwarnings("ignore", category=RuntimeWarning, message="invalid value encountered in cast")
+
 def get_device():
     if torch.cuda.is_available():
         return torch.device("cuda")
@@ -119,6 +125,8 @@ def text_to_speech(text, voice):
         raise
 
 def mel_to_audio(mel):
+    # Implement proper mel to audio conversion here
+    # For now, we'll return a placeholder audio
     return np.zeros(24000, dtype=np.float32)  # Placeholder: 1 second of silence
 
 @spaces.GPU()
@@ -140,6 +148,11 @@ def render_podcast(api_key, script, voice1, voice2, num_hosts):
             return (24000, np.zeros(24000, dtype=np.float32))
 
         podcast_audio = np.concatenate(audio_segments)
+        
+        # Ensure the audio is in the correct format for Gradio
+        podcast_audio = np.clip(podcast_audio, -1, 1)
+        podcast_audio = (podcast_audio * 32767).astype(np.int16)
+        
         return (24000, podcast_audio)
     except Exception as e:
         logger.error(f"Error rendering podcast: {str(e)}")