Update Dockerfile

Dockerfile

@@ -3,16 +3,17 @@ FROM python:3.9-slim
 
 ENV DEBIAN_FRONTEND=noninteractive
 ENV CHROME_BIN=/usr/bin/google-chrome-stable
-
 WORKDIR /app
 
-# Install runtime deps required by Chrome
+# Install system packages needed for Chrome, Xvfb and general runtime.
 RUN apt-get update \
  && apt-get install -y --no-install-recommends \
     ca-certificates \
     wget \
     gnupg \
     unzip \
+    xvfb \
+    x11-utils \
     fonts-liberation \
     libnss3 \
     libxss1 \
@@ -34,7 +35,7 @@ RUN apt-get update \
     libdrm2 \
  && rm -rf /var/lib/apt/lists/*
 
-# Add Google's signing key via gpg and install Google Chrome stable
+# Add Google's signing key and install google-chrome-stable
 RUN wget -q -O - https://dl.google.com/linux/linux_signing_key.pub \
     | gpg --dearmor -o /usr/share/keyrings/google-chrome.gpg \
  && echo "deb [arch=amd64 signed-by=/usr/share/keyrings/google-chrome.gpg] http://dl.google.com/linux/chrome/deb/ stable main" \
@@ -43,15 +44,16 @@ RUN wget -q -O - https://dl.google.com/linux/linux_signing_key.pub \
  && apt-get install -y --no-install-recommends google-chrome-stable \
  && rm -rf /var/lib/apt/lists/*
 
-# Install Python dependencies
+# Copy requirements and install python packages
 COPY requirements.txt /app/requirements.txt
 RUN pip install --no-cache-dir --upgrade pip \
  && pip install --no-cache-dir -r /app/requirements.txt
 
-# Copy app
+# Copy application code
 COPY . /app
 
+# Expose port
 EXPOSE 7860
 
-# Use uvicorn to serve the FastAPI app
+# Run the app (runs as root for simplicity)
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]