dockerfile and app updates

.gitignore

@@ -137,3 +137,5 @@ notebooks/train_model.ipynb
 notebooks/clustering.ipynb
 
 *.db
+
+*.pem

Dockerfile

@@ -0,0 +1,18 @@
+# Use the official Python image as the base image
+FROM python:3.10
+
+# Set the working directory
+WORKDIR /app
+
+# Copy requirements file and install dependencies
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy the rest of the application code
+COPY . .
+
+# Expose the port that the app runs on
+EXPOSE 8000
+
+# Execute the script when the container starts
+CMD ["uvicorn", "src.app.app:app", "--host", "0.0.0.0"]

docker-compose.yml

@@ -0,0 +1,6 @@
+version: '3.10'
+services:
+  app:
+    build: .
+    ports:
+      - '8000:8000'

requirements.txt

@@ -27,5 +27,6 @@ transformers==4.28.1
 torch==2.0.0
 openai
 jupysql
-fastapi==0.95.1
-uvicorn==0.21.1
+fastapi>=0.68.0,<0.69.0
+pydantic>=1.8.0,<2.0.0
+uvicorn>=0.15.0,<0.16.0

src/app/app.py

@@ -54,4 +54,3 @@ async def search(query: Query):
 
     except Exception as e:
         return {"error": f"SQL query failed. {e}"}
-