diff --git a/app.py b/app.py index 4102640..7072f36 100644 --- a/app.py +++ b/app.py @@ -147,6 +147,9 @@ def setup_parser() -> argparse.ArgumentParser: parser = argparse.ArgumentParser( description="A fast and powerful image browser for Stable Diffusion webui." ) + parser.add_argument( + "--host", type=str, default="127.0.0.1", help="The host to use" + ) parser.add_argument( "--port", type=int, help="The port to use", default=default_port ) @@ -186,7 +189,7 @@ def setup_parser() -> argparse.ArgumentParser: return parser -def launch_app(port: int = default_port, *args, **kwargs: dict) -> None: +def launch_app(host: str = default_host, port: int = default_port, *args, **kwargs: dict) -> None: """ Launches the application on the specified port. @@ -196,10 +199,10 @@ def launch_app(port: int = default_port, *args, **kwargs: dict) -> None: """ app_utils = AppUtils(*args, **kwargs) app = app_utils.get_root_browser_app() - uvicorn.run(app, host=default_host, port=port) + uvicorn.run(app, host=host, port=port) -async def async_launch_app(port: int = default_port, *args, **kwargs: dict) -> None: +async def async_launch_app(host: str = default_host, port: int = default_port, *args, **kwargs: dict) -> None: """ Asynchronously launches the application on the specified port. @@ -209,7 +212,7 @@ async def async_launch_app(port: int = default_port, *args, **kwargs: dict) -> N """ app_utils = AppUtils(*args, **kwargs) app = app_utils.get_root_browser_app() - await app_utils.async_run(app, port=port) + await app_utils.async_run(app, host=host, port=port) if __name__ == "__main__":