Update app.py

app.py

@@ -1,8 +1,8 @@
 import requests
 import gradio as gr
-from typing import List, Dict
+from typing import List, Dict, Union
 
-def get_most_liked_spaces(limit: int = 10) -> List[Dict]:
+def get_most_liked_spaces(limit: int = 10) -> Union[List[Dict], str]:
     url = "https://huggingface.co/api/spaces"
     params = {
         "sort": "likes",
@@ -11,35 +11,45 @@ def get_most_liked_spaces(limit: int = 10) -> List[Dict]:
         "full": "true"
     }
     
-    response = requests.get(url, params=params)
-    response.raise_for_status()
-    
-    return response.json()
+    try:
+        response = requests.get(url, params=params)
+        response.raise_for_status()
+        data = response.json()
+        
+        if isinstance(data, list):
+            return data
+        else:
+            return f"Unexpected API response format: {type(data)}"
+    except requests.RequestException as e:
+        return f"API request error: {str(e)}"
+    except ValueError as e:
+        return f"JSON decoding error: {str(e)}"
 
-def format_spaces(spaces: List[Dict]) -> str:
+def format_spaces(spaces: Union[List[Dict], str]) -> str:
+    if isinstance(spaces, str):
+        return spaces  # 이미 오류 메시지인 경우 그대로 반환
+    
     output = ""
     for idx, space in enumerate(spaces, 1):
-        # API 응답 구조 변경을 고려한 안전한 데이터 접근
+        if not isinstance(space, dict):
+            output += f"{idx}. Unexpected space data format: {type(space)}\n\n"
+            continue
+        
         space_id = space.get('id', 'Unknown')
-        space_name = space.get('title', space.get('name', 'Unknown'))  # 'title' 또는 'name' 사용
-        space_author = space.get('author', {}).get('name', 'Unknown')  # 중첩된 구조 고려
+        space_name = space.get('title', space.get('name', 'Unknown'))
+        author_data = space.get('author', {})
+        space_author = author_data.get('name', 'Unknown') if isinstance(author_data, dict) else 'Unknown'
         space_likes = space.get('likes', 'N/A')
         
         output += f"{idx}. {space_name} by {space_author}\n"
         output += f"   Likes: {space_likes}\n"
         output += f"   URL: https://huggingface.co/spaces/{space_id}\n\n"
-    return output
+    
+    return output if output else "No valid space data found."
 
 def get_spaces_list(limit: int) -> str:
-    try:
-        spaces = get_most_liked_spaces(limit)
-        if not spaces:
-            return "No spaces found or empty response from API."
-        return format_spaces(spaces)
-    except requests.RequestException as e:
-        return f"An error occurred while fetching data: {str(e)}"
-    except Exception as e:
-        return f"An unexpected error occurred: {str(e)}"
+    spaces = get_most_liked_spaces(limit)
+    return format_spaces(spaces)
 
 # Gradio 인터페이스 정의
 iface = gr.Interface(