Improve chat (#2)

Co-authored-by: Myx <info@azaaxin.com>

docker-compose.yml

@@ -26,34 +26,43 @@ services:
       - "2333:2333"
       
   ollama:
-    volumes:
-      - ollama:/root/.ollama
-    # comment below to not expose Ollama API outside the container stack
+    image: ollama/ollama:latest
     ports:
-      - 11434:11434
+      - 7869:11434
+    volumes:
+      - .:/code
+      - ./ollama/ollama:/root/.ollama
     container_name: ollama
     pull_policy: always
     tty: true
-    restart: unless-stopped
-    image: ollama/ollama:latest
+    restart: always
+    environment:
+      - OLLAMA_KEEP_ALIVE=24h
+      - OLLAMA_HOST=0.0.0.0
+    networks:
+      - ollama-docker
 
   ollama-webui:
-    build:
-      context: .
-      args:
-        OLLAMA_API_BASE_URL: '/ollama/api'
-      dockerfile: Dockerfile
-    image: ollama-webui:latest
+    image: ghcr.io/open-webui/open-webui:main
     container_name: ollama-webui
+    volumes:
+      - ./ollama/ollama-webui:/app/backend/data
     depends_on:
       - ollama
     ports:
-      - 3000:8080
-    environment:
-      - "OLLAMA_API_BASE_URL=http://ollama:11434/api"
+      - 8080:8080
+    environment: # https://docs.openwebui.com/getting-started/env-configuration#default_models
+      - OLLAMA_BASE_URLS=http://host.docker.internal:7869 #comma separated ollama hosts
+      - ENV=dev
+      - WEBUI_AUTH=False
+      - WEBUI_NAME=valiantlynx AI
+      - WEBUI_URL=http://localhost:8080
+      - WEBUI_SECRET_KEY=t0p-s3cr3t
     extra_hosts:
       - host.docker.internal:host-gateway
     restart: unless-stopped
+    networks:
+      - ollama-docker
 
 volumes:
   ollama: {}
@@ -61,4 +70,6 @@ volumes:
 networks:
   # create a lavalink network you can add other containers to, to give them access to Lavalink
   lavalink:
-    name: lavalink
+    name: lavalink
+  ollama-docker:
+    external: false