Update app.py

app.py

@@ -110,32 +110,38 @@ elif input_mode == "Audio Upload":
 
     if audio_input_mode == "Upload Audio File":
         uploaded_file = st.file_uploader("Upload audio file (WAV, MP3, M4A, MPEG)", type=["wav", "mp3", "m4a", "mpeg"])
+        
         if uploaded_file:
             st.audio(uploaded_file, format='audio/wav')
             audio = AudioSegment.from_file(uploaded_file)
-            wav_io = io.BytesIO()
-            audio.export(wav_io, format="wav")
-            wav_io.seek(0)
-
-            recognizer = sr.Recognizer()
-            with sr.AudioFile(wav_io) as source:
-                audio_data = recognizer.record(source)
-
-            try:
-                text = recognizer.recognize_google(audio_data)
-                st.markdown(f"**Transcribed Text:** _{text}_")
-                values = extract_details_from_text(text)
-                if all(v is not None for v in values):
-                    diagnosis = get_prediction(*values)
-                    st.success(f"🩺 **Predicted Diagnosis:** {diagnosis}")
-                else:
-                    st.warning("Could not extract complete information from audio.")
-            except Exception as e:
-                st.error(f"Audio processing error: {e}")
+            
+            if audio is not None and len(audio) > 0:
+                wav_io = io.BytesIO()
+                audio.export(wav_io, format="wav")
+                wav_io.seek(0)
+
+                recognizer = sr.Recognizer()
+                with sr.AudioFile(wav_io) as source:
+                    audio_data = recognizer.record(source)
+
+                try:
+                    text = recognizer.recognize_google(audio_data)
+                    st.markdown(f"**Transcribed Text:** _{text}_")
+                    values = extract_details_from_text(text)
+                    if all(v is not None for v in values):
+                        diagnosis = get_prediction(*values)
+                        st.success(f"🩺 **Predicted Diagnosis:** {diagnosis}")
+                    else:
+                        st.warning("Could not extract complete information from audio.")
+                except Exception as e:
+                    st.error(f"Audio processing error: {e}")
+            else:
+                st.error("Uploaded audio file is empty or not valid.")
 
     elif audio_input_mode == "Record Audio":
-        audio = audio_recorder("Click to record", "Recording...")
-        if len(audio) > 0:
+        audio = audiorecorder("Click to record", "Recording...")
+
+        if audio is not None and len(audio) > 0:  # Check if audio is not None and has length
             st.audio(audio.tobytes(), format="audio/wav")
             wav_io = io.BytesIO(audio.tobytes())
 
@@ -153,4 +159,6 @@ elif input_mode == "Audio Upload":
                 else:
                     st.warning("Could not extract complete information from recorded audio.")
             except Exception as e:
-                st.error(f"Recording processing error: {e}")
+                st.error(f"Recording processing error: {e}")
+        else:
+            st.error("No audio recorded or audio is empty.")