Update app.py

app.py

@@ -1,17 +1,25 @@
+from pathlib import Path
+import gradio as gr
 from huggingface_hub import snapshot_download
 
 snapshot_download("tiiuae/falcon-7b")
 
-import gradio as gr
-def update(name):
-    return f"Welcome to Gradio, {name}!"
+DATA_PATH = "." # "/data"
+
+def get_storage():
+  dir = Path(DATA_PATH)
+  files = [str(f) for f in dir.glob("**/*") if f.is_file()]
+  usage = sum([f.stat().st_size for f in dir.glob("**/*") if f.is_file()])
+  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, None, [files, storage])
+app.launch()
 
-with gr.Blocks() as demo:
-    gr.Markdown("Start typing below and then click **Run** to see the output.")
-    with gr.Row():
-        inp = gr.Textbox(placeholder="What is your name?")
-        out = gr.Textbox()
-    btn = gr.Button("Run")
-    btn.click(fn=update, inputs=inp, outputs=out)
 
-demo.launch()