feat: zip with zipfile module

app.py

@@ -12,6 +12,7 @@ from demucs.separate import main as demucs
 from pyannote.audio import Pipeline
 from json import dumps, loads
 import shutil
+import zipfile
 
 accelerator = Accelerator()
 device = accelerator.device
@@ -68,7 +69,7 @@ def extract_audio_post(task_id: str) -> str:
     return spec
 
 
-@zero()
+@zero(duration=60 * 2)
 def extract_vocals(task_id: str) -> str:
     audio = path.join("task", task_id, "extracted_48k.wav")
     if not path.exists(audio):
@@ -199,7 +200,10 @@ def generate_clips(task_id: str, speaker: str) -> Tuple[str, str]:
 
     segments_zip = path.join("task", task_id, f"{speaker}.zip")
     if not path.exists(segments_zip):
-        os.system(f"zip -r {segments_zip} {segments}")
+        with zipfile.ZipFile(segments_zip, "w") as zipf:
+            files = [f for f in os.listdir(segments) if f.endswith(".wav")]
+            for file in files:
+                zipf.write(path.join(segments, file), file)
 
     return mp4, segments_zip