Update app.py

app.py

@@ -3,6 +3,8 @@ import yt_dlp
 import numpy as np
 import librosa
 import soundfile as sf
+import os
+import zipfile
 
 # Function to download audio from YouTube and save it as a WAV file
 def download_youtube_audio(url, audio_name):
@@ -134,28 +136,39 @@ class Slicer:
 # Function to slice and save audio chunks
 def slice_audio(file_path, audio_name):
     audio, sr = librosa.load(file_path, sr=None, mono=False)
+    os.makedirs(f'dataset/{audio_name}', exist_ok=True)
     slicer = Slicer(sr=sr, threshold=-40, min_length=5000, min_interval=500, hop_size=10, max_sil_kept=500)
     chunks = slicer.slice(audio)
     for i, chunk in enumerate(chunks):
         if len(chunk.shape) > 1:
             chunk = chunk.T
         sf.write(f'dataset/{audio_name}/split_{i}.wav', chunk, sr)
-    return f"Audio sliced and saved in dataset/{audio_name}/"
+    return f"dataset/{audio_name}/"
+
+# Function to zip the dataset directory
+def zip_directory(directory_path):
+    zip_file = f"{directory_path.rstrip('/').rstrip('\\')}.zip"
+    with zipfile.ZipFile(zip_file, 'w', zipfile.ZIP_DEFLATED) as zipf:
+        for root, dirs, files in os.walk(directory_path):
+            for file in files:
+                file_path = os.path.join(root, file)
+                arcname = os.path.relpath(file_path, start=directory_path)
+                zipf.write(file_path, arcname)
+    return zip_file
 
 # Gradio interface
 def process_audio(url, audio_name):
     file_path = download_youtube_audio(url, audio_name)
-    result = slice_audio(file_path, audio_name)
-    return result
+    dataset_path = slice_audio(file_path, audio_name)
+    zip_file = zip_directory(dataset_path)
+    return zip_file
 
 with gr.Blocks() as demo:
-    gr.Markdown("RVC DATASET MAKER 2.0")
-    with gr.Tabs():
-        with gr.Row():
-            url_input = gr.Textbox(label="YouTube URL")
-            audio_name_input = gr.Textbox(label="Audio Name")
-            result_output = gr.Textbox(label="Output Directory")
-            run_button = gr.Button("Download and Slice Audio")
-            run_button.click(fn=process_audio, inputs=[url_input, audio_name_input], outputs=result_output)
+    url_input = gr.Textbox(label="YouTube URL")
+    audio_name_input = gr.Textbox(label="Audio Name")
+    result_output = gr.File(label="Download Sliced Audio Zip")
+
+    run_button = gr.Button("Download and Slice Audio")
+    run_button.click(fn=process_audio, inputs=[url_input, audio_name_input], outputs=result_output)
 
 demo.launch()