Update Dockerfile

Dockerfile

@@ -26,7 +26,8 @@ WORKDIR /app
 
 # Create cache directories with proper permissions
 RUN mkdir -p /app/cache/pip /tmp/.wdm /tmp/chrome-data && \
-    chmod -R 777 /app/cache /tmp/.wdm /tmp/chrome-data
+    chmod -R 777 /app/cache /tmp/.wdm /tmp/chrome-data && \
+    rm -rf /tmp/*  # Clear /tmp at build time
 
 # Set pip cache directory
 ENV XDG_CACHE_HOME=/app/cache/pip
@@ -45,6 +46,9 @@ RUN pip install --upgrade pip && \
 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 . .
 
@@ -56,13 +60,12 @@ ENV PYTHONUNBUFFERED=1
 ENV STREAMLIT_OPT_OUT_OF_USAGE_STATISTICS=true
 ENV HF_HOME=/app/cache
 ENV WDM_CACHE_DIR=/tmp/.wdm
-ENV CHROME_USER_DATA_DIR=/tmp/chrome-data
 
 # Ensure ChromeDriver and Chromium are executable
 RUN chmod +x /usr/bin/chromedriver /usr/bin/chromium
 
-# Expose Streamlit port
-EXPOSE 8501
+# Clear /tmp at container startup
+CMD ["sh", "-c", "rm -rf /tmp/* && chmod -R 777 /tmp && streamlit run app.py --server.port=8501 --server.address=0.0.0.0 --server.fileWatcherType=none --server.headless=true"]
 
-# Run Streamlit
-CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0", "--server.fileWatcherType=none", "--server.headless=true"]
+# Expose Streamlit port
+EXPOSE 8501