Update app.py

app.py

@@ -84,23 +84,24 @@ def main():
 
     audio_data = mic_recorder(start_prompt="▶️", stop_prompt="🛑", key='recorder')
 
-    if audio_data:
+    if audio_data and 'bytes' in audio_data:
         st.audio(audio_data['bytes'])
 
-        sample_width = audio_data["sample_width"]
-        sample_rate = audio_data["sample_rate"]
+        sample_width = audio_data.get("sample_width")
+        sample_rate = audio_data.get("sample_rate")
         num_channels = 1
-    
-        with wave.open(temp_audio_file_path, 'w') as wave_file:
-            wave_file.setnchannels(num_channels)
-            wave_file.setsampwidth(sample_width)
-            wave_file.setframerate(sample_rate)
-            wave_file.writeframes(audio_data["bytes"])
-
-        audio_text = recognize_speech(audio_data["bytes"])
-        formatted_prompt = format_prompt(audio_text, st.session_state.history)
-        response, audio_file = generate(formatted_prompt, st.session_state.history)
-        display_recognition_result(audio_text, response, audio_file)
+
+        if sample_width and sample_rate:
+            with wave.open(temp_audio_file_path, 'w') as wave_file:
+                wave_file.setnchannels(num_channels)
+                wave_file.setsampwidth(sample_width)
+                wave_file.setframerate(sample_rate)
+                wave_file.writeframes(audio_data['bytes'])
+
+            audio_text = recognize_speech(audio_data['bytes'])
+            formatted_prompt = format_prompt(audio_text, st.session_state.history)
+            response, audio_file = generate(formatted_prompt, st.session_state.history)
+            display_recognition_result(audio_text, response, audio_file)
 
 if __name__ == "__main__":
     main()