File size: 5,063 Bytes
01e655b 1df74c6 f83b1b7 627d3d7 8c22399 f83b1b7 374f426 a2483ba 374f426 8c22399 01e655b 8c22399 01e655b 8c22399 01e655b 374f426 bf13828 8c22399 01e655b 02e90e4 8c22399 02e90e4 ebc4336 c788835 8c22399 02e90e4 da8d589 374f426 01e655b 627d3d7 bf13828 01e655b 8c22399 01e655b 627d3d7 1df74c6 8c22399 a2483ba 8c22399 627d3d7 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 152 153 154 155 156 157 158 |
import os
import sys
import logging
from modules.api.api_setup import (
process_api_args,
process_model_args,
setup_api_args,
setup_model_args,
)
from modules.ffmpeg_env import setup_ffmpeg_path
from modules.utils.env import get_and_update_env
from modules.api.app_config import app_description, app_title, app_version
from modules.utils.torch_opt import configure_torch_optimizations
setup_ffmpeg_path()
logging.basicConfig(
level=os.getenv("LOG_LEVEL", "INFO"),
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
)
from modules.webui import webui_config
from modules import 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)
configure_torch_optimizations()
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,
inbrowser=sys.platform == "win32",
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"
),
},
)
# 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)
demo.block_thread()
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)
|