Update app.py

app.py

@@ -1,29 +1,30 @@
-import os
 import gradio as gr
+from pathlib import Path
 
-HF_TOKEN = os.getenv('HW_TOKEN')
-#print (HF_TOKEN)
-hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "save_audio")
+DATA_PATH = 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"
 
-def calculator(num1, operation, num2):
-    if operation == "add":
-        return num1 + num2
-    elif operation == "subtract":
-        return num1 - num2
-    elif operation == "multiply":
-        return num1 * num2
-    elif operation == "divide":
-        return num1 / num2
+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)
 
-
-iface = gr.Interface(
-    calculator,
-    ["number", gr.Radio(["add", "subtract", "multiply", "divide"]), "number"],
-    "number",
-    #description="Check out the crowd-sourced dataset at: [https://huggingface.co/datasets/aliabd/crowdsourced-calculator-demo](https://huggingface.co/datasets/aliabd/crowdsourced-calculator-demo)",
-    allow_flagging="manual",
-    flagging_options=["wrong sign", "off by one", "other"],
-    flagging_callback=hf_writer
-)
-
-iface.launch()
+# Files that you explicitly allow on allowed_paths 
+app.launch(allowed_paths=[str(DATA_PATH)])