docker-compose.yml

networks:
  my_network:
    driver: bridge

services:
  ollama:
    container_name: "ollama"
    build:
      context: .
      dockerfile: Dockerfile.ollama
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 2
              capabilities: [gpu]
    ports:
       - "11434:11434"
    networks:
      - my_network

  ollamaserver:
    container_name: "ollamaserver"
    build:
      context: .
      dockerfile: Dockerfile
    environment:
      PINECONE_API_KEY: ${PINECONE_API_KEY}
      PINECONE_INDEX_NAME: ${PINECONE_INDEX_NAME}
      SUNO_API_URL: ${SUNO_API_URL}
      OLLAMA_BASE_URL: ${OLLAMA_BASE_URL}
      OLLAMA_MODEL: ${OLLAMA_MODEL}
      VISUAL_MODEL: ${VISUAL_MODEL}
      CORBADO_PROJECTID: ${CORBADO_PROJECTID}
      CORBADO_API_KEY: ${CORBADO_API_KEY}
      CORBADO_FRONTENDAPI: ${CORBADO_FRONTENDAPI}
      CORBADO_BACKENDAPI: ${CORBADO_BACKENDAPI}
      APP_PORT: ${APP_PORT}
      APP_HOST: ${APP_HOST}
      OLLAMA_HOST: ${OLLAMA_HOST}
      SQLLITE_DATABASE_DIR: ${SQLLITE_DATABASE_DIR}
    deploy:
      resources:
        limits:
          cpus: '2'
          memory: 10GB
        reservations:
          cpus: '2'
          memory: 10GB
    restart: always
    depends_on:
      - suno-api
      - ollama
    links:
      - suno-api
      - ollama
    ports:
      - "3000:3000"
    networks:
      - my_network

  suno-api:
    container_name: "suno-api"
    build:
      context: ./suno-api
      dockerfile: Dockerfile.dev
    environment:
      SUNO_COOKIE: ${SUNO_COOKIE}
    volumes:
      - ./public:/app/public
    restart: always
    ports:
      - "3001:3000"
    networks:
      - my_network