From 13a606615c6b0ee31ef16d43f2b3ba79a84d5cc2 Mon Sep 17 00:00:00 2001
From: lcolok <425311101@qq.com>
Date: Sun, 30 Jul 2023 00:01:11 +0800
Subject: [PATCH] Add functionality to customize host

---
 app.py | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

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__":