Update Dockerfile

Dockerfile

@@ -1,29 +1,29 @@
-# Use the same base image
-FROM python:3.9-slim
+# Use an official Python runtime as the base image
+FROM python:3.11-slim
 
-# Set working directory to match the app structure
+# Set working directory in the container
 WORKDIR /app
 
-# Install system dependencies
+# Install system dependencies (for PIL and other libraries)
 RUN apt-get update && apt-get install -y \
-    build-essential \
-    curl \
-    software-properties-common \
-    git \
+    libjpeg-dev \
+    zlib1g-dev \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy requirements and source files
-COPY requirements.txt ./
-COPY src/ ./src/
+# Copy requirements.txt to the working directory
+COPY requirements.txt .
 
 # Install Python dependencies
-RUN pip3 install --no-cache-dir -r requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
 
-# Expose the Streamlit port
+# Copy the app files to the working directory
+COPY app.py .
+
+# Expose the port Streamlit uses
 EXPOSE 8501
 
-# Healthcheck for Spaces
-HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
+# Set environment variables for Hugging Face token (optional, can be passed at runtime)
+ENV HF_TOKEN=""
 
-# Run Streamlit with XSRF protection disabled
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0", "--server.enableXsrfProtection", "false"]
+# Command to run the Streamlit app
+CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0"]