Add server-name and no-autolaunch options

app.py

@@ -244,6 +244,18 @@ if __name__ == "__main__":
     parser.add_argument(
         "--share", action="store_true", help="Share this app publicly", default=False
     )
+    parser.add_argument(
+        "--server-name",
+        type=str,
+        default=None,
+        help="Server name for Gradio app",
+    )
+    parser.add_argument(
+        "--no-autolaunch",
+        action="store_true",
+        default=False,
+        help="Do not launch app automatically",
+    )
     args = parser.parse_args()
     model_dir = args.dir
 
@@ -435,4 +447,6 @@ if __name__ == "__main__":
             outputs=[style, ref_audio_path],
         )
 
-    app.launch(inbrowser=True, share=args.share)
+    app.launch(
+        inbrowser=not args.no_autolaunch, share=args.share, server_name=args.server_name
+    )

webui_dataset.py

@@ -1,3 +1,4 @@
+import argparse
 import os
 
 import gradio as gr
@@ -192,4 +193,19 @@ with gr.Blocks(theme="NoCrypt/miku") as app:
         outputs=[result2],
     )
 
-app.launch(inbrowser=True)
+parser = argparse.ArgumentParser()
+parser.add_argument(
+    "--server-name",
+    type=str,
+    default=None,
+    help="Server name for Gradio app",
+)
+parser.add_argument(
+    "--no-autolaunch",
+    action="store_true",
+    default=False,
+    help="Do not launch app automatically",
+)
+args = parser.parse_args()
+
+app.launch(inbrowser=not args.no_autolaunch, server_name=args.server_name)

webui_merge.py

@@ -1,3 +1,4 @@
+import argparse
 import json
 import os
 import sys
@@ -395,5 +396,19 @@ with gr.Blocks(theme="NoCrypt/miku") as app:
         outputs=[audio_output],
     )
 
+parser = argparse.ArgumentParser()
+parser.add_argument(
+    "--server-name",
+    type=str,
+    default=None,
+    help="Server name for Gradio app",
+)
+parser.add_argument(
+    "--no-autolaunch",
+    action="store_true",
+    default=False,
+    help="Do not launch app automatically",
+)
+args = parser.parse_args()
 
-app.launch(inbrowser=True)
+app.launch(inbrowser=not args.no_autolaunch, server_name=args.server_name)

webui_style_vectors.py

@@ -1,3 +1,4 @@
+import argparse
 import json
 import os
 import shutil
@@ -443,4 +444,20 @@ with gr.Blocks(theme="NoCrypt/miku") as app:
                 inputs=[model_name, audio_files_text, style_names_text],
                 outputs=[info2],
             )
-    app.launch(inbrowser=True)
+
+parser = argparse.ArgumentParser()
+parser.add_argument(
+    "--server-name",
+    type=str,
+    default=None,
+    help="Server name for Gradio app",
+)
+parser.add_argument(
+    "--no-autolaunch",
+    action="store_true",
+    default=False,
+    help="Do not launch app automatically",
+)
+args = parser.parse_args()
+
+app.launch(inbrowser=not args.no_autolaunch, server_name=args.server_name)

webui_train.py

@@ -1,6 +1,8 @@
+import argparse
 import json
 import os
 import shutil
+from datetime import datetime
 from multiprocessing import cpu_count
 
 import gradio as gr
@@ -8,8 +10,6 @@ import yaml
 
 from common.log import logger
 from common.subprocess_utils import run_script_with_log, second_elem_of
-from datetime import datetime
-
 
 logger_handler = None
 
@@ -496,4 +496,19 @@ if __name__ == "__main__":
             second_elem_of(train), inputs=[model_name], outputs=[info_train]
         )
 
-    app.launch(inbrowser=True)
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        "--server-name",
+        type=str,
+        default=None,
+        help="Server name for Gradio app",
+    )
+    parser.add_argument(
+        "--no-autolaunch",
+        action="store_true",
+        default=False,
+        help="Do not launch app automatically",
+    )
+    args = parser.parse_args()
+
+    app.launch(inbrowser=args.no_autolaunch, server_name=args.server_name)