Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import os
-import tarfile
 import allin1
+import zipfile
 
 from pathlib import Path
 
@@ -49,6 +49,16 @@ HEADER = """
 
 CACHE_EXAMPLES = os.getenv('CACHE_EXAMPLES', '1') == '1'
 
+def compress_files(folder_path, dissector_file):
+    """Compresses files in the specified folder into a .zip file"""
+    zip_path = folder_path + ".zip"
+    with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
+        for root, _, files in os.walk(folder_path):
+            for file in files:
+                zipf.write(os.path.join(root, file), arcname=file)
+        zipf.write(dissector_file, arcname=os.path.basename(dissector_file))
+    return zip_path
+
 def analyze(path):
   path = Path(path)
   result = allin1.analyze(
@@ -64,13 +74,9 @@ def analyze(path):
   sonif_path = Path(f'./sonif/{path.stem}.sonif{path.suffix}').resolve().as_posix()
       
   dissector_file = generate_dissector_data(path.stem, result)  
+  compressed_file = compress_files(f"demix/htdemucs/{path.stem}", dissector_file)
 
-  bass_file_path = f"demix/htdemucs/{path.stem}/bass.wav"
-  drums_file_path = f"demix/htdemucs/{path.stem}/drums.wav"
-  other_file_path = f"demix/htdemucs/{path.stem}/other.wav"
-  vocals_file_path = f"demix/htdemucs/{path.stem}/vocals.wav"
-    
-  return result.bpm, fig, sonif_path, bass_file_path, drums_file_path, other_file_path, vocals_file_path, dissector_file
+  return result.bpm, fig, sonif_path, compressed_file
 
 
 with gr.Blocks() as demo:
@@ -94,41 +100,28 @@ with gr.Blocks() as demo:
       show_download_button=False,
       scale=9,
     )
-    output_bass = gr.File(
-      label="Bass File",
-      type="file",
-    )
-    output_drums = gr.File(
-      label="Drums File",
-      type="file",
-    )
-    output_other = gr.File(
-      label="Other File",
-      type="file",
-    )
-    output_vocals = gr.File(
-      label="Vocals File",
-      type="file",
-    )
-    output_dissector = gr.File(
-      label="Dissector File",
-      type="file",
+    output_compressed = gr.File(
+        label="Compressed Files",
+        type="file",
     )
   gr.Examples(
     examples=[
-      './assets/kult.mp3',
+        './assets/kult.mp3',
+        # './assets/krematorii.mp3',
+      # './assets/NewJeans - Super Shy.mp3',
+      # './assets/Bruno Mars - 24k Magic.mp3'
     ],
     inputs=input_audio_path,
-    outputs=[output_bpm, output_viz, output_sonif, output_bass, output_drums, output_other, output_vocals, output_dissector],
+    outputs=[output_bpm, output_viz, output_sonif, output_compressed],
     fn=analyze,
     cache_examples=CACHE_EXAMPLES,
   )
   button.click(
     fn=analyze,
     inputs=input_audio_path,
-    outputs=[output_bpm, output_viz, output_sonif, output_bass, output_drums, output_other, output_vocals, output_dissector],
+    outputs=[output_bpm, output_viz, output_sonif, output_compressed],
     api_name='analyze',
   )
 
 if __name__ == '__main__':
-    demo.launch()
+  demo.launch()