Update app.py

app.py

@@ -46,6 +46,14 @@ HEADER = """
 
 CACHE_EXAMPLES = os.getenv('CACHE_EXAMPLES', '1') == '1'
 
+def compress_files(folder_path):
+    """Compresses files in the specified folder into a .tar.gz"""
+    tar_path = folder_path + ".tar.gz"
+    with tarfile.open(tar_path, "w:gz") as tar:
+        for root, _, files in os.walk(folder_path):
+            for file in files:
+                tar.add(os.path.join(root, file), arcname=file)
+    return tar_path
 
 def analyze(path):
   path = Path(path)
@@ -61,7 +69,9 @@ def analyze(path):
   allin1.sonify(result, out_dir='./sonif')
   sonif_path = Path(f'./sonif/{path.stem}.sonif{path.suffix}').resolve().as_posix()
 
-  return result.bpm, fig, sonif_path
+  compressed_file = compress_files(f"demix/htdemucs/{path.stem}")
+
+  return result.bpm, fig, sonif_path, compressed_file
 
 
 with gr.Blocks() as demo:
@@ -85,13 +95,19 @@ with gr.Blocks() as demo:
       show_download_button=False,
       scale=9,
     )
+    output_compressed = gr.File(
+        label="Compressed Files",
+        type="file",
+        format="tar.gz",
+        show_download_button=True,
+    )
   gr.Examples(
     examples=[
       './assets/NewJeans - Super Shy.mp3',
       './assets/Bruno Mars - 24k Magic.mp3'
     ],
     inputs=input_audio_path,
-    outputs=[output_bpm, output_viz, output_sonif],
+    outputs=[output_bpm, output_viz, output_sonif, output_compressed],
     fn=analyze,
     cache_examples=CACHE_EXAMPLES,
   )