Update app.py

app.py

@@ -50,6 +50,7 @@ def capture_thumbnail(space_id: str) -> bytes:
         pass
     return None
 
+
 def format_spaces(spaces: Union[List[Dict], str]) -> List[Dict]:
     if isinstance(spaces, str):
         return [{"error": spaces}]
@@ -61,12 +62,18 @@ def format_spaces(spaces: Union[List[Dict], str]) -> List[Dict]:
         else:
             print(f"Skipping invalid space: {space}")
     
-    with concurrent.futures.ThreadPoolExecutor(max_workers=10) as executor:
-        return list(executor.map(format_space, valid_spaces))
+    formatted = []
+    for space in valid_spaces:
+        try:
+            formatted.append(format_space(space))
+        except Exception as e:
+            print(f"Error formatting space: {space.get('id', 'Unknown')}. Error: {str(e)}")
+    
+    return formatted
 
 def format_space(space: Dict) -> Dict:
     if not isinstance(space, dict):
-        return {"error": f"Invalid space format: {space}"}
+        raise ValueError(f"Invalid space format: {space}")
     
     space_id = space.get('id', 'Unknown')
     space_name = space_id.split('/')[-1] if '/' in space_id else space_id
@@ -95,12 +102,24 @@ def get_hardware_requirements(space: Dict) -> str:
     if not isinstance(space, dict):
         return "Unknown"
     
-    sdk_version = space.get('sdk', {}).get('hf', {}).get('version')
+    sdk = space.get('sdk', {})
+    if not isinstance(sdk, dict):
+        return "Unknown"
+    
+    hf = sdk.get('hf', {})
+    if not isinstance(hf, dict):
+        return "Unknown"
+    
+    sdk_version = hf.get('version')
     if sdk_version and int(sdk_version.split('.')[0]) >= 3:
-        return space.get('sdk', {}).get('hardware', 'Unknown')
+        return sdk.get('hardware', 'Unknown')
     else:
-        return 'CPU' if space.get('host_requirements', {}).get('hardware', {}).get('gpu') is False else 'GPU'
-
+        host_requirements = space.get('host_requirements', {})
+        if isinstance(host_requirements, dict):
+            hardware = host_requirements.get('hardware', {})
+            if isinstance(hardware, dict):
+                return 'CPU' if hardware.get('gpu') is False else 'GPU'
+    return 'Unknown'
 
 def get_app_py_content(space_id: str) -> str:
     app_py_url = f"https://huggingface.co/spaces/{space_id}/raw/main/app.py"
@@ -127,8 +146,14 @@ def summarize_space(space: Dict) -> str:
 
 def create_ui():
     spaces_list = get_most_liked_spaces()
+    print(f"Type of spaces_list: {type(spaces_list)}")
+    if isinstance(spaces_list, str):
+        print(f"Error in get_most_liked_spaces: {spaces_list}")
+        return gr.Blocks().launch()
+    
     formatted_spaces = format_spaces(spaces_list)
     print(f"Total spaces loaded: {len(formatted_spaces)}")  # 디버깅 출력
+    
 
     css = """
     footer {