Update Dockerfile

Dockerfile

@@ -1,15 +1,33 @@
-# Use a specific version of Python for stability
+# Use a stable Python base image
 FROM python:3.9-slim
 
-# Set the working directory
-WORKDIR /code
+# Set environment variables to prevent interactive prompts during installation
+ENV DEBIAN_FRONTEND=noninteractive
 
-# Copy requirements and install them
-COPY ./requirements.txt /code/requirements.txt
-RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+# Install necessary system dependencies, including Chromium and Xvfb
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+    chromium \
+    chromium-driver \
+    xvfb \
+    && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
 
-# Copy your application
-COPY ./app.py /code/app.py
+# Create a non-root user for security
+RUN useradd -m appuser
+USER appuser
+WORKDIR /home/appuser/app
 
-# Command to run the Uvicorn server
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# Copy the requirements file and install Python packages
+COPY --chown=appuser:appuser requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy the application code
+COPY --chown=appuser:appuser main.py .
+
+# Expose the port the API will run on
+EXPOSE 7860
+
+# The command to run the FastAPI application
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]