tf update

Author: Olawale Abimbola

.env

@@ -0,0 +1,4 @@
+S3_ACCESS_KEY = user
+S3_SECRET_KEY = user@@123
+S3_URL = minio:9000
+S3_BUCKET = recommendation-bucket

.gitignore

@@ -0,0 +1 @@
+data

Dockerfile

@@ -1,8 +1,8 @@
-FROM python:3.7
-# WORKDIR /app
-# COPY ./app /app
-COPY requirements.txt ./requirements.txt
-RUN pip3 install --upgrade pip
-RUN pip3 install -r requirements.txt
-EXPOSE $PORT
-CMD ["uvicorn", "app:app", "--bind 0.0.0.0:$PORT", "--reload"]
+FROM python:3.8
+
+WORKDIR /app
+CMD ["uvicorn", "app:app", "--host=0.0.0.0", "--reload"]
+
+COPY src/* /app/
+COPY data/* /app/
+RUN pip install -r requirements.txt

app.py

@@ -1,7 +1,59 @@
-version: "3.5"
+version: '3.8'
 services:
+  etl:
+    container_name: etl
+    build: .
+    command: python etl.py
+    env_file: 
+      - .env
+    depends_on:
+      - minio
+    networks:
+      - app-net
+  trainer:
+    container_name: trainer
+    build: .
+    command: python trainer.py
+    env_file: 
+      - .env
+    depends_on:
+      - minio
+    networks:
+      - app-net
   fastapi:
+    container_name: api
     restart: "always"
-    build: fastapi
+    env_file: 
+      - .env
+    build: .
+    depends_on:
+      - minio
+      - etl
+      - trainer
+    networks:
+      - app-net
+    command: uvicorn app:app --host=0.0.0.0 --reload
     ports:
-      - 8000:8000
+      - 8000:8000
+  minio:
+    image: minio/minio
+    container_name: minio
+    command: "server /data --console-address ':9001'"
+    environment:
+      MINIO_ROOT_USER: ${S3_ACCESS_KEY}
+      MINIO_ROOT_PASSWORD: ${S3_SECRET_KEY}
+    ports:
+      - 9000:9000
+      - 9001:9001
+    networks:
+      - app-net
+    volumes:
+      - ./data/minio-data:/data
+volumes:
+  cache:
+    driver: local
+  minio-data:
+    driver: local
+networks:
+  app-net:
+    driver: bridge

app/recommender.py

@@ -0,0 +1,10 @@
+from locust import HttpUser, task, between
+import random
+
+class MovieRecommender(HttpUser):
+    wait_time = between(1, 3)  # Add some wait time between requests
+
+    @task
+    def get_recommendation(self):
+        movie_name = random.choice(['Titan A.E.', "Ender's Game", "Independence Day", 'Avatar', 'The Fifth Element', 'Star Trek Into Darkness', 'Jupiter Ascending', 'Battle: Los Angeles'])
+        self.client.get(f"/predict?movie={movie_name}")