Update app.py

app.py

@@ -44,72 +44,47 @@ target_spaces = {
 
 def get_trending_spaces():
     try:
-        # API 엔드포인트 수정
-        url = "https://huggingface.co/api/spaces"  # 기본 URL
+        # 트렌딩 API 엔드포인트 수정
+        url = "https://huggingface.co/api/spaces/trending"
         
         headers = {
             'Authorization': f'Bearer {HF_TOKEN}',
-            'Accept': 'application/json'
+            'Accept': 'application/json',
+            'User-Agent': 'Mozilla/5.0'
         }
         
+        # 모든 결과를 가져오기 위해 limit 증가
         params = {
-            'sort': 'likes',  # 'trending' 대신 'likes'로 변경
-            'direction': -1,  # 내림차순 정렬
-            'limit': 1000,
-            'full': 'true'
+            'limit': 5000,  # 최대한 많은 결과를 가져오기
+            'interval': 'day'  # 일간 트렌딩
         }
         
         response = requests.get(url, headers=headers, params=params)
         
         if response.status_code == 200:
-            return response.json()
+            data = response.json()
+            # target_spaces에 있는 스페이스만 필터링
+            filtered_data = []
+            for space in data:
+                space_id = space.get('id', '')
+                if space_id in target_spaces:
+                    filtered_data.append(space)
+            return filtered_data
         else:
-            # 상세한 디버깅 정보 출력
             print(f"API 요청 실패: {response.status_code}")
             print(f"Response: {response.text}")
-            print(f"URL: {response.url}")  # 실제 요청된 URL 출력
-            print(f"Headers: {headers}")
-            print(f"Params: {params}")
             return None
     except Exception as e:
         print(f"API 호출 중 에러 발생: {str(e)}")
         return None
 
-# 토큰이 없는 경우를 위한 대체 함수
-def get_trending_spaces_without_token():
-    try:
-        url = "https://huggingface.co/api/spaces"
-        params = {
-            'sort': 'likes',
-            'direction': -1,
-            'limit': 1000,
-            'full': 'true'
-        }
-        
-        response = requests.get(url, params=params)
-        
-        if response.status_code == 200:
-            return response.json()
-        else:
-            print(f"API 요청 실패 (토큰 없음): {response.status_code}")
-            print(f"Response: {response.text}")
-            return None
-    except Exception as e:
-        print(f"API 호출 중 에러 발생 (토큰 없음): {str(e)}")
-        return None
-
-# API 토큰 설정 및 함수 선택
-if not HF_TOKEN:
-    get_trending_spaces = get_trending_spaces_without_token
-
-# create_trend_visualization 함수 수정
 def create_trend_visualization(spaces_data):
     if not spaces_data:
         return create_error_plot()
     
     fig = go.Figure()
     
-    # 타겟 스페이스들의 순위 찾기
+    # 순위 데이터 준비
     ranks = []
     for idx, space in enumerate(spaces_data, 1):
         space_id = space.get('id', '')
@@ -118,7 +93,8 @@ def create_trend_visualization(spaces_data):
                 'id': space_id,
                 'rank': idx,
                 'likes': space.get('likes', 0),
-                'title': space.get('title', 'N/A')
+                'title': space.get('title', 'N/A'),
+                'views': space.get('views', 0)
             })
     
     if not ranks:
@@ -131,39 +107,66 @@ def create_trend_visualization(spaces_data):
     ids = [r['id'] for r in ranks]
     rank_values = [r['rank'] for r in ranks]
     likes = [r['likes'] for r in ranks]
-    titles = [r['title'] for r in ranks]
+    views = [r['views'] for r in ranks]
     
-    # 막대 그래프 생성 (개선된 스타일)
+    # 막대 그래프 생성
     fig.add_trace(go.Bar(
         x=ids,
         y=rank_values,
-        text=[f"Rank: {r}<br>Likes: {l}<br>{t}" for r, l, t in zip(rank_values, likes, titles)],
+        text=[f"Rank: {r}<br>Likes: {l}<br>Views: {v}" for r, l, v in zip(rank_values, likes, views)],
         textposition='auto',
         marker_color='rgb(158,202,225)',
-        opacity=0.8,
-        hoverinfo='text'
+        opacity=0.8
     ))
     
     fig.update_layout(
         title={
-            'text': 'Hugging Face Spaces Rankings',
+            'text': 'Current Trending Ranks (All Target Spaces)',
             'y':0.95,
             'x':0.5,
             'xanchor': 'center',
             'yanchor': 'top'
         },
         xaxis_title='Space ID',
-        yaxis_title='Rank',
+        yaxis_title='Trending Rank',
         yaxis_autorange='reversed',
         height=800,
         showlegend=False,
         template='plotly_white',
-        margin=dict(l=50, r=50, t=100, b=100),
         xaxis_tickangle=-45
     )
     
     return fig
 
