Update Dockerfile

Dockerfile

@@ -1,42 +1,84 @@
-# Use Python base image
+# Base image using Python 3.10
 FROM python:3.10-slim
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
     build-essential \
     curl \
+    wget \
     git \
+    shadow \
+    lsof \
+    tar \
+    net-tools \
+    iproute2 \
+    iputils-ping \
+    jq \
+    ca-certificates \
     && rm -rf /var/lib/apt/lists/*
 
+# Define directories
+ENV BASE_DIR=/mlflow
+ENV MINIO_DATA_DIR=/data
+ENV MINIO_ROOT_USER=minioadmin
+ENV MINIO_ROOT_PASSWORD=minioadmin
+ENV MINIO_BUCKET=mlflow-artifacts
+ENV MLFLOW_S3_ENDPOINT_URL=http://localhost:9000
+ENV AWS_ACCESS_KEY_ID=${MINIO_ROOT_USER}
+ENV AWS_SECRET_ACCESS_KEY=${MINIO_ROOT_PASSWORD}
+ENV MLFLOW_TRACKING_URI=http://localhost:5000
+ENV BACKEND_STORE_URI=sqlite:///${BASE_DIR}/mlflow.db
+ENV ARTIFACT_ROOT=s3://${MINIO_BUCKET}/
+
+# Create necessary directories
+RUN mkdir -p ${BASE_DIR} ${MINIO_DATA_DIR}
+
+# Install MinIO client
+RUN wget https://dl.min.io/client/mc/release/linux-amd64/mc -O /usr/local/bin/mc && \
+    chmod +x /usr/local/bin/mc
+
+# Install MinIO server
+RUN wget https://dl.min.io/server/minio/release/linux-amd64/minio -O /usr/local/bin/minio && \
+    chmod +x /usr/local/bin/minio
+
 # Install MLflow and dependencies
 RUN pip install --no-cache-dir \
     mlflow \
     scikit-learn \
     pandas \
-    numpy
-
-# Define a writable base directory for Hugging Face Spaces
-ENV BASE_DIR=/tmp
-RUN mkdir -p ${BASE_DIR}/mlruns ${BASE_DIR}/mlflow-db
-
-# Define environment variables for MLflow
-ENV DB_PATH=${BASE_DIR}/mlflow-db/mlflow.db
-ENV MLFLOW_TRACKING_URI=file://${BASE_DIR}/mlruns
-ENV BACKEND_STORE_URI=sqlite:///${DB_PATH}
-ENV ARTIFACT_ROOT=${BASE_DIR}/mlruns
-
-# Expose port 7860
-EXPOSE 7860
+    numpy \
+    boto3 \
+    s3fs \
+    psutil \
+    alembic
 
 # Create startup script
 RUN echo '#!/bin/bash\n\
-mkdir -p ${BASE_DIR}/mlruns ${BASE_DIR}/mlflow-db\n\
-mlflow server \
+# Start MinIO server in the background\n\
+minio server ${MINIO_DATA_DIR} --console-address ":9001" --address ":9000" &\n\
+\n\
+# Wait for MinIO to start\n\
+sleep 5\n\
+\n\
+# Configure MinIO client\n\
+mc alias set myminio http://localhost:9000 ${MINIO_ROOT_USER} ${MINIO_ROOT_PASSWORD}\n\
+\n\
+# Create bucket if it does not exist\n\
+if ! mc ls myminio | grep -q ${MINIO_BUCKET}; then\n\
+    mc mb myminio/${MINIO_BUCKET}\n\
+    echo "Created bucket: ${MINIO_BUCKET}"\n\
+fi\n\
+\n\
+# Start MLflow server\n\
+exec mlflow server \
     --host 0.0.0.0 \
-    --port 7860 \
+    --port 5000 \
     --backend-store-uri ${BACKEND_STORE_URI} \
-    --default-artifact-root ${ARTIFACT_ROOT}' > /start.sh \
-    && chmod +x /start.sh
+    --default-artifact-root ${ARTIFACT_ROOT} \
+    --expose-metrics\n' > /start.sh && chmod +x /start.sh
+
+# Expose ports for MLflow (5000) and MinIO (9000, 9001)
+EXPOSE 5000 9000 9001
 
 # Set the entrypoint
-ENTRYPOINT ["/start.sh"]
+ENTRYPOINT ["/start.sh"]