Update Docker configuration

sahi-mfg · Dec 16, 2023 · 0e50826 · 0e50826
1 parent dff3a8b
commit 0e50826
Show file tree

Hide file tree

Showing 4 changed files with 22 additions and 16 deletions.
diff --git a/Dockerfile.fastapi → Dockerfile b/Dockerfile.fastapi → Dockerfile
@@ -1,7 +1,10 @@
 FROM tiangolo/uvicorn-gunicorn:python3.10
+
 WORKDIR /app
+
 COPY requirements.txt /tmp/
 RUN pip install --no-cache-dir --upgrade -r /tmp/requirements.txt
+
 COPY . /app/
 
 CMD ["uvicorn","main:app", "--port","5001", "--host","0.0.0.0"]
diff --git a/Dockerfile.streamlit b/Dockerfile.streamlit
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -1,16 +1,16 @@
-version: '3.10'
+version: "3.10"
 
 services:
   fastapi:
     build:
       context: .
-      dockerfile: Dockerfile.fastapi
+      dockerfile: Dockerfile
     ports:
       - 5001:5000
 
   streamlit:
     build:
       context: .
-      dockerfile: Dockerfile.streamlit
+      dockerfile: ui/Dockerfile
     ports:
-      - 8501:8501
+      - 8501:8501
diff --git a/ui/Dockerfile b/ui/Dockerfile
@@ -0,0 +1,15 @@
+FROM python:3.10-slim
+
+WORKDIR /ui
+
+COPY requirements.txt /tmp/
+RUN pip install --no-cache-dir --upgrade -r /tmp/requirements.txt
+
+COPY . /ui/
+
+EXPOSE 8501
+
+HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
+
+
+ENTRYPOINT ["streamlit", "run", "ui/app.py", "--server.port=8501", "--server.address=0.0.0.0"]