Update app.py

app.py

@@ -115,53 +115,58 @@ def get_card(item: dict, index: int, card_type: str = "space") -> str:
         gpu_memory = gpu_info.get('memory', 'N/A')
         gpu_info = f"{gpu_type} ({gpu_memory}GB)"
 
-# 카드 타입별 그라데이션 설정
-if card_type == "space":
-    gradient_colors = """
-        rgba(255, 182, 193, 0.7),    /* 파스텔 핑크 */
-        rgba(173, 216, 230, 0.7),    /* 파스텔 블루 */
-        rgba(255, 218, 185, 0.7)     /* 파스텔 피치 */
-    """
-    bg_content = f"""
-        background-image: url(data:image/png;base64,{get_cached_screenshot(url) if get_cached_screenshot(url) else ''});
-        background-size: cover;
-        background-position: center;
-    """
-    content_bg = f"""
-        background: linear-gradient(135deg, {gradient_colors});
-        backdrop-filter: blur(10px);
-    """
-elif card_type == "model":
-    gradient_colors = """
-        rgba(110, 142, 251, 0.7),    /* 모델 블루 */
-        rgba(130, 158, 251, 0.7),
-        rgba(150, 174, 251, 0.7)
-    """
-    bg_content = f"""
-        background: linear-gradient(135deg, #6e8efb, #4a6cf7);
-        padding: 15px;
-    """
-    content_bg = f"""
-        background: linear-gradient(135deg, {gradient_colors});
-        backdrop-filter: blur(10px);
-    """
-else:  # dataset
-    gradient_colors = """
-        rgba(255, 107, 107, 0.7),    /* 데이터셋 레드 */
-        rgba(255, 127, 127, 0.7),
-        rgba(255, 147, 147, 0.7)
-    """
-    bg_content = f"""
-        background: linear-gradient(135deg, #ff6b6b, #ff8787);
-        padding: 15px;
-    """
-    content_bg = f"""
-        background: linear-gradient(135deg, {gradient_colors});
-        backdrop-filter: blur(10px);
-    """
-
+    # 카드 타입별 그라데이션 설정
+    if card_type == "space":
+        gradient_colors = """
+            rgba(255, 182, 193, 0.7),    /* 파스텔 핑크 */
+            rgba(173, 216, 230, 0.7),    /* 파스텔 블루 */
+            rgba(255, 218, 185, 0.7)     /* 파스텔 피치 */
+        """
+        bg_content = f"""
+            background-image: url(data:image/png;base64,{get_cached_screenshot(url) if get_cached_screenshot(url) else ''});
+            background-size: cover;
+            background-position: center;
+        """
+        content_bg = f"""
+            background: linear-gradient(135deg, {gradient_colors});
+            backdrop-filter: blur(10px);
+        """
+        type_icon = "🎯"
+        type_label = "SPACE"
+    elif card_type == "model":
+        gradient_colors = """
+            rgba(110, 142, 251, 0.7),    /* 모델 블루 */
+            rgba(130, 158, 251, 0.7),
+            rgba(150, 174, 251, 0.7)
+        """
+        bg_content = f"""
+            background: linear-gradient(135deg, #6e8efb, #4a6cf7);
+            padding: 15px;
+        """
+        content_bg = f"""
+            background: linear-gradient(135deg, {gradient_colors});
+            backdrop-filter: blur(10px);
+        """
+        type_icon = "🤖"
+        type_label = "MODEL"
+    else:  # dataset
+        gradient_colors = """
+            rgba(255, 107, 107, 0.7),    /* 데이터셋 레드 */
+            rgba(255, 127, 127, 0.7),
+            rgba(255, 147, 147, 0.7)
+        """
+        bg_content = f"""
+            background: linear-gradient(135deg, #ff6b6b, #ff8787);
+            padding: 15px;
+        """
+        content_bg = f"""
+            background: linear-gradient(135deg, {gradient_colors});
+            backdrop-filter: blur(10px);
+        """
+        type_icon = "📊"
+        type_label = "DATASET"
 
-    # 하드웨어 정보 HTML (모든 카드 타입에 대해 동일하게 적용)
+    # 하드웨어 정보 HTML
     hardware_info = f"""
     <div style='
         margin-top: 15px;
@@ -187,6 +192,7 @@ else:  # dataset
     """
 
     # 태그 표시 (models와 datasets용)
+    tags_html = ""
     if card_type != "space":
         tags_html = f"""
         <div style='
@@ -211,9 +217,8 @@ else:  # dataset
             ''' for tag in tags[:5]])}
         </div>
         """
-    else:
-        tags_html = ""
 
+    # 카드 HTML 반환
     return f"""
     <div class="card" style='
         position: relative;