Update app.py

app.py

@@ -36,16 +36,19 @@ 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:
-        # Ensure that the `audio` is a raw bytes object
+        # Check if audio is a bytes object (raw audio data)
         if isinstance(audio, bytes):
-            # Convert the raw byte data to an AudioSegment instance
-            audio_segment = AudioSegment.from_file(io.BytesIO(audio), format="wav")
+            # Try to load audio from bytes using AudioSegment
+            try:
+                # Convert the raw byte data to an AudioSegment instance
+                audio_segment = AudioSegment.from_file(io.BytesIO(audio), format="wav")
+                prompt = inference(audio_segment)
+            except Exception as e:
+                st.error(f"Error processing audio: {e}")
+                prompt = "Sorry, there was an error processing your audio."
         else:
-            st.error("The audio data format is not correct.")
-            audio_segment = None
-        
-        if audio_segment:
-            prompt = inference(audio_segment)
+            st.error("The audio data is not in the expected format.")
+            prompt = "Sorry, the audio format is not correct."
 
     # Display user message in chat message container
     st.chat_message("user").markdown(prompt)