add webui option

Author: wsmlby

cli/build.sh

@@ -19,7 +19,7 @@ echo "Building the binary with PyInstaller..."
 # The entry point is the main function in the homl_cli package
 ENTRY_POINT="homl_cli/main.py"
 # We need to bundle the docker-compose template
-DATA_FILE="homl_cli/docker-compose.yml.template"
+DATA_FILE="homl_cli/data/*"
 # The binary will be named 'homl'
 BINARY_NAME="homl"
 if [ -z "$CLI_VERSION" ]; then
@@ -33,7 +33,7 @@ pyinstaller \
     --onefile \
     --console \
     --add-data "homl_cli/__version.txt:." \
-    --add-data "$DATA_FILE:." \
+    --add-data "$DATA_FILE:data/" \
     "$ENTRY_POINT"
 
 echo "---"

cli/homl_cli/docker-compose.yml.template …oml_cli/data/docker-compose.yml.templatecli/homl_cli/docker-compose.yml.template renamed to cli/homl_cli/data/docker-compose.yml.template cli/homl_cli/docker-compose.yml.template renamed to cli/homl_cli/data/docker-compose.yml.template

@@ -17,4 +17,4 @@ services:
     pid: "host"
     ports:
       - "${HOML_PORT}:8080"
-    deploy: ${HOML_DEPLOY_RESOURCES}
+    deploy: ${HOML_DEPLOY_RESOURCES}

cli/homl_cli/data/openui.template

@@ -0,0 +1,9 @@
+  open-webui:
+    image: ghcr.io/open-webui/open-webui:main
+    ports:
+      - "${WEBUI_PORT}:8080"
+    environment:
+      - WEBUI_AUTH=false
+      - ENABLE_OLLAMA_API=false
+      - OPENAI_API_BASE_URL=http://homl-server:8080/v1
+      - OFFLINE_MODE=true

cli/homl_cli/main.py

@@ -128,8 +128,9 @@ def restart():
 @click.option('--insecure-socket', is_flag=True, help="Use a world-writable socket (less secure).")
 @click.option('--upgrade', is_flag=True, help="Force reinstallation even if the server is already running.")
 @click.option('--gptoss', is_flag=True, help="Use the GPTOSS image instead of the default.")
-def install(insecure_socket, upgrade, gptoss):
-    install_utils.install(insecure_socket=insecure_socket, upgrade=upgrade, gptoss=gptoss)
+@click.option('--webui', is_flag=True, help="Install the Open WebUI alongside the server.")
+def install(insecure_socket, upgrade, gptoss, webui):
+    install_utils.install(insecure_socket=insecure_socket, upgrade=upgrade, gptoss=gptoss, install_webui=webui)
 
 @main.command()
 @click.argument('model_name')

cli/homl_cli/utils/install_utils.py

@@ -135,7 +135,7 @@ def check_and_install_nvidia_runtime():
     return False
 
 
-def install(insecure_socket: bool, upgrade: bool, gptoss: bool):
+def install(insecure_socket: bool, upgrade: bool, gptoss: bool, install_webui: bool = False):
     """Installs and starts the HoML server using Docker Compose."""
     click.echo("🚀 Starting HoML server installation...")
 
@@ -179,7 +179,7 @@ def install(insecure_socket: bool, upgrade: bool, gptoss: bool):
             f"Removing old module info cache at {module_info_cache_path}")
         shutil.rmtree(module_info_cache_path)
     compose_path = homl_dir / "docker-compose.yml"
-    template_path = get_resource_path("docker-compose.yml.template")
+    template_path = get_resource_path("data/docker-compose.yml.template")
     user_etc_path = homl_dir / "etc_pwd_tmp"
 
     socket_dir.mkdir(parents=True, exist_ok=True)
@@ -212,6 +212,14 @@ def install(insecure_socket: bool, upgrade: bool, gptoss: bool):
         HOML_MODEL_UNLOAD_IDLE_TIME=str(model_unload_idle_time)
     )
 
+    if install_webui:
+        template_path = get_resource_path("data/openui.template")
+        with open(template_path, 'r') as f:
+            openui_template = Template(f.read())
+        compose_content += openui_template.substitute(
+            WEBUI_PORT=str(7457)
+        )
+
     with open(compose_path, 'w') as f:
         f.write(compose_content)
 
@@ -228,6 +236,9 @@ def print_post_install_message():
         click.echo("  3. Chat with it: 'homl chat qwen3:0.6b'")
         click.echo("\nYour OpenAI-compatible API is available at:")
         click.secho(f"  http://0.0.0.0:{port}", fg="cyan")
+        if install_webui:
+            click.echo("\nYour Open WebUI is available at:")
+            click.secho(f"  http://0.0.0.0:7457", fg="cyan")
 
     try:
         if upgrade: