Update app.py

app.py

@@ -50,13 +50,18 @@ def cached_list_items(username, kind):
 @lru_cache(maxsize=1)
 def get_trending_accounts(limit=100):
     try:
-        trending_data = {"spaces": []}
+        trending_data = {"spaces": [], "models": []}
         
         # Get spaces for stats calculation
         spaces_response = requests.get("https://huggingface.co/api/spaces", 
                                       params={"limit": 10000}, 
                                       timeout=30)
         
+        # Get models for stats calculation
+        models_response = requests.get("https://huggingface.co/api/models", 
+                                      params={"limit": 10000}, 
+                                      timeout=30)
+        
         # Process spaces data
         spaces_owners = []
         if spaces_response.status_code == 200:
@@ -78,14 +83,48 @@ def get_trending_accounts(limit=100):
             trending_data["spaces"] = top_owners_spaces
             spaces_owners = [owner for owner, _ in top_owners_spaces]
         
-        # Use the spaces ranking as the overall trending
-        trending_authors = spaces_owners
+        # Process models data
+        models_owners = []
+        if models_response.status_code == 200:
+            models = models_response.json()
+            
+            # Count models by owner
+            owner_counts_models = {}
+            for model in models:
+                if '/' in model.get('id', ''):
+                    owner, _ = model.get('id', '').split('/', 1)
+                else:
+                    owner = model.get('owner', '')
+                
+                if owner != 'None':
+                    owner_counts_models[owner] = owner_counts_models.get(owner, 0) + 1
+            
+            # Get top owners by count for models
+            top_owners_models = sorted(owner_counts_models.items(), key=lambda x: x[1], reverse=True)[:limit]
+            trending_data["models"] = top_owners_models
+            models_owners = [owner for owner, _ in top_owners_models]
+        
+        # Combine rankings for overall trending based on appearance in both lists
+        combined_score = {}
+        for i, owner in enumerate(spaces_owners):
+            if owner not in combined_score:
+                combined_score[owner] = 0
+            combined_score[owner] += (limit - i)  # Higher rank gives more points
+            
+        for i, owner in enumerate(models_owners):
+            if owner not in combined_score:
+                combined_score[owner] = 0
+            combined_score[owner] += (limit - i)  # Higher rank gives more points
+        
+        # Sort by combined score
+        sorted_combined = sorted(combined_score.items(), key=lambda x: x[1], reverse=True)[:limit]
+        trending_authors = [owner for owner, _ in sorted_combined]
         
-        return trending_authors, trending_data["spaces"], []  # Empty list for models as we don't need it
+        return trending_authors, trending_data["spaces"], trending_data["models"]
     except Exception as e:
         st.error(f"Error fetching trending accounts: {str(e)}")
         fallback_authors = ["ritvik77", "facebook", "google", "stabilityai", "Salesforce", "tiiuae", "bigscience"]
-        return fallback_authors, [(author, 0) for author in fallback_authors], []
+        return fallback_authors, [(author, 0) for author in fallback_authors], [(author, 0) for author in fallback_authors]
 
 
 # Rate limiting
@@ -239,10 +278,12 @@ with st.spinner("Loading trending accounts..."):
 with st.sidebar:
     st.title("👤 Contributor")
     
-    # Create tabs for Spaces rankings only
-    tab1, tab2 = st.tabs([
+    # Create tabs for Spaces and Models rankings
+    tab1, tab2, tab3, tab4 = st.tabs([
         "SPACES TOP 30", 
-        "SPACES TOP 100"
+        "SPACES TOP 100", 
+        "MODELS TOP 30", 
+        "MODELS TOP 100"
     ])
     
     with tab1:
@@ -410,6 +451,24 @@ if username:
                     spaces_rank = i+1
                     st.info(f"🚀 Spaces Ranking: #{spaces_rank} with {count} spaces")
                     break
+            
+            # Find user in models ranking
+            models_rank = None
+            for i, (owner, count) in enumerate(top_owners_models):
+                if owner == username:
+                    models_rank = i+1
+                    st.info(f"🧠 Models Ranking: #{models_rank} with {count} models")
+                    break
+                    
+            # Display combined ranking info
+            combined_info = []
+            if spaces_rank and spaces_rank <= 100:
+                combined_info.append(f"Spaces: #{spaces_rank}")
+            if models_rank and models_rank <= 100:
+                combined_info.append(f"Models: #{models_rank}")
+                
+            if combined_info:
+                st.success(f"Combined Rankings (Top 100): {', '.join(combined_info)}")
         
         # Create a dictionary to store commits by type
         commits_by_type = {}