Update app.py

app.py

@@ -7,6 +7,8 @@ from datetime import datetime
 from concurrent.futures import ThreadPoolExecutor, as_completed
 from functools import lru_cache
 import time
+import requests
+from collections import Counter
 
 st.set_page_config(page_title="HF Contributions", layout="wide")
 api = HfApi()
@@ -34,28 +36,57 @@ def cached_list_items(username, kind):
     return []
 
 
-# Function to fetch trending model repositories
+# Function to fetch trending accounts and create stats
 @lru_cache(maxsize=1)
 def get_trending_accounts(limit=100):
     try:
-        # Fetch trending models, datasets, and spaces
-        trending_models = list(api.list_models(sort="trending", limit=limit))
-        trending_datasets = list(api.list_datasets(sort="trending", limit=limit))
-        trending_spaces = list(api.list_spaces(sort="trending", limit=limit))
+        # Get spaces for stats calculation
+        spaces_response = requests.get("https://huggingface.co/api/spaces", 
+                                      params={"limit": 10000}, 
+                                      timeout=30)
         
-        # Extract unique authors
-        authors = set()
-        for item in trending_models + trending_datasets + trending_spaces:
-            if hasattr(item, "author"):
-                authors.add(item.author)
-            elif hasattr(item, "id") and "/" in item.id:
-                authors.add(item.id.split("/")[0])
-        
-        # Return sorted list of unique authors
-        return sorted(list(authors))[:limit]
+        if spaces_response.status_code == 200:
+            spaces = spaces_response.json()
+            
+            # Count spaces by owner
+            owner_counts = {}
+            for space in spaces:
+                if '/' in space.get('id', ''):
+                    owner, _ = space.get('id', '').split('/', 1)
+                else:
+                    owner = space.get('owner', '')
+                
+                if owner != 'None':
+                    owner_counts[owner] = owner_counts.get(owner, 0) + 1
+            
+            # Get top owners by count
+            top_owners = sorted(owner_counts.items(), key=lambda x: x[1], reverse=True)[:limit]
+            
+            # Extract just the owner names for dropdown
+            trending_authors = [owner for owner, count in top_owners]
+            
+            return trending_authors, top_owners
+        else:
+            # Fallback to API method if HTTP request fails
+            trending_models = list(api.list_models(sort="trending", limit=limit))
+            trending_datasets = list(api.list_datasets(sort="trending", limit=limit))
+            trending_spaces = list(api.list_spaces(sort="trending", limit=limit))
+            
+            # Extract unique authors
+            authors = set()
+            for item in trending_models + trending_datasets + trending_spaces:
+                if hasattr(item, "author"):
+                    authors.add(item.author)
+                elif hasattr(item, "id") and "/" in item.id:
+                    authors.add(item.id.split("/")[0])
+            
+            # Return sorted list of unique authors and empty stats
+            author_list = sorted(list(authors))[:limit]
+            return author_list, [(author, 0) for author in author_list[:30]]
     except Exception as e:
         st.error(f"Error fetching trending accounts: {str(e)}")
-        return ["ritvik77", "facebook", "google", "stabilityai", "Salesforce", "tiiuae", "bigscience"]
+        fallback_authors = ["ritvik77", "facebook", "google", "stabilityai", "Salesforce", "tiiuae", "bigscience"]
+        return fallback_authors, [(author, 0) for author in fallback_authors]
 
 
 # Rate limiting
@@ -207,51 +238,61 @@ with st.sidebar:
     
     # Fetch trending accounts with a loading spinner
     with st.spinner("Loading top trending accounts..."):
-        trending_accounts = get_trending_accounts(limit=100)
+        trending_accounts, top_owners = get_trending_accounts(limit=100)
     
-    # Create a tab interface for selection method
-    tab1, tab2 = st.tabs(["Top 100 Trending", "Custom User"])
+    # Show trending accounts list with search filter
+    st.subheader("🔥 Top Trending Accounts")
     
