File size: 4,776 Bytes
01e655b f83b1b7 8c22399 f83b1b7 8c22399 374f426 8c22399 01e655b 8c22399 01e655b 8c22399 01e655b 374f426 bf13828 8c22399 01e655b 02e90e4 8c22399 02e90e4 ebc4336 c788835 8c22399 02e90e4 da8d589 374f426 01e655b bf13828 01e655b 8c22399 01e655b 8c22399 01e655b 8c22399 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 |
import os
import logging
logging.basicConfig(
level=os.getenv("LOG_LEVEL", "INFO"),
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
)
from launch import (
get_and_update_env,
setup_api_args,
setup_model_args,
process_api_args,
process_model_args,
app_description,
app_title,
app_version,
)
from modules.webui import webui_config
from modules.webui.app import webui_init, create_interface
import argparse
from modules.gradio_dcls_fix import dcls_patch
dcls_patch()
def setup_webui_args(parser: argparse.ArgumentParser):
parser.add_argument("--server_name", type=str, help="server name")
parser.add_argument("--server_port", type=int, help="server port")
parser.add_argument(
"--share", action="store_true", help="share the gradio interface"
)
parser.add_argument("--debug", action="store_true", help="enable debug mode")
parser.add_argument("--auth", type=str, help="username:password for authentication")
parser.add_argument(
"--tts_max_len",
type=int,
help="Max length of text for TTS",
)
parser.add_argument(
"--ssml_max_len",
type=int,
help="Max length of text for SSML",
)
parser.add_argument(
"--max_batch_size",
type=int,
help="Max batch size for TTS",
)
# webui_Experimental
parser.add_argument(
"--webui_experimental",
action="store_true",
help="Enable webui_experimental features",
)
parser.add_argument(
"--language",
type=str,
help="Set the default language for the webui",
)
parser.add_argument(
"--api",
action="store_true",
help="use api=True to launch the API together with the webui (run launch.py for only API server)",
)
def process_webui_args(args):
server_name = get_and_update_env(args, "server_name", "0.0.0.0", str)
server_port = get_and_update_env(args, "server_port", 7860, int)
share = get_and_update_env(args, "share", False, bool)
debug = get_and_update_env(args, "debug", False, bool)
auth = get_and_update_env(args, "auth", None, str)
language = get_and_update_env(args, "language", "zh-CN", str)
api = get_and_update_env(args, "api", False, bool)
webui_config.experimental = get_and_update_env(
args, "webui_experimental", False, bool
)
webui_config.tts_max = get_and_update_env(args, "tts_max_len", 1000, int)
webui_config.ssml_max = get_and_update_env(args, "ssml_max_len", 5000, int)
webui_config.max_batch_size = get_and_update_env(args, "max_batch_size", 8, int)
webui_config.experimental = get_and_update_env(
args, "webui_experimental", False, bool
)
webui_config.tts_max = get_and_update_env(args, "tts_max_len", 1000, int)
webui_config.ssml_max = get_and_update_env(args, "ssml_max_len", 5000, int)
webui_config.max_batch_size = get_and_update_env(args, "max_batch_size", 8, int)
webui_init()
demo = create_interface()
if auth:
auth = tuple(auth.split(":"))
app, local_url, share_url = demo.queue().launch(
server_name=server_name,
server_port=server_port,
share=share,
debug=debug,
auth=auth,
show_api=False,
prevent_thread_lock=True,
app_kwargs={
"title": app_title,
"description": app_description,
"version": app_version,
# "redoc_url": (
# None
# if api is False
# else None if config.runtime_env_vars.no_docs else "/redoc"
# ),
# "docs_url": (
# None
# if api is False
# else None if config.runtime_env_vars.no_docs else "/docs"
# ),
"docs_url": "/docs",
},
)
# gradio uses a very open CORS policy via app.user_middleware, which makes it possible for
# an attacker to trick the user into opening a malicious HTML page, which makes a request to the
# running web ui and do whatever the attacker wants, including installing an extension and
# running its code. We disable this here. Suggested by RyotaK.
app.user_middleware = [
x for x in app.user_middleware if x.cls.__name__ != "CustomCORSMiddleware"
]
if api:
process_api_args(args, app)
if __name__ == "__main__":
import dotenv
dotenv.load_dotenv(
dotenv_path=os.getenv("ENV_FILE", ".env.webui"),
)
parser = argparse.ArgumentParser(description="Gradio App")
setup_webui_args(parser)
setup_model_args(parser)
setup_api_args(parser)
args = parser.parse_args()
process_model_args(args)
process_webui_args(args)
|