Update Dockerfile

Dockerfile

@@ -1,6 +1,6 @@
 FROM python:3.10-slim
 
-# Install system diagnostics, networking utilities, and analysis tools
+# Install system utilities, network tools, and Chromium with its driver
 USER root
 RUN apt-get update && apt-get install -y --no-install-recommends \
     nmap \
@@ -17,22 +17,26 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     jq \
     socat \
     libcap2-bin \
+    chromium \
+    chromium-driver \
     && rm -rf /var/lib/apt/lists/*
 
 # Set up non-root user (Hugging Face default UID 1000)
 RUN useradd -m -u 1000 user
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH \
-    PYTHONUNBUFFERED=1
+    PYTHONUNBUFFERED=1 \
+    CHROME_BIN=/usr/bin/chromium \
+    CHROMEDRIVER_PATH=/usr/bin/chromedriver
 
 WORKDIR /app
 
-# Ensure application directory permissions
+# Set permissions for application directory
 RUN chown -R user:user /app
 
 USER user
 
-# Install Python runtime dependencies
+# Install Python dependencies including Selenium
 RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir \
     gradio>=4.44.0 \
@@ -40,6 +44,7 @@ RUN pip install --no-cache-dir --upgrade pip && \
     pydantic>=2.5.0 \
     httpx>=0.27.0 \
     beautifulsoup4>=4.12.0 \
+    selenium>=4.18.0 \
     python-multipart>=0.0.9
 
 COPY --chown=user:user . /app