version: '3.8'

services:
  pyspark-catalyst:
    build: .
    ports:
      - "8501:8501"
    environment:
      - GROQ_API_KEY=${GROQ_API_KEY}
      - MODEL_NAME=${MODEL_NAME:-llama3-70b-8192}
      - TEMPERATURE=${TEMPERATURE:-0.3}
    volumes:
      # Persist FAISS index between container restarts
      - faiss_data:/app/faiss_index
    restart: unless-stopped
    healthcheck:
      test: ["CMD", "curl", "-f", "http://localhost:8501/_stcore/health"]
      interval: 30s
      timeout: 10s
      retries: 3
      start_period: 40s

volumes:
  faiss_data: