Update app.py

app.py

@@ -9,13 +9,21 @@ from scipy.io.wavfile import write
 tts = None
 audio_base64 = ""
 
-
 def audio_to_base64(sample_rate, audio_data):
+    # Create a buffer to hold the WAV file
     buffer = io.BytesIO()
-    write(buffer, sample_rate, audio_data)
-    base64_audio = base64.b64encode(buffer.getvalue()).decode('utf-8')
-    buffer.close()
-    return base64_audio
+    with wave.open(buffer, 'w') as wav_file:
+        # Set the parameters for the WAV file
+        wav_file.setnchannels(1)
+        wav_file.setsampwidth(2) # 16-bit
+        wav_file.setframerate(sample_rate)
+        wav_file.writeframes(audio_data.tobytes())
+
+    # Get the content of the buffer and encode it as Base64
+    wav_bytes = buffer.getvalue()
+    base64_str = base64.b64encode(wav_bytes).decode('utf-8')
+
+    return base64_str
     
 def main():
     logging.basicConfig(level=logging.INFO)