docker-compose.local-llm.yml

version: '3.8'

services:
  form-flow-local:
    build:
      context: ./form-flow-backend
      dockerfile: Dockerfile.local-llm
    ports:
      - "8000:8000"
    environment:
      - DATABASE_URL=sqlite:///./sql_app.db
      - SECRET_KEY=your_secret_key_here
      - ALGORITHM=HS256
      - ACCESS_TOKEN_EXPIRE_MINUTES=30
      # Optional: Add Gemini API key for fallback
      # - GOOGLE_API_KEY=your_gemini_key_here
    volumes:
      - ./models:/app/models  # Cache models locally
      - ./data:/app/data      # Persist database
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    restart: unless-stopped

  # Optional: Redis for caching
  redis:
    image: redis:7-alpine
    ports:
      - "6379:6379"
    volumes:
      - redis_data:/data
    restart: unless-stopped

volumes:
  redis_data: