Update Dockerfile

Dockerfile

@@ -1,20 +1,32 @@
-FROM python:3.13.5-slim
+# Use lightweight official Python image
+FROM python:3.11-slim
 
+# Prevent Python from writing .pyc files and enable buffering
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV PYTHONUNBUFFERED=1
+
+# Set working directory inside container
 WORKDIR /app
 
-RUN apt-get update && apt-get install -y \
-    build-essential \
-    curl \
-    git \
-    && rm -rf /var/lib/apt/lists/*
+# Copy only requirements first (best practice for caching)
+COPY requirements.txt .
+
+# Install dependencies (fast & clean)
+RUN pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir -r requirements.txt
 
-COPY requirements.txt ./
-COPY src/ ./src/
+# Copy your entire app (code + model files)
+COPY . .
 
-RUN pip3 install -r requirements.txt
+# Create a non-root user (security best practice)
+RUN useradd -m appuser && chown -R appuser:appuser /app
+USER appuser
 
+# Expose Streamlit port
 EXPOSE 8501
 
-HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
+# Health check (optional but recommended)
+HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health || exit 1
 
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+# Run Streamlit app
+CMD ["streamlit", "run", "mushroom_app.py", "--server.port=8501", "--server.address=0.0.0.0", "--server.enableCORS=false", "--server.enableXsrfProtection=false"]