#!/bin/bash

HF_DATASET_REPO="InstaDeepAI/mlipaudit-results"
HF_DATASET_LOCAL_PATH="./results"

echo "Attempting to download dataset from ${HF_DATASET_REPO} to ${HF_DATASET_LOCAL_PATH}..."

# Ensure the local data directory exists
mkdir -p ${HF_DATASET_LOCAL_PATH}

huggingface-cli download --repo-id ${HF_DATASET_REPO} --local-dir ${HF_DATASET_LOCAL_PATH} --repo-type dataset

if [ $? -eq 0 ]; then
    echo "Dataset downloaded successfully!"
else
    echo "Failed to download dataset."
    exit 1 # Exit if download fails
fi

echo "Starting Streamlit app..."
exec streamlit run src/app.py --server.port=8501 --server.address=0.0.0.0