Add functionality to customize host

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