vpavlenko commited on
Commit
00b2148
·
1 Parent(s): b12171d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -49,14 +49,15 @@ HEADER = """
49
 
50
  CACHE_EXAMPLES = os.getenv('CACHE_EXAMPLES', '1') == '1'
51
 
52
- def compress_files(folder_path, dissector_file):
53
  """Compresses files in the specified folder into a .zip file"""
54
  zip_path = folder_path + ".zip"
55
  with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
56
  for root, _, files in os.walk(folder_path):
57
  for file in files:
58
  zipf.write(os.path.join(root, file), arcname=file)
59
- zipf.write(dissector_file, arcname=os.path.basename(dissector_file))
 
60
  return zip_path
61
 
62
  def analyze(path):
@@ -74,7 +75,7 @@ def analyze(path):
74
  sonif_path = Path(f'./sonif/{path.stem}.sonif{path.suffix}').resolve().as_posix()
75
 
76
  dissector_file = generate_dissector_data(path.stem, result)
77
- compressed_file = compress_files(f"demix/htdemucs/{path.stem}", dissector_file)
78
 
79
  return result.bpm, fig, sonif_path, compressed_file
80
 
 
49
 
50
  CACHE_EXAMPLES = os.getenv('CACHE_EXAMPLES', '1') == '1'
51
 
52
+ def compress_files(folder_path, dissector_file, original_audio):
53
  """Compresses files in the specified folder into a .zip file"""
54
  zip_path = folder_path + ".zip"
55
  with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
56
  for root, _, files in os.walk(folder_path):
57
  for file in files:
58
  zipf.write(os.path.join(root, file), arcname=file)
59
+ zipf.write(dissector_file, arcname='dissector.json') # Fixed the name
60
+ zipf.write(original_audio, arcname='mixdown.mp3') # Added original audio
61
  return zip_path
62
 
63
  def analyze(path):
 
75
  sonif_path = Path(f'./sonif/{path.stem}.sonif{path.suffix}').resolve().as_posix()
76
 
77
  dissector_file = generate_dissector_data(path.stem, result)
78
+ compressed_file = compress_files(f"demix/htdemucs/{path.stem}", dissector_file, path.as_posix())
79
 
80
  return result.bpm, fig, sonif_path, compressed_file
81