Update app.py

app.py

@@ -1,6 +1,8 @@
-import requests
 import gradio as gr
+import requests
+
 from huggingface_hub import list_models, list_datasets, list_spaces
+from typing import Union
 
 # Helper function to get the total storage for models, datasets, or spaces
 def get_total_storage(namespace, resource_type, oauth_token: gr.OAuthToken):
@@ -25,7 +27,7 @@ def get_total_storage(namespace, resource_type, oauth_token: gr.OAuthToken):
 
     return total_storage, len(resources)
 
-def get_report(namespace, oauth_token: gr.OAuthToken):
+def get_report(namespace, oauth_token: Union[OAuthToken, None]):
     # Fetch storage and counts for models, datasets, and spaces
     model_storage, n_models = get_total_storage(namespace, "model", oauth_token)
     dataset_storage, n_datasets = get_total_storage(namespace, "dataset", oauth_token)
@@ -52,11 +54,6 @@ def get_report(namespace, oauth_token: gr.OAuthToken):
     
     return report
 
-# Gradio interface
-def gradio_interface(namespace, oauth_token: gr.OAuthToken):
-    report = get_report(api_token, author)
-    return report
-
 css = """
 .main_ui_logged_out{opacity: 0.3; pointer-events: none}
 """
@@ -71,7 +68,7 @@ with gr.Blocks(css=css) as demo:
     
     # Button to trigger the report generation
     report_button = gr.Button("Generate Report")
-    report_button.click(fn=gradio_interface, inputs=namespace, outputs=output)
+    report_button.click(fn=get_report, inputs=namespace, outputs=output)
 
 # Launch the Gradio app
 demo.launch()