Update app.py

app.py

@@ -21,7 +21,7 @@ import shutil
 import urllib.request
 import gdown
 import subprocess
-
+from argparse import ArgumentParser
 main_dir = Path().resolve()
 print(main_dir)
 
@@ -198,8 +198,20 @@ def download_online_model(url, dir_name):
 
 
 
+
+if __name__ == '__main__':
+    parser = ArgumentParser(description='Generate a AI song in the song_output/id directory.', add_help=True)
+    parser.add_argument("--share", action="store_true", dest="share_enabled", default=False, help="Enable sharing")
+    parser.add_argument("--listen", action="store_true", default=False, help="Make the UI reachable from your local network.")
+    parser.add_argument('--listen-host', type=str, help='The hostname that the server will use.')
+    parser.add_argument('--listen-port', type=int, help='The listening port that the server will use.')
+    args = parser.parse_args()
+
+
+
+
 # Gradio Blocks Interface with Tabs
-with gr.Blocks(title="Hex RVC") as app:
+with gr.Blocks(title="Hex RVC", theme="gradio/soft") as app:
     gr.Markdown("# Hex RVC")
     
     with gr.Tab("Inference"):
@@ -303,4 +315,8 @@ with gr.Blocks(title="Hex RVC") as app:
 
 
 # Launch the Gradio app
-app.launch(share=True, debug=True)
+app.launch(
+    share=args.share_enabled,
+    server_name=None if not args.listen else (args.listen_host or '0.0.0.0'),
+    server_port=args.listen_port,
+)