Update app.py

app.py

@@ -15,7 +15,11 @@ def classify_audio(audio_file_path):
             headers=headers,
             files={"file": audio_file}
         )
-    return response.json()
+    # Check the response status and return JSON if successful, or an error message
+    if response.status_code == 200:
+        return response.json()
+    else:
+        return {"error": f"Failed to classify audio. Status code: {response.status_code}"}
 
 # Streamlit interface
 st.title("Audio Classifier")
@@ -24,8 +28,12 @@ st.title("Audio Classifier")
 audio_folder = "audio_files"
 
 # List the audio files in the folder
-audio_files = os.listdir(audio_folder)
-audio_file_options = [f for f in audio_files if f.endswith(('.mp3', '.wav'))]
+try:
+    audio_files = os.listdir(audio_folder)
+    audio_file_options = [f for f in audio_files if f.endswith(('.mp3', '.wav'))]
+except Exception as e:
+    st.error(f"Error accessing the audio files: {e}")
+    st.stop()
 
 # Dropdown to select an audio file
 selected_file = st.selectbox("Select an audio file:", audio_file_options)
@@ -40,6 +48,21 @@ if st.button("Classify"):
     
     # Get and display the classification results
     results = classify_audio(audio_file_path)
+    
+    # Check for errors
+    if "error" in results:
+        st.error(results["error"])
+    else:
+        st.write("Results:")
+        # Loop through each result in the list of results
+        for result in results:
+            # Check if result is a dictionary with 'label' and 'score' keys
+            if isinstance(result, dict) and 'label' in result and 'score' in result:
+                st.write(f"Label: {result['label']}, Confidence: {result['score']:.2f}")
+            else:
+                st.error(f"Unexpected result format: {result}")
+    # Get and display the classification results
+    results = classify_audio(audio_file_path)
     st.write("Results:")
     for result in results:
         st.write(f"Label: {result['label']}, Confidence: {result['score']:.2f}")