Update app.py

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 from clarifai_grpc.channel.clarifai_channel import ClarifaiChannel
 from clarifai_grpc.grpc.api import resources_pb2, service_pb2, service_pb2_grpc
 from clarifai_grpc.grpc.api.status import status_code_pb2
+import numpy as np
+
 
 # Streamlit page configuration
 st.set_page_config(page_title="Text to Speech", layout="centered")
@@ -24,6 +26,14 @@ APP_ID = 'audio-generation'
 MODEL_ID = 'speech-synthesis'
 MODEL_VERSION_ID = 'f2cead3a965f4c419a61a4a9b501095c'
 
+def pad_buffer(audio):
+    # Pad buffer to multiple of 2 bytes
+    buffer_size = len(audio)
+    element_size = np.dtype(np.int16).itemsize
+    if buffer_size % element_size != 0:
+        audio = audio + b'\0' * (element_size - (buffer_size % element_size))
+    return audio 
+
 # Streamlit UI
 st.title("Text to Speech Conversion")
 input_text = st.text_area("Enter text to convert to speech:", "I love your product very much")
@@ -57,6 +67,10 @@ if submit_button:
 
         # Assuming the output is in audio format
         audio_output = response.outputs[0].data.audio.content
-        st.audio(audio_output, format="audio/wav")
+
+        audio_data = np.frombuffer(pad_buffer(audio_output), dtype=np.int16)
+        audio_bytes = audio_data.tobytes()
+        
+        st.audio(audio_bytes, format="audio/wav")
     except Exception as e:
         st.error(f"An error occurred: {e}")