Update Dockerfile

Dockerfile

@@ -1,31 +1,40 @@
-# Use a lightweight Python base image
+# 1. Use an official lightweight Python image
 FROM python:3.9-slim
 
-# Set the working directory inside the container
+# 2. Set environment variables to prevent Python from buffering stdout/stderr
+ENV PYTHONUNBUFFERED=1 \
+    PYTHONDONTWRITEBYTECODE=1 \
+    PIP_NO_CACHE_DIR=1
+
+# 3. Create a non-root user (Security Best Practice for HF Spaces)
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+# 4. Set the working directory
 WORKDIR /app
 
-# 1. Install System Dependencies (REQUIRED for RDKit & File Parsing)
-# - libxrender1, libxext6, libsm6: Required for RDKit Image Drawing
-# - libexpat1: Required for XML/SVG parsing
+# 5. Install system dependencies (Git is often needed for cloning repos)
+USER root
 RUN apt-get update && apt-get install -y \
-    libxrender1 \
-    libxext6 \
-    libsm6 \
-    libexpat1 \
+    git \
+    build-essential \
     && rm -rf /var/lib/apt/lists/*
+USER user
 
-# 2. Copy requirements first (from src folder) to cache dependencies
-COPY src/requirements.txt .
+# 6. Copy requirements first (to leverage Docker caching)
+COPY --chown=user requirements.txt ./
 
-# 3. Install Python dependencies
-RUN pip install --no-cache-dir --upgrade pip && \
-    pip install --no-cache-dir -r requirements.txt
+# 7. Install Python dependencies
+RUN pip install --upgrade pip && \
+    pip install -r requirements.txt
 
-# 4. Copy the rest of the application (src folder content)
-COPY src/ .
+# 8. Copy the rest of the application files
+# This includes app.py and your .pkl model files
+COPY --chown=user . .
 
-# 5. Expose port 7860 (Hugging Face Default)
+# 9. Expose the port Hugging Face Spaces expects (7860)
 EXPOSE 7860
 
-# 6. Run the application
-CMD ["streamlit", "run", "app.py", "--server.port", "7860", "--server.address", "0.0.0.0"]
+# 10. Run the Streamlit app on port 7860
+CMD ["streamlit", "run", "app.py", "--server.address", "0.0.0.0", "--server.port", "7860"]