FROM python:3.10

# Install dependencies
RUN pip install --no-cache-dir mlflow huggingface_hub psycopg2-binary

# Expose port
EXPOSE 7860

# Environment variables
ENV HF_TOKEN=${HF_TOKEN}
ENV MLFLOW_HF_ARTIFACT_REPO=${MLFLOW_HF_ARTIFACT_REPO}
ENV DATABASE_URL=${DATABASE_URL}
ENV PYTHONPATH=/app:${PYTHONPATH}

# Copy files
WORKDIR /app
COPY hf_artifact_repo.py .
COPY sitecustomize.py .
COPY start_mlflow.py .

# Start custom MLflow server
CMD ["python", "start_mlflow.py"]