Update app.py

app.py

@@ -322,25 +322,9 @@ def get_korea_models():
 def get_all_models(limit=1000):
     """모든 모델과 Korea 관련 모델 가져오기"""
     all_models = []
+    existing_ids = set()
     
-    # 1. 일반 모델 리스트 가져오기
-    params = {
-        "limit": limit,
-        "full": "True",
-        "config": "True"
-    }
-    
-    response = requests.get(
-        "https://huggingface.co/api/models",
-        headers={'Accept': 'application/json'},
-        params=params
-    )
-    
-    if response.status_code == 200:
-        all_models.extend(response.json())
-        print(f"Fetched {len(all_models)} general models")
-    
-    # 2. Korea 검색 결과 가져오기
+    # 1. Korea 검색 결과 가져오기
     korea_params = {
         "search": "korea",
         "full": "True",
@@ -356,16 +340,14 @@ def get_all_models(limit=1000):
     
     if korea_response.status_code == 200:
         korea_models = korea_response.json()
-        print(f"Fetched {len(korea_models)} Korea-related models")
-        
-        # 중복 제거하면서 Korea 모델 추가
-        existing_ids = {model.get('id', '') for model in all_models}
+        print(f"Found {len(korea_models)} Korea-related models")
         for model in korea_models:
-            if model.get('id', '') not in existing_ids:
+            model_id = model.get('id', '')
+            if model_id not in existing_ids:
                 all_models.append(model)
-                existing_ids.add(model.get('id', ''))
+                existing_ids.add(model_id)
     
-    # 3. Korean 검색 결과 가져오기
+    # 2. Korean 검색 결과 가져오기
     korean_params = {
         "search": "korean",
         "full": "True",
@@ -381,13 +363,33 @@ def get_all_models(limit=1000):
     
     if korean_response.status_code == 200:
         korean_models = korean_response.json()
-        print(f"Fetched {len(korean_models)} Korean-related models")
-        
-        # 중복 제거하면서 Korean 모델 추가
+        print(f"Found {len(korean_models)} Korean-related models")
         for model in korean_models:
-            if model.get('id', '') not in existing_ids:
+            model_id = model.get('id', '')
+            if model_id not in existing_ids:
                 all_models.append(model)
-                existing_ids.add(model.get('id', ''))
+                existing_ids.add(model_id)
+    
+    # 3. 일반 모델 리스트 가져오기
+    params = {
+        "limit": limit,
+        "full": "True",
+        "config": "True"
+    }
+    
+    response = requests.get(
+        "https://huggingface.co/api/models",
+        headers={'Accept': 'application/json'},
+        params=params
+    )
+    
+    if response.status_code == 200:
+        general_models = response.json()
+        for model in general_models:
+            model_id = model.get('id', '')
+            if model_id not in existing_ids:
+                all_models.append(model)
+                existing_ids.add(model_id)
     
     print(f"Total unique models: {len(all_models)}")
     return all_models[:limit]
@@ -476,12 +478,11 @@ def get_models_data(progress=gr.Progress()):
             likes = [m['likes'] for m in valid_models]
             downloads = [m['downloads'] for m in valid_models]
             
-            # Y축 값을 반전 (높은 순위가 위로 가도록)
-            y_values = [1001 - r for r in ranks]
-            
+            # 막대 그래프 생성 (각 순위에서 1000까지의 길이로 막대 생성)
             fig.add_trace(go.Bar(
                 x=ids,
-                y=y_values,
+                y=ranks,  # 실제 순위 사용
+                base=1000,  # 막대의 기준점을 1000으로 설정
                 text=[f"Global Rank: #{r}<br>Downloads: {format(d, ',')}<br>Likes: {format(l, ',')}" 
                       for r, d, l in zip(ranks, downloads, likes)],
                 textposition='auto',
@@ -498,10 +499,12 @@ def get_models_data(progress=gr.Progress()):
                     'yanchor': 'top'
                 },
                 xaxis_title='Model ID',
-                yaxis_title='Global Rank',
+                yaxis_title='Rank',
                 yaxis=dict(
-                    ticktext=[f"#{i}" for i in range(1, 1001, 50)],
-                    tickvals=[1001 - i for i in range(1, 1001, 50)],
+                    autorange='reversed',  # Y축을 반전
+                    tickmode='linear',
+                    tick0=0,
+                    dtick=50,
                     range=[0, 1000]
                 ),
                 height=800,