Update app.py

app.py

@@ -336,27 +336,153 @@ def get_trending_spaces(progress=gr.Progress()) -> Tuple[str, str]:
         error_html = f'<div style="color: red; padding: 20px;">Error: {str(e)}</div>'
         return error_html, f"Error: {str(e)}"
 
+def get_models(progress=gr.Progress()) -> Tuple[str, str]:
+    """인기 모델 가져오기"""
+    url = "https://huggingface.co/api/models"
+    
+    try:
+        progress(0, desc="Fetching models data...")
+        response = requests.get(url)
+        response.raise_for_status()
+        models = response.json()
+        
+        # 상위 10개만 선택
+        top_models = models[:10]
+        
+        progress(0.1, desc="Creating gallery...")
+        html_content = """
+        <div style='padding: 20px; background: #f5f5f5;'>
+            <div style='display: grid; grid-template-columns: repeat(auto-fill, minmax(300px, 1fr)); gap: 20px;'>
+        """
+        
+        for idx, model in enumerate(top_models):
+            model_id = model.get('id', '')
+            author = model_id.split('/')[0]
+            title = model_id.split('/')[-1]
+            likes = format(model.get('likes', 0), ',')
+            downloads = format(model.get('downloads', 0), ',')
+            created = model.get('createdAt', '').split('T')[0]
+            url = f"https://huggingface.co/{model_id}"
+            
+            screenshot = get_cached_screenshot(url)
+            html_content += get_space_card({"id": model_id, "likes": model.get('likes', 0), 
+                                          "sdk": "model", "createdAt": created}, idx)
+            
+            progress((0.1 + 0.9 * idx/10), desc=f"Loading model {idx+1}/10...")
+            
+        html_content += "</div></div>"
+        
+        progress(1.0, desc="Complete!")
+        return html_content, "Models gallery refresh complete!"
+        
+    except Exception as e:
+        error_html = f'<div style="color: red; padding: 20px;">Error: {str(e)}</div>'
+        return error_html, f"Error: {str(e)}"
+
+def get_datasets(progress=gr.Progress()) -> Tuple[str, str]:
+    """인기 데이터셋 가져오기"""
+    url = "https://huggingface.co/api/datasets"
+    
+    try:
+        progress(0, desc="Fetching datasets data...")
+        response = requests.get(url)
+        response.raise_for_status()
+        datasets = response.json()
+        
+        # 상위 10개만 선택
+        top_datasets = datasets[:10]
+        
+        progress(0.1, desc="Creating gallery...")
+        html_content = """
+        <div style='padding: 20px; background: #f5f5f5;'>
+            <div style='display: grid; grid-template-columns: repeat(auto-fill, minmax(300px, 1fr)); gap: 20px;'>
+        """
+        
+        for idx, dataset in enumerate(top_datasets):
+            dataset_id = dataset.get('id', '')
+            author = dataset_id.split('/')[0]
+            title = dataset_id.split('/')[-1]
+            likes = format(dataset.get('likes', 0), ',')
+            downloads = format(dataset.get('downloads', 0), ',')
+            created = dataset.get('createdAt', '').split('T')[0]
+            url = f"https://huggingface.co/datasets/{dataset_id}"
+            
+            screenshot = get_cached_screenshot(url)
+            html_content += get_space_card({"id": dataset_id, "likes": dataset.get('likes', 0), 
+                                          "sdk": "dataset", "createdAt": created}, idx)
+            
+            progress((0.1 + 0.9 * idx/10), desc=f"Loading dataset {idx+1}/10...")
+            
+        html_content += "</div></div>"
+        
+        progress(1.0, desc="Complete!")
+        return html_content, "Datasets gallery refresh complete!"
+        
+    except Exception as e:
+        error_html = f'<div style="color: red; padding: 20px;">Error: {str(e)}</div>'
+        return error_html, f"Error: {str(e)}"
+
 def create_interface():
     """Gradio 인터페이스 생성"""
-    with gr.Blocks(title="Hugging Face Trending Spaces") as interface:
-        gr.Markdown("# 🤗 Hugging Face Top 10 Trending Spaces")
-        gr.Markdown("Shows top 10 most liked spaces on Hugging Face")
-        
-        with gr.Row():
-            refresh_btn = gr.Button("Refresh Gallery", variant="primary")
+    with gr.Blocks(title="Hugging Face Trending Board") as interface:
+        gr.Markdown("# 🤗 Hugging Face Trending Board")
         
-        gallery_html = gr.HTML()
-        status = gr.Markdown("Ready")
+        with gr.Tabs() as tabs:
+            # Spaces 탭
+            with gr.Tab("🎯 Trending Spaces"):
+                gr.Markdown("Shows top 10 trending spaces on Hugging Face")
+                with gr.Row():
+                    spaces_refresh_btn = gr.Button("Refresh Spaces", variant="primary")
+                spaces_gallery = gr.HTML()
+                spaces_status = gr.Markdown("Ready")
+            
+            # Models 탭
+            with gr.Tab("🤖 Trending Models"):
+                gr.Markdown("Shows top 10 trending models on Hugging Face")
+                with gr.Row():
+                    models_refresh_btn = gr.Button("Refresh Models", variant="primary")
+                models_gallery = gr.HTML()
+                models_status = gr.Markdown("Ready")
+            
+            # Datasets 탭
+            with gr.Tab("📊 Trending Datasets"):
+                gr.Markdown("Shows top 10 trending datasets on Hugging Face")
+                with gr.Row():
+                    datasets_refresh_btn = gr.Button("Refresh Datasets", variant="primary")
+                datasets_gallery = gr.HTML()
+                datasets_status = gr.Markdown("Ready")
         
-        refresh_btn.click(
+        # Event handlers
+        spaces_refresh_btn.click(
             fn=get_trending_spaces,
-            outputs=[gallery_html, status],
+            outputs=[spaces_gallery, spaces_status],
+            show_progress=True
+        )
+        
+        models_refresh_btn.click(
+            fn=get_models,
+            outputs=[models_gallery, models_status],
             show_progress=True
         )
         
+        datasets_refresh_btn.click(
+            fn=get_datasets,
+            outputs=[datasets_gallery, datasets_status],
+            show_progress=True
+        )
+        
+        # 초기 로드
         interface.load(
             fn=get_trending_spaces,
-            outputs=[gallery_html, status]
+            outputs=[spaces_gallery, spaces_status]
+        )
+        interface.load(
+            fn=get_models,
+            outputs=[models_gallery, models_status]
+        )
+        interface.load(
+            fn=get_datasets,
+            outputs=[datasets_gallery, datasets_status]
         )
     
     return interface