Update app.py

app.py

@@ -23,9 +23,7 @@ def init_huggingface(token: str):
     except Exception as e:
         print(f"Error logging in: {str(e)}")
         return False
-        
-# First, define format_link before it's used in search_hub
-def format_link(item: Dict, number: int, search_type: str) -> str:
+        def format_link(item: Dict, number: int, search_type: str) -> str:
     """Format a link for display in the UI."""
     link = item['link']
     readme_link = f"{link}/blob/main/README.md"
@@ -45,6 +43,40 @@ def format_link(item: Dict, number: int, search_type: str) -> str:
     """
     return html
 
+def display_results(df: pd.DataFrame):
+    """Display search results in HTML format."""
+    if df is not None and not df.empty:
+        html = "<div style='max-height: 400px; overflow-y: auto;'>"
+        for _, row in df.iterrows():
+            html += row['formatted_link']
+        html += "</div>"
+        return html
+    else:
+        return "<p>No results found.</p>"
+
+def SwarmyTime(data: List[Dict]) -> Dict:
+    """Aggregates all content from the given data."""
+    aggregated = {
+        "total_items": len(data),
+        "unique_authors": set(),
+        "total_downloads": 0,
+        "item_types": {"Models": 0, "Datasets": 0, "Spaces": 0}
+    }
+
+    for item in data:
+        aggregated["unique_authors"].add(item.get("author", "Unknown"))
+        aggregated["total_downloads"] += item.get("downloads", 0)
+        
+        if "modelId" in item:
+            aggregated["item_types"]["Models"] += 1
+        elif "dataset" in item.get("id", ""):
+            aggregated["item_types"]["Datasets"] += 1
+        else:
+            aggregated["item_types"]["Spaces"] += 1
+
+    aggregated["unique_authors"] = len(aggregated["unique_authors"])
+    return aggregated
+
 def search_hub(query: str, search_type: str, token: str = None) -> pd.DataFrame:
     """Search the Hugging Face Hub for models, datasets, or spaces."""
     api = HfApi(token=token)
@@ -176,7 +208,7 @@ def create_repo_zip(data: List[Dict], search_type: str, token: str) -> tuple[str
         
         return download_link, f"Successfully downloaded {len(successful_downloads)} repositories"
 
-# Rest of the functions remain the same (display_results, SwarmyTime)...
+
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("""