Update tts.py

tts.py

@@ -23,12 +23,18 @@ def synthesize_speech(text):
             return None
 
         result = pipe(text)
-        waveform = result["waveform"]
+        logging.debug(f"Pipeline result: {result}")
 
-        # Save waveform to an audio file
-        audio_path = "output.wav"
-        sf.write(audio_path, waveform, 16000)  # Write the waveform using soundfile
-        return audio_path
+        # Check if the output contains 'waveform' key
+        if 'waveform' in result:
+            waveform = result['waveform']
+            # Save waveform to an audio file
+            audio_path = "output.wav"
+            sf.write(audio_path, waveform, 16000)  # Write the waveform using soundfile
+            return audio_path
+        else:
+            logging.error(f"Unexpected pipeline output: {result}")
+            return None
     except Exception as e:
         logging.error(f"Error during speech synthesis: {e}")
         return None