Update app.py

app.py

@@ -18,6 +18,7 @@ target_models = {
     
     "moreh/Llama-3-Motif-102B-Instruct": "https://huggingface.co/moreh/Llama-3-Motif-102B-Instruct",
     "moreh/Llama-3-Motif-102B": "https://huggingface.co/moreh/Llama-3-Motif-102B",
+    "Samsung/TinyClick": "https://huggingface.co/Samsung/TinyClick",
     
     "Saxo/Linkbricks-Horizon-AI-Korean-Gemma-2-sft-dpo-27B": "https://huggingface.co/Saxo/Linkbricks-Horizon-AI-Korean-Gemma-2-sft-dpo-27B",
     "AALF/gemma-2-27b-it-SimPO-37K": "https://huggingface.co/AALF/gemma-2-27b-it-SimPO-37K",
@@ -538,38 +539,44 @@ target_spaces = {
     "VIDraft/Mouse-Hackathon": "https://huggingface.co/spaces/VIDraft/Mouse-Hackathon",
     "upstage/open-ko-llm-leaderboard": "https://huggingface.co/spaces/upstage/open-ko-llm-leaderboard",
     "LGAI-EXAONE/EXAONE-3.5-Instruct-Demo": "https://huggingface.co/spaces/LGAI-EXAONE/EXAONE-3.5-Instruct-Demo",
+
+    "kolaslab/RC4-EnDecoder": "https://huggingface.co/spaces/kolaslab/RC4-EnDecoder",
+    "kolaslab/simulator": "https://huggingface.co/spaces/kolaslab/simulator",
+    "kolaslab/calculator": "https://huggingface.co/spaces/kolaslab/calculator",
+    "etri-vilab/Ko-LLaVA": "https://huggingface.co/spaces/etri-vilab/Ko-LLaVA",
+    "etri-vilab/KOALA": "https://huggingface.co/spaces/etri-vilab/KOALA",
+    "naver-clova-ix/donut-base-finetuned-cord-v2": "https://huggingface.co/spaces/naver-clova-ix/donut-base-finetuned-cord-v2",    
+    
     "NCSOFT/VARCO_Arena": "https://huggingface.co/spaces/NCSOFT/VARCO_Arena"
 }
 
 def get_spaces_data(sort_type="trending", progress=gr.Progress()):
     """스페이스 데이터 가져오기 (trending 또는 modes)"""
-    url = f"https://huggingface.co/api/spaces"
-    params = {
-        'full': 'true',
-        'limit': 300
-    }
-    
-    if sort_type == "modes":
-        params['sort'] = 'likes'  # modes는 좋아요 순으로 정렬
+    url = "https://huggingface.co/api/spaces"
     
     try:
         progress(0, desc=f"Fetching {sort_type} spaces data...")
+        params = {
+            'full': 'true',
+            'limit': 300
+        }
+        
         response = requests.get(url, params=params)
         response.raise_for_status()
         all_spaces = response.json()
         
         # 순위 정보 저장
-        space_ranks = {space['id']: idx + 1 for idx, space in enumerate(all_spaces)}
-        
-        # target_spaces 필터링 및 순위 정보 포함
-        spaces = []
-        for space in all_spaces:
-            if space.get('id', '') in target_spaces:
-                space['rank'] = space_ranks.get(space['id'], 'N/A')
-                spaces.append(space)
+        space_ranks = {}
+        for idx, space in enumerate(all_spaces, 1):
+            space_id = space.get('id', '')
+            if space_id in target_spaces:
+                # 전체 space 정보 저장
+                space['rank'] = idx
+                space_ranks[space_id] = space
         
-        # 순위별로 정렬
-        spaces.sort(key=lambda x: x['rank'])
+        # target_spaces 중 순위권 내 space 필터링
+        spaces = [space_ranks[space_id] for space_id in space_ranks.keys()]
+        spaces.sort(key=lambda x: x['rank'])  # 순위별로 정렬
         
         progress(0.3, desc="Creating visualization...")
         
@@ -580,15 +587,17 @@ def get_spaces_data(sort_type="trending", progress=gr.Progress()):
         ids = [space['id'] for space in spaces]
         ranks = [space['rank'] for space in spaces]
         likes = [space.get('likes', 0) for space in spaces]
+        titles = [space.get('title', 'No Title') for space in spaces]
         
-        # Y축 값을 반전 (300 - rank + 1)
-        y_values = [301 - r for r in ranks]  # 순위를 반전된 값으로 변환
+        # Y축 값을 반전
+        y_values = [301 - r for r in ranks]
         
         # 막대 그래프 생성
         fig.add_trace(go.Bar(
             x=ids,
             y=y_values,
-            text=[f"Rank: {r}<br>Likes: {l}" for r, l in zip(ranks, likes)],
+            text=[f"Rank: {r}<br>Title: {t}<br>Likes: {l}" 
+                  for r, t, l in zip(ranks, titles, likes)],
             textposition='auto',
             marker_color='rgb(158,202,225)',
             opacity=0.8
@@ -605,9 +614,9 @@ def get_spaces_data(sort_type="trending", progress=gr.Progress()):
             xaxis_title='Space ID',
             yaxis_title='Rank',
             yaxis=dict(
-                ticktext=[str(i) for i in range(1, 301, 20)],  # 1부터 300까지 20 간격
-                tickvals=[301 - i for i in range(1, 301, 20)],  # 반전된 값
-                range=[0, 300]  # y축 범위 설정
+                ticktext=[str(i) for i in range(1, 301, 20)],
+                tickvals=[301 - i for i in range(1, 301, 20)],
+                range=[0, 300]
             ),
             height=800,
             showlegend=False,
@@ -625,11 +634,19 @@ def get_spaces_data(sort_type="trending", progress=gr.Progress()):
         """
         
         for space in spaces:
-            space_id = space.get('id', '')
-            rank = space.get('rank', 'N/A')
-            likes = space.get('likes', 0)
+            space_id = space['id']
+            rank = space['rank']
             title = space.get('title', 'No Title')
-            description = space.get('description', 'No Description')[:100]
+            likes = space.get('likes', 0)
+            description = space.get('description', '')
+            
+            # cardData에서 추가 정보 가져오기
+            card_data = space.get('cardData', {})
+            if not description and card_data:
+                description = card_data.get('description', 'No Description')
+            
+            # description이 너무 길면 자르기
+            short_description = description[:150] + '...' if description and len(description) > 150 else description
             
             html_content += f"""
             <div style='
@@ -640,9 +657,9 @@ def get_spaces_data(sort_type="trending", progress=gr.Progress()):
                 transition: transform 0.2s;
             '>
                 <h3 style='color: #34495e;'>Rank #{rank} - {space_id}</h3>
-                <p style='color: #7f8c8d;'>👍 Likes: {likes}</p>
-                <p style='color: #2c3e50;'>{title}</p>
-                <p style='color: #7f8c8d; font-size: 0.9em;'>{description}...</p>
+                <h4 style='color: #2c3e50; margin: 10px 0; font-size: 1.1em;'>{title}</h4>
+                <p style='color: #7f8c8d; margin-bottom: 10px;'>👍 Likes: {likes}</p>
+                <p style='color: #7f8c8d; font-size: 0.9em; margin-bottom: 15px; line-height: 1.4;'>{short_description}</p>
                 <a href='{target_spaces[space_id]}' 
                    target='_blank' 
                    style='
@@ -663,24 +680,24 @@ def get_spaces_data(sort_type="trending", progress=gr.Progress()):
         
         # 데이터프레임 생성
         df = pd.DataFrame([{
-            'Rank': space.get('rank', 'N/A'),
-            'Space ID': space.get('id', ''),
-            'Likes': space.get('likes', 'N/A'),
-            'Title': space.get('title', 'N/A'),
-            'URL': target_spaces[space.get('id', '')]
+            'Rank': space['rank'],
+            'Space ID': space['id'],
+            'Title': space.get('title', 'No Title'),
+            'Description': (space.get('description', '') or space.get('cardData', {}).get('description', 'No Description'))[:100] + '...',
+            'Likes': space.get('likes', 0),
+            'URL': target_spaces[space['id']]
         } for space in spaces])
         
         progress(1.0, desc="Complete!")
         return fig, html_content, df
         
     except Exception as e:
+        print(f"Error in get_spaces_data: {str(e)}")
         error_html = f'<div style="color: red; padding: 20px;">Error: {str(e)}</div>'
         error_plot = create_error_plot()
         return error_plot, error_html, pd.DataFrame()
 
 
-
-
 def create_trend_visualization(spaces_data):
     if not spaces_data:
         return create_error_plot()