Update app.py

app.py

@@ -50,26 +50,24 @@ def capture_thumbnail(space_id: str) -> bytes:
         pass
     return None
 
-def get_hardware_requirements(space: Dict) -> str:
-    sdk_version = space.get('sdk', {}).get('hf', {}).get('version')
-    if sdk_version and int(sdk_version.split('.')[0]) >= 3:
-        return space.get('sdk', {}).get('hardware', 'Unknown')
-    else:
-        return 'CPU' if space.get('host_requirements', {}).get('hardware', {}).get('gpu') is False else 'GPU'
-
 def format_spaces(spaces: Union[List[Dict], str]) -> List[Dict]:
     if isinstance(spaces, str):
         return [{"error": spaces}]
     
-    valid_spaces = [space for space in spaces if isinstance(space, dict)]
+    valid_spaces = []
+    for space in spaces:
+        if isinstance(space, dict):
+            valid_spaces.append(space)
+        else:
+            print(f"Skipping invalid space: {space}")
     
     with concurrent.futures.ThreadPoolExecutor(max_workers=10) as executor:
         return list(executor.map(format_space, valid_spaces))
 
-def format_space(space: Union[Dict, str]) -> Dict:
+def format_space(space: Dict) -> Dict:
     if not isinstance(space, dict):
-        return {"error": "Invalid space data"}
-
+        return {"error": f"Invalid space format: {space}"}
+    
     space_id = space.get('id', 'Unknown')
     space_name = space_id.split('/')[-1] if '/' in space_id else space_id
     
@@ -93,6 +91,16 @@ def format_space(space: Union[Dict, str]) -> Dict:
         "hardware": hardware
     }
 
+def get_hardware_requirements(space: Dict) -> str:
+    if not isinstance(space, dict):
+        return "Unknown"
+    
+    sdk_version = space.get('sdk', {}).get('hf', {}).get('version')
+    if sdk_version and int(sdk_version.split('.')[0]) >= 3:
+        return space.get('sdk', {}).get('hardware', 'Unknown')
+    else:
+        return 'CPU' if space.get('host_requirements', {}).get('hardware', {}).get('gpu') is False else 'GPU'
+
 
 def get_app_py_content(space_id: str) -> str:
     app_py_url = f"https://huggingface.co/spaces/{space_id}/raw/main/app.py"