- name: Setup open-webui
  hosts: openwebui
  vars:
    podman_user: "{{ open_webui_user | default('openwebui') }}"
    podman_group: "{{ open_webui_group | default('openwebui') }}"
    podman_network_name: "{{ open_webui_network_name | default('openwebui-network') }}"
    podman_user_home: "{{ open_webui_home | default('/home/open-webui') }}"
    pods:
      - name: ollama
        build:
          dockerfile: "{{ lookup('file', 'dockerfiles/Dockerfile.ollama') }}"
        ports:
          - 0.0.0.0:11434:11434/tcp
        volumes:
          - "{{ podman_user_home }}/ollama:/home/worker/.ollama:Z"
        device:
          - "nvidia.com/gpu=all"
      - name: open-webui
        repo:
          image: ghcr.io/open-webui/open-webui
          tag: main
        ports:
          - 0.0.0.0:8080:8080/tcp
        volumes:
          - "{{ podman_user_home }}/open-webui:/app/backend/data:Z"
        device:
          - "nvidia.com/gpu=all"
        env:
          OLLAMA_BASE_URL: "http://ollama:11434"

  tasks:
    - name: Setup nvidia-cdi-generator
      ansible.builtin.include_tasks:
        file: tasks/nvidia-cdi-generator.yml

    - name: Setup Pods
      ansible.builtin.include_tasks:
        file: tasks/podman.yml