Update Dockerfile

Dockerfile

@@ -1,29 +1,32 @@
-# Use the official Python image from the Docker Hub
-FROM python:3.9-slim
+FROM python:3.9
 
-# Set the working directory
-WORKDIR /app
+# Set up working directory and install dependencies
+WORKDIR /code
 
-# Copy the requirements file into the container
-COPY requirements.txt .
+COPY ./requirements.txt /code/requirements.txt
 
-# Install the dependencies
-RUN apt-get update && apt-get install -y git && \
-    pip install torch && \
-    pip install -r requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
-# Set environment variables for cache and API key
-ENV HF_HOME=/app/.cache/huggingface
-ENV OPENAI_API_KEY=your_openai_api_key_here
+# Create a new user and set permissions
+RUN useradd -m -u 1000 user
 
-# Create cache directory and set permissions
-RUN mkdir -p /app/.cache/huggingface && chmod -R 777 /app/.cache
+USER user
+
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH \
+    HF_HOME=/home/user/.cache/huggingface \
+    OPENAI_API_KEY=your_openai_api_key_here
+
+WORKDIR $HOME/app
 
 # Copy the rest of the application code into the container
-COPY . .
+COPY --chown=user . $HOME/app
+
+# Ensure the cache directory has proper permissions
+RUN mkdir -p $HF_HOME && chmod -R 777 $HF_HOME
 
 # Expose the port the app runs on
 EXPOSE 8000
 
-# Run the application
-CMD ["python", "main.py"]
+# Run the application using uvicorn
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]