Update app.py

app.py

@@ -23,8 +23,30 @@ 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:
+    """Format a link for display in the UI."""
+    link = item['link']
+    readme_link = f"{link}/blob/main/README.md"
+    title = f"{number}. {item['id']}"
+    
+    metadata = f"Author: {item['author']}"
+    if 'downloads' in item:
+        metadata += f", Downloads: {item['downloads']}"
+    
+    html = f"""
+    <div style="margin-bottom: 10px;">
+        <strong>{title}</strong><br>
+        <a href="{link}" target="_blank" style="color: #4a90e2; text-decoration: none;">View {search_type[:-1]}</a> | 
+        <a href="{readme_link}" target="_blank" style="color: #4a90e2; text-decoration: none;">View README</a><br>
+        <small>{metadata}</small>
+    </div>
+    """
+    return html
 
 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)
     if search_type == "Models":
         results = api.list_models(search=query)