Update app.py

app.py

@@ -295,7 +295,11 @@ target_models = {
 
 def get_models_data(progress=gr.Progress()):
     """모델 데이터 가져오기"""
-    url = "https://huggingface.co/api/models"  # 기본 모델 API 엔드포인트
+    def normalize_model_id(model_id):
+        """모델 ID를 정규화"""
+        return model_id.strip().lower()
+        
+    url = "https://huggingface.co/api/models"
     
     try:
         progress(0, desc="Fetching models data...")
@@ -324,38 +328,36 @@ def get_models_data(progress=gr.Progress()):
         
         # 중복 제거
         seen_ids = set()
-        unique_models = []
+        filtered_models = []
         for model in all_found_models:
-            model_id = model.get('id', '')
-            if model_id not in seen_ids and model_id in target_models:
+            model_id = normalize_model_id(model.get('id', ''))
+            if model_id not in seen_ids and model_id in [normalize_model_id(tid) for tid in target_models.keys()]:
                 seen_ids.add(model_id)
-                unique_models.append(model)
+                filtered_models.append(model)
         
         # 다운로드 수로 정렬
-        unique_models.sort(key=lambda x: x.get('downloads', 0), reverse=True)
+        filtered_models.sort(key=lambda x: x.get('downloads', 0), reverse=True)
         
         # 순위 할당
-        for idx, model in enumerate(unique_models, 1):
+        for idx, model in enumerate(filtered_models, 1):
             model['rank'] = idx
         
-        if not unique_models:
+        if not filtered_models:
             return create_error_plot(), "<div>선택된 모델의 데이터를 찾을 수 없습니다.</div>", pd.DataFrame()
         
-        # 나머지 시각화 코드는 동일하게 유지
         progress(0.3, desc="Creating visualization...")
         
         # 시각화 생성
         fig = go.Figure()
         
         # 데이터 준비
-        ids = [model['id'] for model in unique_models]
-        ranks = [model['rank'] for model in unique_models]
-        likes = [model.get('likes', 0) for model in unique_models]
-        downloads = [model.get('downloads', 0) for model in unique_models]
+        ids = [model['id'] for model in filtered_models]
+        ranks = [model['rank'] for model in filtered_models]
+        likes = [model.get('likes', 0) for model in filtered_models]
+        downloads = [model.get('downloads', 0) for model in filtered_models]
         
         # Y축 값을 반전
         y_values = [1001 - r for r in ranks]
-
         
         # 막대 그래프 생성
         fig.add_trace(go.Bar(
@@ -492,60 +494,10 @@ def get_models_data(progress=gr.Progress()):
         print(f"Error in get_models_data: {str(e)}")
         return create_error_plot(), f"<div>에러 발생: {str(e)}</div>", pd.DataFrame()
         
-# API 응답 형식에 맞게 모델 ID 수정
-def normalize_model_id(model_id):
-    """모델 ID를 정규화"""
-    return model_id.strip().lower()
-
-# 매칭 로직 수정
-filtered_models = []
-for model in all_models:
-    model_id = normalize_model_id(model.get('id', ''))
-    target_id = next(
-        (tid for tid in target_models.keys() 
-         if normalize_model_id(tid) == model_id),
-        None
-    )
-    
-    if target_id:
-        model['rank'] = len(filtered_models) + 1
-        filtered_models.append(model)
-        print(f"Matched model: {model_id} with target: {target_id}")
-
-print(f"\nMatched {len(filtered_models)} models out of {len(target_models)} targets")
-
-def try_different_sorts():
-    """다양한 정렬 방식으로 모델 검색"""
-    sort_options = [
-        {'sort': 'downloads', 'direction': -1},
-        {'sort': 'lastModified', 'direction': -1},
-        {'sort': 'likes', 'direction': -1}
-    ]
-    
-    all_found_models = set()
-    for sort_params in sort_options:
-        params = {
-            'full': 'true',
-            'limit': 1000,
-            **sort_params
-        }
-        
-        response = requests.get(url, params=params, headers=headers)
-        if response.status_code == 200:
-            models = response.json()
-            for model in models:
-                model_id = normalize_model_id(model.get('id', ''))
-                if model_id in [normalize_model_id(tid) for tid in target_models.keys()]:
-                    all_found_models.add(model_id)
-    
-    return all_found_models
-
-# 메인 함수에서 사용
-all_found_models = try_different_sorts()
-print(f"\nTotal unique models found across all sorts: {len(all_found_models)}")        
-        
 # 관심 스페이스 URL 리스트와 정보
 target_spaces = {
+
+    "openfree/Korean-Leaderboard": "https://huggingface.co/spaces/openfree/Korean-Leaderboard",
     "ginipick/FLUXllama": "https://huggingface.co/spaces/ginipick/FLUXllama",
     "ginipick/SORA-3D": "https://huggingface.co/spaces/ginipick/SORA-3D",
     "fantaxy/Sound-AI-SFX": "https://huggingface.co/spaces/fantaxy/Sound-AI-SFX",