Update app.py

app.py

@@ -6,12 +6,12 @@ from whispercpp import Whisper
 # Download whisper.cpp
 w = Whisper('tiny')
 
-def inference(audio):
-    # Save audio to a file:
-    with NamedTemporaryFile(suffix=".mp3") as temp:
-        with open(f"{temp.name}", "wb") as f:
-            f.write(audio.tobytes())
-        result = w.transcribe(f"{temp.name}")
+def inference(audio_data):
+    # Save the raw audio data to a temporary file
+    with NamedTemporaryFile(suffix=".wav", delete=False) as temp:
+        temp.write(audio_data)  # write the raw audio bytes
+        temp.close()  # Ensure the file is written and closed before passing it to Whisper
+        result = w.transcribe(temp.name)
         text = w.extract_text(result)
     return text[0]
 
@@ -32,7 +32,7 @@ for message in st.session_state.messages:
 # React to user input
 if (prompt := st.chat_input("Your message")) or len(audio):
     # If it's coming from the audio recorder transcribe the message with whisper.cpp
-    if len(audio)>0:
+    if len(audio) > 0:
         prompt = inference(audio)
 
     # Display user message in chat message container
@@ -45,4 +45,4 @@ if (prompt := st.chat_input("Your message")) or len(audio):
     with st.chat_message("assistant"):
         st.markdown(response)
     # Add assistant response to chat history
-    st.session_state.messages.append({"role": "assistant", "content": response})
+    st.session_state.messages.append({"role": "assistant", "content": response})