Update app.py

app.py

@@ -17,7 +17,14 @@ def record_audio(duration=3, sr=44100, channels=1):
     print("Recording stopped.")
     return audio, sr
 
-def audio_to_spectrogram(audio, sr): 
+def audio_to_spectrogram(audio_file, sr): 
+        if audio_file.endswith('.mp3'):
+        with tempfile.NamedTemporaryFile(suffix='.wav') as temp_wav:
+            audio = AudioSegment.from_mp3(audio_file)
+            audio.export(temp_wav.name, format='wav')
+            y, sr = librosa.load(temp_wav.name, sr=None)
+    else:
+        y, sr = librosa.load(audio_file, sr=None)
     S = librosa.feature.melspectrogram(y=audio[:, 0], sr=sr, n_mels=128, fmax=8000)
     S_dB = librosa.power_to_db(S, ref=np.max)
     fig, ax = plt.subplots()