Update Dockerfile

Dockerfile

@@ -1,74 +1,57 @@
 FROM python:3.10-slim
 
-# Run as root to avoid permission issues
-USER root
-
-# Install system dependencies
+# Install system dependencies for Playwright
 RUN apt-get update && apt-get install -y \
-    chromium \
-    chromium-driver \
-    fonts-liberation \
+    curl \
+    net-tools \
+    wget \
+    unzip \
+    libglib2.0-0 \
     libnss3 \
     libatk1.0-0 \
     libatk-bridge2.0-0 \
-    libdrm2 \
     libxkbcommon0 \
     libxcomposite1 \
     libxdamage1 \
     libxrandr2 \
     libgbm1 \
     libasound2 \
-    curl \
-    net-tools \
-    wget \
-    unzip \
     && rm -rf /var/lib/apt/lists/*
 
 # Set working directory
 WORKDIR /app
 
-# Create cache and tmp directories with proper permissions
-RUN mkdir -p /app/cache/pip /app/tmp/.wdm && \
-    chmod -R 777 /app/cache /app/tmp
+# Create cache directory
+RUN mkdir -p /app/cache/pip && \
+    chmod -R 777 /app/cache
 
-# Set pip cache and tmp directory
+# Set pip cache directory
 ENV XDG_CACHE_HOME=/app/cache/pip
-# Disable pip cache to avoid permission issues
 ENV PIP_NO_CACHE_DIR=1
-ENV TMPDIR=/app/tmp
 
-# Copy requirements first to leverage Docker cache
+# Copy requirements
 COPY requirements.txt .
 
-# Update pip and install dependencies
+# Install dependencies
 RUN pip install --upgrade pip && \
     pip install -r requirements.txt && \
-    pip install tokenizers==0.19.1
+    pip install tokenizers==0.19.1 && \
+    playwright install
 
-# Verify transformers and tokenizers
+# Verify transformers
 RUN python -c "import transformers; from transformers import pipeline; print('Transformers pipeline imported successfully, version:', transformers.__version__)" && \
     python -c "import tokenizers; print('Tokenizers imported successfully, version:', tokenizers.__version__)"
 
-# Verify ChromeDriver
-RUN chromedriver --version && chromium --version
-
-# Copy the rest of the application
+# Copy application
 COPY . .
 
 # Set environment variables
-ENV CHROMEDRIVER_PATH=/usr/bin/chromedriver
-ENV CHROMIUM_PATH=/usr/bin/chromium
-ENV DISPLAY=:99
 ENV PYTHONUNBUFFERED=1
 ENV STREAMLIT_OPT_OUT_OF_USAGE_STATISTICS=true
 ENV HF_HOME=/app/cache
-ENV WDM_CACHE_DIR=/app/tmp/.wdm
-
-# Ensure ChromeDriver and Chromium are executable
-RUN chmod +x /usr/bin/chromedriver /usr/bin/chromium
-
-# Clear /app/tmp at container startup
-CMD ["sh", "-c", "rm -rf /app/tmp/* && streamlit run app.py --server.port=8501 --server.address=0.0.0.0 --server.fileWatcherType=none --server.headless=true"]
 
 # Expose Streamlit port
-EXPOSE 8501
+EXPOSE 8501
+
+# Run Streamlit
+CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0", "--server.fileWatcherType=none", "--server.headless=true"]