Update Dockerfile

Dockerfile

@@ -1,22 +1,58 @@
 FROM python:3.10
 
+# Install Chromium + dependencies
+USER root
+RUN apt-get update && apt-get install -y \
+    chromium \
+    chromium-driver \
+    fonts-liberation \
+    libasound2 \
+    libatk1.0-0 \
+    libc6 \
+    libcairo2 \
+    libcups2 \
+    libdbus-1-3 \
+    libexpat1 \
+    libfontconfig1 \
+    libgcc1 \
+    libglib2.0-0 \
+    libgtk-3-0 \
+    libnspr4 \
+    libnss3 \
+    libpango-1.0-0 \
+    libx11-6 \
+    libx11-xcb1 \
+    libxcb1 \
+    libxcomposite1 \
+    libxcursor1 \
+    libxdamage1 \
+    libxext6 \
+    libxfixes3 \
+    libxi6 \
+    libxrandr2 \
+    libxrender1 \
+    libxss1 \
+    libxtst6 \
+    wget \
+    && rm -rf /var/lib/apt/lists/*
+
+# Create user
 RUN useradd -m -u 1000 user
 
 WORKDIR /code
 
 COPY ./requirements.txt /code/requirements.txt
 
-USER root
 RUN chown -R user:user /code
-
 USER user
 
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
-RUN pip show uvicorn
-
 COPY --chown=user:user . /code
 
 ENV PATH="/home/user/.local/bin:${PATH}"
 
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+# Tell html2image where Chromium is
+ENV HTML2IMAGE_CHROME_PATH="/usr/bin/chromium"
+
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]