Update app.py

app.py

@@ -87,26 +87,62 @@ def take_screenshot(url):
 
 def get_hardware_info(item: dict) -> tuple:
     """하드웨어 정보 추출"""
-    hardware = item.get('hardware', {})
-    
-    # CPU 정보 처리
-    cpu_info = hardware.get('cpu', 'Standard')
-    
-    # GPU 정보 처리
-    gpu_info = hardware.get('gpu', {})
-    if isinstance(gpu_info, dict):
-        gpu_name = gpu_info.get('name', '')
-        gpu_memory = gpu_info.get('memory', '')
-        gpu_info = f"{gpu_name} ({gpu_memory}GB)" if gpu_name and gpu_memory else "None"
-    elif isinstance(gpu_info, str):
-        gpu_info = gpu_info if gpu_info else "None"
-    else:
+    try:
+        # 디버그용 출력
+        print("Hardware Info:", item.get('hardware', {}))
+        
+        hardware = item.get('hardware', {})
+        resources = item.get('resources', {})  # resources 필드도 확인
+        
+        # CPU 정보 처리
+        cpu_info = hardware.get('cpu', 'Standard')
+        
+        # GPU 정보 처리 - 여러 가능한 경로 확인
         gpu_info = "None"
-    
-    # SDK 정보 처리
-    sdk = item.get('sdk', 'N/A')
-    
-    return cpu_info, gpu_info, sdk
+        
+        # 1. hardware.gpu에서 확인
+        if 'gpu' in hardware:
+            gpu_data = hardware['gpu']
+            if isinstance(gpu_data, dict):
+                gpu_name = gpu_data.get('name', '')
+                gpu_memory = gpu_data.get('memory', '')
+                if gpu_name or gpu_memory:
+                    gpu_info = f"{gpu_name} ({gpu_memory}GB)" if gpu_memory else gpu_name
+            elif isinstance(gpu_data, str) and gpu_data:
+                gpu_info = gpu_data
+
+        # 2. resources.gpu에서 확인
+        if gpu_info == "None" and 'gpu' in resources:
+            gpu_data = resources['gpu']
+            if isinstance(gpu_data, dict):
+                gpu_name = gpu_data.get('name', '')
+                gpu_memory = gpu_data.get('memory', '')
+                if gpu_name or gpu_memory:
+                    gpu_info = f"{gpu_name} ({gpu_memory}GB)" if gpu_memory else gpu_name
+            elif isinstance(gpu_data, str) and gpu_data:
+                gpu_info = gpu_data
+
+        # 3. runtime.gpu에서 확인
+        runtime = item.get('runtime', {})
+        if gpu_info == "None" and 'gpu' in runtime:
+            gpu_data = runtime['gpu']
+            if isinstance(gpu_data, dict):
+                gpu_name = gpu_data.get('name', '')
+                gpu_memory = gpu_data.get('memory', '')
+                if gpu_name or gpu_memory:
+                    gpu_info = f"{gpu_name} ({gpu_memory}GB)" if gpu_memory else gpu_name
+            elif isinstance(gpu_data, str) and gpu_data:
+                gpu_info = gpu_data
+
+        # SDK 정보 처리
+        sdk = item.get('sdk', 'N/A')
+        
+        return cpu_info, gpu_info, sdk
+        
+    except Exception as e:
+        print(f"Error parsing hardware info: {str(e)}")
+        print(f"Item data: {item}")
+        return 'Standard', 'None', 'N/A'
     
 def get_card(item: dict, index: int, card_type: str = "space") -> str:
     """통합 카드 HTML 생성"""
@@ -346,19 +382,26 @@ def get_card(item: dict, index: int, card_type: str = "space") -> str:
         </div>
     </div>
     """
-
-
 def get_trending_spaces(progress=gr.Progress()) -> Tuple[str, str]:
     """트렌딩 스페이스 가져오기"""
     url = "https://huggingface.co/api/spaces"
     
     try:
         progress(0, desc="Fetching spaces data...")
-        response = requests.get(url)
+        # 더 자세한 정보를 포함하도록 파라미터 추가
+        params = {
+            'full': 'true',
+            'limit': 10,
+            'hardware': 'true'
+        }
+        response = requests.get(url, params=params)
         response.raise_for_status()
         spaces = response.json()
         
-        # 상위 10개만 선택 (원본 순서 유지)
+        # 디버그용 출력
+        print("First space hardware info:", spaces[0].get('hardware', {}))
+        
+        # 상위 10개만 선택
         top_spaces = spaces[:10]
         
         progress(0.1, desc="Creating gallery...")