plant-disease-rag-assistant/docker-compose.yml at main · mcherif/plant-disease-rag-assistant · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
services:
  api:
    build: .
    image: plant-disease-rag:latest
    container_name: plant_rag_api
    environment:
      PYTHONPATH: /app
      OPENAI_API_KEY: ${OPENAI_API_KEY}
      INDEX_DIR: models/index/kb-faiss-bge
      RETRIEVAL_DEVICE: cpu
      TOP_K: 3
    ports:
      - "8000:8000"
    command: uvicorn src.interface.api:app --host 0.0.0.0 --port 8000
    healthcheck:
      test: ["CMD", "curl", "-f", "http://localhost:8000/health"]
      interval: 30s
      timeout: 5s
      retries: 3

  ui:
    image: plant-disease-rag:latest
    container_name: plant_rag_ui
    depends_on:
      - api
    environment:
      PYTHONPATH: /app
      OPENAI_API_KEY: ${OPENAI_API_KEY}
      INDEX_DIR: models/index/kb-faiss-bge
      API_URL: http://api:8000
    ports:
      - "8501:8501"
    command: streamlit run src/interface/streamlit_app.py --server.port 8501 --server.address 0.0.0.0