Update app.py

app.py

@@ -5,6 +5,8 @@ import allin1
 
 from pathlib import Path
 
+from dissector import generate_dissector_data
+
 HEADER = """
 <header style="text-align: center;">
   <h1>
@@ -47,13 +49,14 @@ HEADER = """
 
 CACHE_EXAMPLES = os.getenv('CACHE_EXAMPLES', '1') == '1'
 
-def compress_files(folder_path):
+def compress_files(folder_path, dissector_file):
     """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)
+        tar.add(dissector_file)
     return tar_path
 
 def analyze(path):
@@ -70,7 +73,8 @@ def analyze(path):
   allin1.sonify(result, out_dir='./sonif')
   sonif_path = Path(f'./sonif/{path.stem}.sonif{path.suffix}').resolve().as_posix()
 
-  compressed_file = compress_files(f"demix/htdemucs/{path.stem}")
+  dissector_file = generate_dissector_data(path.stem, result)  
+  compressed_file = compress_files(f"demix/htdemucs/{path.stem}", dissector_file)
 
   return result.bpm, fig, sonif_path, compressed_file
 
@@ -113,7 +117,7 @@ with gr.Blocks() as demo:
   button.click(
     fn=analyze,
     inputs=input_audio_path,
-    outputs=[output_bpm, output_viz, output_sonif],
+    outputs=[output_bpm, output_viz, output_sonif, output_compressed],
     api_name='analyze',
   )