Update app.py

app.py

@@ -1,21 +1,28 @@
 import streamlit as st
 import tempfile
+import os
+from pydub import AudioSegment
 from utils.noise_removal import remove_noise
 from utils.vad_segmentation import vad_segmentation
 from utils.speaker_diarization import diarize_speakers
 from utils.noise_classification import classify_noise
 
 st.set_page_config(page_title="Audio Analyzer", layout="wide")
-st.title(" Audio Analysis Pipeline")
+st.title("Audio Analysis Pipeline")
 
-uploaded_file = st.file_uploader("Upload an audio file", type=["wav", "mp3"])
+uploaded_file = st.file_uploader("Upload an audio file", type=["wav", "mp3", "m4a", "mp4a"])
+
+def convert_to_wav(uploaded_file):
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as out_wav:
+        audio = AudioSegment.from_file(uploaded_file, format=uploaded_file.name.split('.')[-1])
+        audio.export(out_wav.name, format="wav")
+        return out_wav.name
 
 if uploaded_file:
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp:
-        tmp.write(uploaded_file.read())
-        tmp_path = tmp.name
+    st.audio(uploaded_file, format="audio/wav")
 
-    st.audio(tmp_path, format='audio/wav')
+    with st.spinner(" Converting to WAV..."):
+        tmp_path = convert_to_wav(uploaded_file)
 
     try:
         st.subheader("1️⃣ Noise Removal")