-    with tab1:
-        # Show trending accounts list with search filter
-        st.subheader("🔥 Top Trending Accounts")
-        search_filter = st.text_input("Filter accounts", key="trending_filter")
-        
-        # Filter accounts based on search
-        filtered_accounts = [acc for acc in trending_accounts if search_filter.lower() in acc.lower()] if search_filter else trending_accounts
-        
-        # Show account count
-        st.caption(f"Showing {len(filtered_accounts)} accounts")
-        
-        # Create a scrollable container for the accounts list
-        account_container = st.container()
-        with account_container:
-            selected_trending = st.selectbox(
-                "Select trending account",
-                options=filtered_accounts,
-                index=0 if filtered_accounts else None,
-                key="trending_selectbox"
-            )
+    # Add stats expander
+    with st.expander("View Top 30 Contributor Stats"):
+        # Create a bar chart for top 30 contributors
+        if top_owners:
+            chart_data = pd.DataFrame(top_owners[:30], columns=["Owner", "Spaces Count"])
             
-            if selected_trending:
-                username = selected_trending
+            fig, ax = plt.subplots(figsize=(10, 8))
+            bars = ax.barh(chart_data["Owner"], chart_data["Spaces Count"])
+            
+            # Add color gradient to bars
+            for i, bar in enumerate(bars):
+                bar.set_color(plt.cm.viridis(i/len(bars)))
+            
+            ax.set_title("Top 30 Contributors by Number of Spaces")
+            ax.set_xlabel("Number of Spaces")
+            plt.tight_layout()
+            st.pyplot(fig)
+    
+    # Search filter
+    search_filter = st.text_input("Filter accounts", key="trending_filter")
+    
+    # Filter accounts based on search
+    filtered_accounts = [acc for acc in trending_accounts if search_filter.lower() in acc.lower()] if search_filter else trending_accounts
+    
+    # Show account count
+    st.caption(f"Showing {len(filtered_accounts)} of {len(trending_accounts)} accounts")
     
-    with tab2:
-        st.subheader("🔎 Custom Account")
-        default_accounts = ["ritvik77", "facebook", "google", "stabilityai", "Salesforce", "tiiuae", "bigscience"]
-        custom_account = st.selectbox(
-            "Select or type a username",
-            options=default_accounts,
-            index=0
+    # Create a scrollable container for the accounts list
+    account_container = st.container()
+    with account_container:
+        selected_trending = st.selectbox(
+            "Select trending account",
+            options=filtered_accounts,
+            index=0 if filtered_accounts else None,
+            key="trending_selectbox"
         )
+        
+        # Custom account input option
         st.markdown("<div style='text-align: center; margin: 10px 0;'>OR</div>", unsafe_allow_html=True)
         custom = st.text_input("", placeholder="Enter custom username/org")
+        
+        # Set username based on selection or custom input
         if custom.strip():
             username = custom.strip()
-        elif tab2._active and not tab1._active:  # Only set if tab2 is active
-            username = custom_account
+        elif selected_trending:
+            username = selected_trending
+        else:
+            username = "facebook"  # Default fallback
     
-    # Year selection (outside tabs to always be visible)
+    # Year selection
     st.subheader("🗓️ Time Period")
     year_options = list(range(datetime.now().year, 2017, -1))
     selected_year = st.selectbox("Select Year", options=year_options)
@@ -266,6 +307,11 @@ with st.sidebar:
 st.title("🤗 Hugging Face Contributions")
 if username:
     with st.spinner(f"Fetching commit data for {username}..."):
+        # Display contributor rank if in top 100
+        if username in trending_accounts:
+            rank = trending_accounts.index(username) + 1
+            st.success(f"🏆 {username} is ranked #{rank} in the top trending contributors!")
+        
         # Create a dictionary to store commits by type
         commits_by_type = {}
         commit_counts_by_type = {}
@@ -342,6 +388,13 @@ if username:
         
         with profile_col2:
             st.metric("Total Commits", total_commits)
+            
+            # Show contributor rank if in top owners
+            for owner, count in top_owners:
+                if owner.lower() == username.lower():
+                    st.metric("Spaces Count", count)
+                    break
+            
             st.markdown(f"[View Profile on Hugging Face](https://huggingface.co/{username})")
 
         # Create DataFrame for all commits