new dockerfile

Dockerfile

@@ -1,19 +1,33 @@
 # Use the official Python image from the Docker Hub
 FROM python:3.10-slim
 
-# Set the working directory in the container
-WORKDIR /app
+# Set environment variables
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH
 
-COPY requirements.txt .
-RUN pip install -r requirements.txt
+# Set up a new user named "user" with user ID 1000
+RUN useradd -m -u 1000 user
 
-# Copy the rest of the application code to the working directory
-COPY . .
+# Switch to the "user" user
+USER user
+
+# Set the working directory to the user's home directory
+WORKDIR $HOME/app
+
+# Install pip
+RUN pip install --no-cache-dir --upgrade pip
 
-ENV SENTENCE_TRANSFORMERS_HOME ./.cache
+# Copy only the requirements file and install dependencies
+COPY --chown=user requirements.txt .
+
+# Install the application dependencies
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy the rest of the application code to the working directory
+COPY --chown=user . .
 
 # Expose the port FastAPI will run on
 EXPOSE 7860
 
 # Command to run the FastAPI application
-CMD ["python", "run.py"]
+CMD ["python", "run.py"]

run.py

@@ -5,9 +5,6 @@ from src.search_engine import PromptSearchEngine
 from src.prompt_loader import PromptLoader
 import os
 
-os.environ["SENTENCE_TRANSFORMERS_HOME"] = "./.cache"
-os.environ["HF_HOME"] = "./.cache"
-
 # Constants
 SEED = 42
 DATA_SIZE = 100