Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 from huggingface_hub import HfApi
 import asyncio
 import os
+import plotly.express as px
 
 # Initialize the Hugging Face API
 api = HfApi()
@@ -89,7 +90,7 @@ def get_cached_html_page(username):
     user_data = asyncio.run(fetch_user_content(username))
     if "error" in user_data:
         return None, user_data["error"]
-    return generate_html_page(username, user_data["models"], user_data["datasets"]), None
+    return generate_html_page(username, user_data["models"], user_data["datasets"]), user_data
 
 # Streamlit app setup - the nerve center of our operation! 🎛️
 st.title("Hugging Face User Content Display - Let's Automate Some Fun! 🎉")
@@ -105,6 +106,9 @@ if st.button("Show User Content"):
     if usernames:
         username_list = [username.strip() for username in usernames.split('\n') if username.strip()]
 
+        # Collect statistics for Plotly graphs
+        stats = {"username": [], "models_count": [], "datasets_count": []}
+
         st.markdown("### User Content Overview")
         for username in username_list:
             with st.container():
@@ -112,15 +116,50 @@ if st.button("Show User Content"):
                 st.markdown(f"**{username}** [🔗 Profile](https://huggingface.co/{username})")
 
                 # Generate HTML page and provide download link - because who wouldn't want a custom webpage? 🌐
-                html_file_path, error = get_cached_html_page(username)
+                html_file_path, user_data = get_cached_html_page(username)
 
-                if error:
-                    st.warning(f"{username}: {error} - Looks like the AI needs a coffee break ☕")
+                if not html_file_path:
+                    st.warning(f"{username}: {user_data} - Looks like the AI needs a coffee break ☕")
                 else:
                     st.markdown(f"[📄 Download {username}'s HTML Page]({html_file_path})")
 
+                    # Add to statistics for Plotly graphs
+                    stats["username"].append(username)
+                    stats["models_count"].append(len(user_data["models"]))
+                    stats["datasets_count"].append(len(user_data["datasets"]))
+
+                    # Models section with expander - 🧠 because AI models are brainy! 🧠
+                    with st.expander(f"🧠 Models ({len(user_data['models'])})", expanded=False):
+                        if user_data['models']:
+                            for model in user_data['models']:
+                                model_name = model.modelId.split("/")[-1]
+                                st.markdown(f"- [{model_name}](https://huggingface.co/{model.modelId})")
+                        else:
+                            st.markdown("No models found. Did you check under the rug? 🕵️‍♂️")
+
+                    # Datasets section with expander - 📚 because data is the foundation of AI! 📚
+                    with st.expander(f"📚 Datasets ({len(user_data['datasets'])})", expanded=False):
+                        if user_data['datasets']:
+                            for dataset in user_data['datasets']:
+                                dataset_name = dataset.id.split("/")[-1]
+                                st.markdown(f"- [{dataset_name}](https://huggingface.co/datasets/{dataset.id})")
+                        else:
+                            st.markdown("No datasets found. Maybe they’re still baking in the oven? 🍪")
+
                 st.markdown("---")
 
+        # Plotly graphs to visualize the number of models and datasets each user has
+        if stats["username"]:
+            st.markdown("### User Content Statistics")
+
+            # Plotting the number of models per user
+            fig_models = px.bar(x=stats["username"], y=stats["models_count"], labels={'x':'Username', 'y':'Number of Models'}, title="Number of Models per User")
+            st.plotly_chart(fig_models)
+
+            # Plotting the number of datasets per user
+            fig_datasets = px.bar(x=stats["username"], y=stats["datasets_count"], labels={'x':'Username', 'y':'Number of Datasets'}, title="Number of Datasets per User")
+            st.plotly_chart(fig_datasets)
+
     else:
         st.warning("Please enter at least one username. Don't be shy! 😅")
 
@@ -131,4 +170,5 @@ st.sidebar.markdown("""
 2. Click 'Show User Content'.
 3. View the user's models and datasets along with a link to their Hugging Face profile.
 4. Download an HTML page for each user to use the absolute links offline!
+5. Check out the statistics visualizations at the end!
 """)