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}"}
-