Update app.py

app.py

@@ -1,20 +1,18 @@
 import requests
 import gradio as gr
-from huggingface_hub import list_models, list_datasets, list_spaces, login
+from huggingface_hub import list_models, list_datasets, list_spaces
 
 # Helper function to get the total storage for models, datasets, or spaces
-def get_total_storage(author, resource_type, oauth_token: gr.OAuthToken | None):
+def get_total_storage(namespace, resource_type, oauth_token: gr.OAuthToken):
     if resource_type == "model":
-        resources = list_models(author=author, token=oauth_token.token)
+        resources = list_models(author=namespace, token=oauth_token.token)
         url_base = "https://huggingface.co/api/models"
     elif resource_type == "dataset":
-        resources = list_datasets(author=author, token=oauth_token.token)
+        resources = list_datasets(author=namespace, token=oauth_token.token)
         url_base = "https://huggingface.co/api/datasets"
     elif resource_type == "space":
-        resources = list_spaces(author=author, token=oauth_token.token)
+        resources = list_spaces(author=namespace, token=oauth_token.token)
         url_base = "https://huggingface.co/api/spaces"
-    else:
-        raise ValueError("Invalid resource type. Choose 'model', 'dataset', or 'space'.")
     
     total_storage = 0
     for resource in resources:
@@ -27,17 +25,11 @@ def get_total_storage(author, resource_type, oauth_token: gr.OAuthToken | None):
 
     return total_storage, len(resources)
 
-def get_report(oauth_token: gr.OAuthToken | None):
-    # Log in to Hugging Face using the provided API token
-    login(api_token)
-    
-    # Replace with the actual author namespace
-    author = "your-namespace"  # Example: 'osanseviero'
-
+def get_report(namespace, oauth_token: gr.OAuthToken):
     # Fetch storage and counts for models, datasets, and spaces
-    model_storage, n_models = get_total_storage(author, "model")
-    dataset_storage, n_datasets = get_total_storage(author, "dataset")
-    space_storage, n_spaces = get_total_storage(author, "space")
+    model_storage, n_models = get_total_storage(namespace, "model", oauth_token)
+    dataset_storage, n_datasets = get_total_storage(namespace, "dataset", oauth_token)
+    space_storage, n_spaces = get_total_storage(namespace, "space", oauth_token)
     
     # Total storage
     total_storage = model_storage + dataset_storage + space_storage
@@ -49,8 +41,7 @@ def get_report(oauth_token: gr.OAuthToken | None):
     
     # Generate a report
     report = f"""
-    ## Hugging Face Storage Report for {author}
-
+    ## Hugging Face Storage Report for {namespace}
     - **Number of Models**: {n_models}
     - **Number of Datasets**: {n_datasets}
     - **Number of Spaces**: {n_spaces}
@@ -62,8 +53,8 @@ def get_report(oauth_token: gr.OAuthToken | None):
     return report
 
 # Gradio interface
-def gradio_interface(api_token):
-    report = get_report(api_token)
+def gradio_interface(namespace, oauth_token: gr.OAuthToken):
+    report = get_report(api_token, author)
     return report
 
 css = """
@@ -75,12 +66,12 @@ with gr.Blocks(css=css) as demo:
     gr.Markdown("# Hugging Face Storage Report")
     
     gr.LoginButton()
-    
+    namespace = gr.Textbox(label="Enter Namespace (username or org)")
     output = gr.Markdown()
     
     # Button to trigger the report generation
     report_button = gr.Button("Generate Report")
-    report_button.click(fn=gradio_interface, inputs=None, outputs=output)
+    report_button.click(fn=gradio_interface, inputs=namespace, outputs=output)
 
 # Launch the Gradio app
-demo.launch()
+demo.launch()