+# 토큰이 없는 경우를 위한 대체 함수
+def get_trending_spaces_without_token():
+    try:
+        url = "https://huggingface.co/api/spaces"
+        params = {
+            'sort': 'likes',
+            'direction': -1,
+            'limit': 1000,
+            'full': 'true'
+        }
+        
+        response = requests.get(url, params=params)
+        
+        if response.status_code == 200:
+            return response.json()
+        else:
+            print(f"API 요청 실패 (토큰 없음): {response.status_code}")
+            print(f"Response: {response.text}")
+            return None
+    except Exception as e:
+        print(f"API 호출 중 에러 발생 (토큰 없음): {str(e)}")
+        return None
+
+# API 토큰 설정 및 함수 선택
+if not HF_TOKEN:
+    get_trending_spaces = get_trending_spaces_without_token
+
+
+
 def create_error_plot():
     fig = go.Figure()
     fig.add_annotation(
@@ -182,20 +185,21 @@ def create_error_plot():
     return fig
 
 
-
 def create_space_info_html(spaces_data):
     if not spaces_data:
         return "<div style='padding: 20px;'><h2>데이터를 불러오는데 실패했습니다.</h2></div>"
     
     html_content = """
     <div style='padding: 20px;'>
-    <h2 style='color: #2c3e50;'>Current Rankings</h2>
+    <h2 style='color: #2c3e50;'>Current Trending Rankings</h2>
     <div style='display: grid; grid-template-columns: repeat(auto-fill, minmax(300px, 1fr)); gap: 20px;'>
     """
     
-    for idx, space in enumerate(spaces_data, 1):
-        space_id = space.get('id', '')
-        if space_id in target_spaces:
+    # 모든 target spaces를 포함하도록 수정
+    for space_id in target_spaces.keys():
+        space_info = next((s for s in spaces_data if s.get('id') == space_id), None)
+        if space_info:
+            rank = next((idx for idx, s in enumerate(spaces_data, 1) if s.get('id') == space_id), 'N/A')
             html_content += f"""
             <div style='
                 background: white;
@@ -203,12 +207,12 @@ def create_space_info_html(spaces_data):
                 border-radius: 10px;
                 box-shadow: 0 2px 4px rgba(0,0,0,0.1);
                 transition: transform 0.2s;
-                hover: transform: translateY(-5px);
             '>
-                <h3 style='color: #34495e;'>#{idx} - {space_id}</h3>
-                <p style='color: #7f8c8d;'>👍 Likes: {space.get('likes', 'N/A')}</p>
-                <p style='color: #2c3e50;'>{space.get('title', 'N/A')}</p>
-                <p style='color: #7f8c8d; font-size: 0.9em;'>{space.get('description', 'N/A')[:100]}...</p>
+                <h3 style='color: #34495e;'>#{rank} - {space_id}</h3>
+                <p style='color: #7f8c8d;'>👍 Likes: {space_info.get('likes', 'N/A')}</p>
+                <p style='color: #7f8c8d;'>👀 Views: {space_info.get('views', 'N/A')}</p>
+                <p style='color: #2c3e50;'>{space_info.get('title', 'N/A')}</p>
+                <p style='color: #7f8c8d; font-size: 0.9em;'>{space_info.get('description', 'N/A')[:100]}...</p>
                 <a href='{target_spaces[space_id]}' 
                    target='_blank' 
                    style='
@@ -219,7 +223,30 @@ def create_space_info_html(spaces_data):
                     text-decoration: none;
                     border-radius: 5px;
                     transition: background 0.3s;
-                    hover: background: #2980b9;
+                   '>
+                   Visit Space 🔗
+                </a>
+            </div>
+            """
+        else:
+            html_content += f"""
+            <div style='
+                background: #f8f9fa;
+                padding: 20px;
+                border-radius: 10px;
+                box-shadow: 0 2px 4px rgba(0,0,0,0.1);
+            '>
+                <h3 style='color: #34495e;'>{space_id}</h3>
+                <p style='color: #7f8c8d;'>Not in trending</p>
+                <a href='{target_spaces[space_id]}' 
+                   target='_blank' 
+                   style='
+                    display: inline-block;
+                    padding: 8px 16px;
+                    background: #95a5a6;
+                    color: white;
+                    text-decoration: none;
+                    border-radius: 5px;
                    '>
                    Visit Space 🔗
                 </a>