from pathlib import Path
import gradio as gr
from huggingface_hub import snapshot_download

snapshot_download("tiiuae/falcon-7b")

DATA_PATH = Path("/data")

def get_storage():
    files = [
        {
            "orig_name": file.name,
            "name": file.resolve(),
            "size": file.stat().st_size,
            "data": None,
            "is_file": True,
        }
        for file in DATA_PATH.glob("**/*")
        if file.is_file()
    ]
    usage = sum([f['size'] for f in files])

    return files, f"{usage/(1024.0 ** 3):.3f}GB"


with gr.Blocks() as app:
    with gr.Row():
        with gr.Column():
            btn = gr.Button("Run")
        with gr.Column():
            files = gr.Files(label="Files")
            storage = gr.Text(label="Total Usage")
    btn.click(get_storage, inputs=None, outputs=[files, storage], postprocess=False)

# Files that you explicitly allow on allowed_paths
app.launch(allowed_paths=["/data"])