$ErrorActionPreference = "Stop"

$BACKEND_STORE_DB_PATH = "./mlflow/mlruns.sqlite"
$DEFAULT_ARTIFACT_ROOT = "./mlflow/mlartifacts/"

$MLFLOW_PORT = if ($env:MLFLOW_PORT) { $env:MLFLOW_PORT } else { 5000 }

$BACKEND_URI = "sqlite:///$BACKEND_STORE_DB_PATH"

New-Item -ItemType Directory -Force -Path $DEFAULT_ARTIFACTS_ROOT | Out-Null

& mlflow server `
    --backend-store-uri="$BACKEND_URI" `
    --default-artifact-root="$DEFAULT_ARTIFACT_ROOT" `
    -p $MLFLOW_PORT