[Structure] Dockerizing the project

Makefile

@@ -25,3 +25,19 @@ test:  ## Run tests
 build:
 	make format
 	make coverage
+
+# For building the docker compose
+docker:
+	docker-compose -f docker-compose.yml up --build
+
+# For starting the docker comppose,
+up:
+	docker-compose -f docker-compose.yml up
+
+## For removing containers
+remove:
+	docker-compose down --remove-orphans
+
+# List all containers
+ps:
+	docker-compose ps -a

backend/Dockerfile

@@ -0,0 +1,33 @@
+# Get image for Python
+FROM python:3.8-slim-buster
+
+# Set working directory
+WORKDIR /app/
+
+# pip does not like being run as root, so create a user
+RUN useradd --create-home jina
+
+# Add the data folder locally to container
+COPY . .
+
+# Give read permissions for the data folder
+RUN chown jina data
+
+# Switch to user
+USER jina
+
+# Path change needed for huggingface-cli and jina
+ENV PYTHONPATH "${PYTHONPATH}:/home/jina/.local/bin"
+ENV PATH "${PATH}:/home/jina/.local/bin"
+
+# Add requirement contents
+COPY ./requirements.txt requirements.txt
+
+# Install dependencies
+RUN pip3 install -r requirements.txt
+
+# Expose port
+EXPOSE 8020
+
+# Run the application
+CMD [ "python", "src/app.py" ]

backend/requirements.txt

@@ -0,0 +1,20 @@
+torch==1.9.0
+transformers==4.9.0
+jina==2.0.10
+asgiref==3.4.1
+click==8.0.1
+h11==0.12.0
+uvicorn==0.14.0
+fastapi==0.67.0
+pydantic==1.8.2
+starlette==0.14.2
+typing-extensions==3.10.0.0
+uvloop==0.15.3
+aiohttp==3.7.4.post0
+async-timeout==3.0.1
+attrs==21.2.0
+chardet==4.0.0
+idna==3.2
+multidict==5.1.0
+typing-extensions==3.10.0.0
+yarl==1.6.3

backend/app.py → backend/src/app.py

@@ -25,7 +25,7 @@ def main():
         proteins = DocumentArray(docs_generator)[0:42]
 
     flow = (
-        Flow(port_expose=12345, protocol='http')
+        Flow(port_expose=8020, protocol='http')
         .add(uses=ProtBertExecutor)
         .add(uses=MyIndexer)
     )

docker-compose.yml

@@ -0,0 +1,31 @@
+version: "3"
+
+services:
+
+  backend:
+    image: rubix982/jina-backend:v1.0.0    
+    container_name: protein-search-backend
+    build:
+      context: ./backend
+    stdin_open: true      
+    restart: always
+    volumes: 
+        - "./backend:/app"
+        - "./backend/requirements.txt:/app/requirements.txt"
+    ports:
+      - "8020:8020"
+
+  frontend:
+    image: rubix982/jina-frontend:v1.0.0
+    container_name: protein-search-frontend
+    build: 
+      context: ./frontend
+    stdin_open: true
+    restart: always
+    volumes: 
+      - "./frontend:/app"
+      - "./frontend/requirements.txt:/app/requirements.txt"    
+    ports:
+      - "8501:8501"
+    depends_on:
+      - "backend"

frontend/Dockerfile

@@ -0,0 +1,30 @@
+# Get image for Python
+FROM python:3.8-slim-buster
+
+# Set working directory
+WORKDIR /app/
+
+# pip does not like being run as root, so create a user
+RUN useradd --create-home jina
+
+# Add the data folder locally to container
+COPY . .
+
+# Switch to user
+USER jina
+
+# Path change needed for huggingface-cli and jina
+ENV PYTHONPATH "${PYTHONPATH}:/home/jina/.local/bin"
+ENV PATH "${PATH}:/home/jina/.local/bin"
+
+# Add requirement contents
+COPY ./requirements.txt requirements.txt
+
+# Install dependencies
+RUN pip3 install -r requirements.txt
+
+# Expose Port
+EXPOSE 8501
+
+# Run the application
+CMD [ "streamlit", "run", "app.py" ]

frontend/app.py

@@ -38,7 +38,7 @@ def protein_3d(pdb_id="1YCR", width=200, height=200):
         width=width,
     )
 
-endpoint = "http://localhost:12345/search"
+endpoint = "http://localhost:8020/search"
 
 st.title("Proteins Search")
 

frontend/requirements.txt

@@ -0,0 +1,2 @@
+streamlit==0.85.0
+requests==2.26.0