Update app.py

app.py

@@ -6,63 +6,28 @@ import time
 
 app = Flask(__name__)
 
-# Function to fetch spaces from Huggingface
-def fetch_spaces(limit=100):
+# Function to fetch trending spaces from Huggingface
+def fetch_trending_spaces(limit=100):
     try:
-        # Get trending spaces sorted by likes count (since trending API is giving 400 error)
+        # 트렌딩 스페이스 가져오기 (원래의 API 파라미터 사용)
         url = "https://huggingface.co/api/spaces"
         params = {
             "limit": limit,
-            "sort": "likes",
-            "order": "desc"
+            "sort": "trending"  # 단순히 trending 파라미터만 사용
         }
         response = requests.get(url, params=params, timeout=10)
         
         if response.status_code == 200:
-            return response.json()
+            # None 값이 있는 항목 필터링
+            spaces = response.json()
+            filtered_spaces = [space for space in spaces if space.get('owner') != 'None' and space.get('id', '').split('/', 1)[0] != 'None']
+            return filtered_spaces
         else:
-            print(f"Error fetching spaces: {response.status_code}")
-            # Fallback to hardcoded list if API fails
-            return get_fallback_spaces()
+            print(f"Error fetching trending spaces: {response.status_code}")
+            return []
     except Exception as e:
-        print(f"Exception when fetching spaces: {e}")
-        # Fallback to hardcoded list if API fails
-        return get_fallback_spaces()
-
-# Fallback function to provide some spaces if the API call fails
-def get_fallback_spaces():
-    return [
-        {
-            "id": "stabilityai/stable-diffusion-xl-base-1.0",
-            "owner": "stabilityai",
-            "title": "Stable Diffusion XL Base 1.0",
-            "likes": 4857
-        },
-        {
-            "id": "meta-llama/Llama-2-7b-chat-hf",
-            "owner": "meta-llama",
-            "title": "Llama 2 7B Chat",
-            "likes": 3821
-        },
-        {
-            "id": "mistralai/Mistral-7B-Instruct-v0.2",
-            "owner": "mistralai",
-            "title": "Mistral 7B Instruct v0.2",
-            "likes": 2953
-        },
-        {
-            "id": "runwayml/stable-diffusion-v1-5",
-            "owner": "runwayml",
-            "title": "Stable Diffusion v1.5",
-            "likes": 2649
-        },
-        {
-            "id": "microsoft/phi-2",
-            "owner": "microsoft",
-            "title": "Phi-2",
-            "likes": 2134
-        }
-    ]
+        print(f"Exception when fetching trending spaces: {e}")
+        return []
 
 # Transform Huggingface URL to direct space URL
 def transform_url(owner, name):
@@ -71,19 +36,28 @@ def transform_url(owner, name):
 # Get space details
 def get_space_details(space_data):
     try:
-        # Basic info
-        owner = space_data.get('owner')
-        name = space_data.get('id')
-        title = space_data.get('title') or name
+        # 공통 정보 추출
+        if '/' in space_data.get('id', ''):
+            owner, name = space_data.get('id', '').split('/', 1)
+        else:
+            owner = space_data.get('owner', '')
+            name = space_data.get('id', '')
+        
+        # None이 포함된 경우 무시
+        if owner == 'None' or name == 'None':
+            return None
         
-        # URL construction
+        # URL 구성
         original_url = f"https://huggingface.co/spaces/{owner}/{name}"
         embed_url = transform_url(owner, name)
         
-        # Get likes count
+        # 좋아요 수
         likes_count = space_data.get('likes', 0)
         
-        # Tags
+        # 제목 추출
+        title = space_data.get('title', name)
+        
+        # 태그
         tags = space_data.get('tags', [])
         
         return {
@@ -103,18 +77,18 @@ def get_space_details(space_data):
 def home():
     return render_template('index.html')
 
-# Spaces API
+# Trending spaces API
 @app.route('/api/trending-spaces', methods=['GET'])
 def trending_spaces():
     search_query = request.args.get('search', '').lower()
     limit = int(request.args.get('limit', 100))
     
-    # Fetch spaces
-    spaces_data = fetch_spaces(limit)
+    # Fetch trending spaces
+    spaces_data = fetch_trending_spaces(limit)
     
     # Process and filter spaces
     results = []
-    for space_data in spaces_data:
+    for index, space_data in enumerate(spaces_data):
         space_info = get_space_details(space_data)
         
         if not space_info:
@@ -459,9 +433,14 @@ if __name__ == '__main__':
         return;
       }
       
-      spaces.forEach(item => {
+      spaces.forEach((item, index) => {
         const { url, embedUrl, title, likes_count, owner } = item;
         
+        // Skip if owner is 'None'
+        if (owner === 'None') {
+          return;
+        }
+        
         // Create grid item
         const gridItem = document.createElement('div');
         gridItem.className = 'grid-item';