Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import numpy as np
+import zipfile
 from pydub import AudioSegment
 import streamlit as st
 from io import BytesIO
@@ -108,24 +109,35 @@ def main():
     # Process and save files
     if st.button("Process Audio"):
         if uploaded_files:
-            noisy_audios = {}
-            for uploaded_file in uploaded_files:
-                st.write(f"Processing {uploaded_file.name}...")
-                audio_data = BytesIO(uploaded_file.read())
-
-                noisy_audio = add_noise(audio_data, noise_type, noise_level, start_time, end_time)
-                if noisy_audio:
-                    # Save to in-memory buffer
-                    buffer = BytesIO()
-                    noisy_audio.export(buffer, format="wav")
-                    buffer.seek(0)
-                    noisy_audios[uploaded_file.name] = buffer
-                    st.success(f"Added {noise_type} noise to {uploaded_file.name}.")
+            zip_buffer = BytesIO()
+            with zipfile.ZipFile(zip_buffer, "w") as zf:
+                for uploaded_file in uploaded_files:
+                    st.write(f"Processing {uploaded_file.name}...")
+                    audio_data = BytesIO(uploaded_file.read())
+                    
+                    noisy_audio = add_noise(audio_data, noise_type, noise_level, start_time, end_time)
+                    if noisy_audio:
+                        # Rename file based on noise type
+                        base_name, ext = os.path.splitext(uploaded_file.name)
+                        output_name = f"{noise_type}_{base_name}.wav"
+
+                        # Save to in-memory buffer
+                        buffer = BytesIO()
+                        noisy_audio.export(buffer, format="wav")
+                        buffer.seek(0)
+
+                        # Add to zip archive
+                        zf.writestr(output_name, buffer.read())
+                        st.success(f"Added {noise_type} noise to {uploaded_file.name}.")
             
-            if noisy_audios:
-                st.write("### Download Processed Files")
-                for name, buffer in noisy_audios.items():
-                    st.download_button(label=f"Download {name}", data=buffer, file_name=f"noisy_{name}", mime="audio/wav")
+            zip_buffer.seek(0)
+            st.write("### Download All Processed Files")
+            st.download_button(
+                label="Download All as ZIP",
+                data=zip_buffer,
+                file_name="processed_audio_files.zip",
+                mime="application/zip"
+            )
         else:
             st.warning("Please upload at least one audio file.")