Update app.py

app.py

@@ -88,60 +88,37 @@ def take_screenshot(url):
 def get_hardware_info(item: dict) -> tuple:
     """하드웨어 정보 추출"""
     try:
-        # 디버그용 출력
-        print("Hardware Info:", item.get('hardware', {}))
-        
-        hardware = item.get('hardware', {})
-        resources = item.get('resources', {})  # resources 필드도 확인
+        # runtime 정보 확인
+        runtime = item.get('runtime', {})
         
         # CPU 정보 처리
-        cpu_info = hardware.get('cpu', 'Standard')
+        cpu_info = runtime.get('cpu', 'Standard')
         
-        # GPU 정보 처리 - 여러 가능한 경로 확인
+        # GPU 정보 처리
         gpu_info = "None"
+        if runtime.get('accelerator') == "gpu":
+            gpu_type = runtime.get('gpu', {}).get('name', '')
+            gpu_memory = runtime.get('gpu', {}).get('memory', '')
+            if gpu_type:
+                gpu_info = f"{gpu_type}"
+                if gpu_memory:
+                    gpu_info += f" ({gpu_memory}GB)"
+        
+        # spaces decorator 확인
+        if '@spaces.GPU' in str(item.get('sdk_version', '')):
+            if gpu_info == "None":
+                gpu_info = "GPU Enabled"
         
-        # 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')
         
+        print(f"Debug - Runtime Info: {runtime}")  # 디버그 출력
+        print(f"Debug - GPU Info: {gpu_info}")    # 디버그 출력
+        
         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:
@@ -382,6 +359,7 @@ 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"
@@ -392,17 +370,24 @@ def get_trending_spaces(progress=gr.Progress()) -> Tuple[str, str]:
         params = {
             'full': 'true',
             'limit': 10,
-            'hardware': 'true'
+            'hardware': 'true',
+            'runtime': 'true'  # runtime 정보 포함
+        }
+        headers = {
+            'Accept': 'application/json'
         }
-        response = requests.get(url, params=params)
+        
+        response = requests.get(url, params=params, headers=headers)
         response.raise_for_status()
         spaces = response.json()
         
         # 디버그용 출력
-        print("First space hardware info:", spaces[0].get('hardware', {}))
+        print("First space full info:", spaces[0])
         
         # 상위 10개만 선택
         top_spaces = spaces[:10]
+    
+
         
         progress(0.1, desc="Creating gallery...")
         html_content = """