version: '3.8'

services:
  ollama:
    image: ollama/ollama:latest
    container_name: ollama-server
    ports:
      - "11434:11434"
    environment:
      - OLLAMA_HOST=0.0.0.0:11434
    volumes:
      - ollama_data:/root/.ollama
    command: serve

  api:
    build: .
    container_name: ollama-api
    ports:
      - "7860:7860"
    depends_on:
      - ollama
    environment:
      - OLLAMA_HOST=http://ollama:11434
      - PORT=7860
    volumes:
      - .:/app
    command: python backend.py

volumes:
  ollama_data:
    driver: local