Update app.py

app.py

@@ -36,9 +36,16 @@ for message in st.session_state.messages:
 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:
-        # Convert audio (from audiorecorder) to AudioSegment
-        audio_segment = AudioSegment.from_file(io.BytesIO(audio), format="wav")
-        prompt = inference(audio_segment)
+        # Ensure that the `audio` is a raw bytes object
+        if isinstance(audio, bytes):
+            # Convert the raw byte data to an AudioSegment instance
+            audio_segment = AudioSegment.from_file(io.BytesIO(audio), format="wav")
+        else:
+            st.error("The audio data format is not correct.")
+            audio_segment = None
+        
+        if audio_segment:
+            prompt = inference(audio_segment)
 
     # Display user message in chat message container
     st.chat_message("user").markdown(prompt)