Update app.py

app.py

@@ -5,6 +5,7 @@ import librosa
 import soundfile as sf
 import os
 import zipfile
+from argparse import ArgumentParser
 
 
 # Function to download audio from YouTube and save it as a WAV file
@@ -166,17 +167,25 @@ def process_audio(url, audio_name):
     zip_file = zip_directory(dataset_path, audio_name)
     return zip_file, print(f"{zip_file} successfully processed")
 
+if __name__ == '__main__':
+    parser = ArgumentParser()
+    parser.add_argument("--colab", action="store_true", dest="share_enabled", default=False, help="Enable sharing")
+    
+    args = parser.parse_args()
 
 
-
-negra = gr.themes.Base(
+thme = gr.themes.Soft(
     primary_hue="blue",
     secondary_hue="blue")
 
 
-with gr.Blocks(theme=negra) as demo:
+with gr.Blocks(theme=thme) as demo:
     gr.Markdown("# <div style='text-align: center;'> RVC DATASET MAKER</div>")
     gr.Markdown(" <div style='text-align: center;'> only work for vocal only audio!</div>")
+    if config.iscolab:
+        pass
+    else:
+        gr.Markdown("<div style='text-align: center;'> please `❤` this spaces 🤗 if helpfull</dif>")
     with gr.Tabs():
         with gr.TabItem("make dataset"):
             with gr.Row():
@@ -189,16 +198,7 @@ with gr.Blocks(theme=negra) as demo:
             with gr.Row():
                 run_button = gr.Button("Slice 👾 Audio")
                 run_button.click(fn=process_audio, inputs=[url_input, audio_name_input], outputs=[result_output, result_process])
-        with gr.TabItem("note"):
-            with gr.Row():
-                gr.Markdown(
-                    f"""
-                    
-                   # colab version:
-                  [click here](https://colab.research.google.com/drive/1b7CpTBpMtd26KgZ8IhpEAkR2ehU11Amp?usp=sharing)
-                    """
-                )
-
+        
     with gr.Tabs():
         gr.Markdown(f"### <div style='text-align: center;'>made with ❤ by <a href='https://huggingface.co/Hev832'>Hev832</a></div>")