Update app.py

app.py

@@ -444,30 +444,34 @@ def get_trending_spaces(search_query="", sort_by="rank", progress=gr.Progress())
         progress(0, desc="Fetching spaces data...")
         params = {
             'full': 'true',
-            'limit': 1000,  # 검색을 위해 더 많은 항목 로드
-            'search': search_query
+            'limit': 10  # 기본 300개 유지
         }
 
         response = requests.get(url, params=params)
         response.raise_for_status()
         spaces = response.json()
         
-        # 검색어로 필터링
+        # 검색어로 필터링 (검색어가 있는 경우에만)
         if search_query:
-            spaces = [space for space in spaces if search_query.lower() in 
-                     (space.get('id', '') + space.get('title', '')).lower()]
-        
-        # 정렬
-        if sort_by == "rising_rate":
-            spaces.sort(key=lambda x: calculate_rising_rate(x.get('createdAt', ''), 0), reverse=True)
-        elif sort_by == "popularity":
-            spaces.sort(key=lambda x: get_popularity_grade(
-                int(str(x.get('likes', '0')).replace(',', '')),
-                calculate_rising_rate(x.get('createdAt', ''), 0))[1],
-                reverse=True)
-        
-        # 상위 300개만 선택
-        spaces = spaces[:300]
+            filtered_spaces = []
+            # 추가 데이터 가져오기 (검색용)
+            params['limit'] = 1000
+            response = requests.get(url, params=params)
+            all_spaces = response.json()
+            
+            filtered_spaces = [space for space in all_spaces if search_query.lower() in 
+                             (space.get('id', '') + space.get('title', '')).lower()]
+            spaces = filtered_spaces[:300]  # 상위 300개만 유지
+        
+        # 정렬 (rank가 아닌 경우에만)
+        if sort_by != "rank":  # rank인 경우 기존 순서 유지
+            if sort_by == "rising_rate":
+                spaces.sort(key=lambda x: calculate_rising_rate(x.get('createdAt', ''), 0), reverse=True)
+            elif sort_by == "popularity":
+                spaces.sort(key=lambda x: get_popularity_grade(
+                    int(str(x.get('likes', '0')).replace(',', '')),
+                    calculate_rising_rate(x.get('createdAt', ''), 0))[1],
+                    reverse=True)
         
         progress(0.1, desc="Creating gallery...")
         html_content = """
@@ -496,29 +500,33 @@ def get_models(search_query="", sort_by="rank", progress=gr.Progress()) -> Tuple
         progress(0, desc="Fetching models data...")
         params = {
             'full': 'true',
-            'limit': 1000,
-            'search': search_query
+            'limit': 300  # 기본 300개 유지
         }
         response = requests.get(url, params=params)
         response.raise_for_status()
         models = response.json()
         
-        # 검색어로 필터링
+        # 검색어로 필터링 (검색어가 있는 경우에만)
         if search_query:
-            models = [model for model in models if search_query.lower() in 
-                     (model.get('id', '') + model.get('title', '')).lower()]
-        
-        # 정렬
-        if sort_by == "rising_rate":
-            models.sort(key=lambda x: calculate_rising_rate(x.get('createdAt', ''), 0), reverse=True)
-        elif sort_by == "popularity":
-            models.sort(key=lambda x: get_popularity_grade(
-                int(str(x.get('likes', '0')).replace(',', '')),
-                calculate_rising_rate(x.get('createdAt', ''), 0))[1],
-                reverse=True)
-        
-        # 상위 300개만 선택
-        models = models[:300]
+            filtered_models = []
+            # 추가 데이터 가져오기 (검색용)
+            params['limit'] = 1000
+            response = requests.get(url, params=params)
+            all_models = response.json()
+            
+            filtered_models = [model for model in all_models if search_query.lower() in 
+                             (model.get('id', '') + model.get('title', '')).lower()]
+            models = filtered_models[:300]  # 상위 300개만 유지
+        
+        # 정렬 (rank가 아닌 경우에만)
+        if sort_by != "rank":  # rank인 경우 기존 순서 유지
+            if sort_by == "rising_rate":
+                models.sort(key=lambda x: calculate_rising_rate(x.get('createdAt', ''), 0), reverse=True)
+            elif sort_by == "popularity":
+                models.sort(key=lambda x: get_popularity_grade(
+                    int(str(x.get('likes', '0')).replace(',', '')),
+                    calculate_rising_rate(x.get('createdAt', ''), 0))[1],
+                    reverse=True)
         
         progress(0.1, desc="Creating gallery...")
         html_content = """
@@ -547,29 +555,33 @@ def get_datasets(search_query="", sort_by="rank", progress=gr.Progress()) -> Tup
         progress(0, desc="Fetching datasets data...")
         params = {
             'full': 'true',
-            'limit': 1000,
-            'search': search_query
+            'limit': 300  # 기본 300개 유지
         }
         response = requests.get(url, params=params)
         response.raise_for_status()
         datasets = response.json()
         
-        # 검색어로 필터링
+        # 검색어로 필터링 (검색어가 있는 경우에만)
         if search_query:
-            datasets = [dataset for dataset in datasets if search_query.lower() in 
-                       (dataset.get('id', '') + dataset.get('title', '')).lower()]
-        
-        # 정렬
-        if sort_by == "rising_rate":
-            datasets.sort(key=lambda x: calculate_rising_rate(x.get('createdAt', ''), 0), reverse=True)
-        elif sort_by == "popularity":
-            datasets.sort(key=lambda x: get_popularity_grade(
-                int(str(x.get('likes', '0')).replace(',', '')),
-                calculate_rising_rate(x.get('createdAt', ''), 0))[1],
-                reverse=True)
-        
-        # 상위 300개만 선택
-        datasets = datasets[:300]
+            filtered_datasets = []
+            # 추가 데이터 가져오기 (검색용)
+            params['limit'] = 1000
+            response = requests.get(url, params=params)
+            all_datasets = response.json()
+            
+            filtered_datasets = [dataset for dataset in all_datasets if search_query.lower() in 
+                               (dataset.get('id', '') + dataset.get('title', '')).lower()]
+            datasets = filtered_datasets[:300]  # 상위 300개만 유지
+        
+        # 정렬 (rank가 아닌 경우에만)
+        if sort_by != "rank":  # rank인 경우 기존 순서 유지
+            if sort_by == "rising_rate":
+                datasets.sort(key=lambda x: calculate_rising_rate(x.get('createdAt', ''), 0), reverse=True)
+            elif sort_by == "popularity":
+                datasets.sort(key=lambda x: get_popularity_grade(
+                    int(str(x.get('likes', '0')).replace(',', '')),
+                    calculate_rising_rate(x.get('createdAt', ''), 0))[1],
+                    reverse=True)
         
         progress(0.1, desc="Creating gallery...")
         html_content = """