import streamlit as st
import huggingface_hub as hf_hub
import monai
import os
import zipfile

hf_hub.login(token=st.secrets["HF_TOKEN"])

with st.spinner("Downloading Dataset"):
    data_path = hf_hub.hf_hub_download(repo_id="osbm/prostate158", filename="data.zip", repo_type="dataset")

st.write(data_path)
with st.spinnet("Unzipping..."):
    with zipfile.ZipFile(data_path, 'r') as zip_ref:
        zip_ref.extractall(".")


st.write(os.listdir(os.getcwd()))
st.write(os.getcwd())