Update app.py

app.py

@@ -294,53 +294,54 @@ def get_models_data(progress=gr.Progress()):
     
     try:
         progress(0, desc="Fetching models data...")
-        params = {
-            'full': 'true',
-            'limit': 1000,
-            'sort': 'downloads',
-            'direction': -1
-        }
+        
+        # 여러 정렬 방식으로 모델 검색
+        all_found_models = []
+        sort_options = [
+            {'sort': 'downloads', 'direction': -1},
+            {'sort': 'lastModified', 'direction': -1},
+            {'sort': 'likes', 'direction': -1}
+        ]
         
         headers = {
             'Accept': 'application/json',
             'User-Agent': 'Mozilla/5.0'
         }
         
-        response = requests.get(url, params=params, headers=headers)
-        if response.status_code != 200:
-            print(f"API 요청 실패: {response.status_code}")
-            print(f"Response: {response.text}")
+        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()
+                all_found_models.extend(models)
+                print(f"Found {len(models)} models with {sort_params['sort']} sort")
+        
+        if not all_found_models:
+            print("No models found from API")
             return create_error_plot(), "<div>모델 데이터를 가져오는데 실패했습니다.</div>", pd.DataFrame()
         
-        all_models = response.json()
-        print(f"Total models fetched: {len(all_models)}")
+        # 중복 제거
+        unique_models = {model['id']: model for model in all_found_models}.values()
+        print(f"Total unique models: {len(unique_models)}")
         
-        # API 응답 확인을 위한 디버깅
-        print("First few model IDs from API:")
-        for model in all_models[:5]:
-            print(f"API Model ID: {model.get('id', 'No ID')}")
-        
-        print("\nTarget model IDs:")
-        for model_id in list(target_models.keys())[:5]:
-            print(f"Target Model ID: {model_id}")
-        
-        # 매칭 로직 수정
+        # target_models와 매칭
         filtered_models = []
-        for model in all_models:
-            model_id = model.get('id', '').strip()
-            for target_id in target_models:
-                if model_id.lower() == target_id.lower():  # 대소문자 구분 없이 비교
-                    model['rank'] = len(filtered_models) + 1
-                    filtered_models.append(model)
-                    print(f"Matched model: {model_id}")
-                    break
+        for model in unique_models:
+            if model['id'] in target_models:
+                model['rank'] = len(filtered_models) + 1
+                filtered_models.append(model)
+                print(f"Matched model: {model['id']}")
         
-        print(f"\nMatched {len(filtered_models)} models out of {len(target_models)} targets")
+        print(f"Matched {len(filtered_models)} models out of {len(target_models)} targets")
         
         if not filtered_models:
             return create_error_plot(), "<div>선택된 모델의 데이터를 찾을 수 없습니다.</div>", pd.DataFrame()
         
-        
         progress(0.3, desc="Creating visualization...")
         
         # 시각화 생성
@@ -351,7 +352,7 @@ def get_models_data(progress=gr.Progress()):
         downloads = [model.get('downloads', 0) for model in filtered_models]
         likes = [model.get('likes', 0) for model in filtered_models]
         
-        # 다운로드 수를 기준으로 순위 계산
+        # 다운로드 수를 기준으로 순위 재계산
         sorted_models = sorted(filtered_models, key=lambda x: x.get('downloads', 0), reverse=True)
         for idx, model in enumerate(sorted_models):
             model['rank'] = idx + 1
@@ -403,8 +404,8 @@ def get_models_data(progress=gr.Progress()):
         """
         
         for model in sorted_models:
-            model_id = model.get('id', '')
-            rank = model.get('rank', 'N/A')
+            model_id = model['id']
+            rank = model['rank']
             downloads = model.get('downloads', 0)
             likes = model.get('likes', 0)
             
@@ -439,11 +440,11 @@ def get_models_data(progress=gr.Progress()):
         
         # 데이터프레임 생성
         df = pd.DataFrame([{
-            'Rank': model.get('rank', 'N/A'),
-            'Model ID': model.get('id', ''),
+            'Rank': model['rank'],
+            'Model ID': model['id'],
             'Downloads': model.get('downloads', 'N/A'),
             'Likes': model.get('likes', 'N/A'),
-            'URL': target_models[model.get('id', '')]
+            'URL': target_models[model['id']]
         } for model in sorted_models])
         
         progress(1.0, desc="Complete!")