Update app.py

app.py

@@ -85,6 +85,124 @@ def take_screenshot(url):
         if 'driver' in locals():
             driver.quit()
 
+from datetime import datetime, timedelta
+
+def calculate_rising_rate(created_date: str, rank: int) -> int:
+    """AI Rising Rate 계산"""
+    # 생성일 기준 점수 계산
+    created = datetime.strptime(created_date.split('T')[0], '%Y-%m-%d')
+    today = datetime.now()
+    days_diff = (today - created).days
+    date_score = max(0, 300 - days_diff)  # 최대 300점
+    
+    # 순위 기준 점수 계산
+    rank_score = max(0, 300 - rank)  # 최대 300점
+    
+    # 총점 계산
+    total_score = date_score + rank_score
+    
+    # 별 개수 계산 (0~5)
+    if total_score <= 100:
+        stars = 1
+    elif total_score <= 200:
+        stars = 2
+    elif total_score <= 300:
+        stars = 3
+    elif total_score <= 400:
+        stars = 4
+    else:
+        stars = 5
+        
+    return stars
+
+def get_popularity_grade(likes: int, stars: int) -> tuple:
+    """AI Popularity Score 등급 계산"""
+    # 기본 점수 (likes)
+    base_score = min(likes, 10000)  # 최대 10000점
+    
+    # 별점 추가 점수 (별 하나당 500점)
+    star_score = stars * 500
+    
+    # 총점
+    total_score = base_score + star_score
+    
+    # 등급 테이블 (18단계)
+    grades = [
+        (9000, "AAA+"), (8500, "AAA"), (8000, "AAA-"),
+        (7500, "AA+"), (7000, "AA"), (6500, "AA-"),
+        (6000, "A+"), (5500, "A"), (5000, "A-"),
+        (4500, "BBB+"), (4000, "BBB"), (3500, "BBB-"),
+        (3000, "BB+"), (2500, "BB"), (2000, "BB-"),
+        (1500, "B+"), (1000, "B"), (500, "B-")
+    ]
+    
+    for threshold, grade in grades:
+        if total_score >= threshold:
+            return grade, total_score
+            
+    return "B-", total_score
+
+# get_card 함수 내의 hardware_info 부분을 다음으로 교체:
+def get_rating_info(item: dict, index: int) -> str:
+    """평가 정보 HTML 생성"""
+    created = item.get('createdAt', '').split('T')[0]
+    likes = int(str(item.get('likes', '0')).replace(',', ''))
+    
+    # AI Rising Rate 계산
+    stars = calculate_rising_rate(created, index + 1)
+    star_html = "★" * stars + "☆" * (5 - stars)  # 채워진 별과 빈 별 조합
+    
+    # AI Popularity Score 계산
+    grade, score = get_popularity_grade(likes, stars)
+    
+    # 등급별 색상 설정
+    grade_colors = {
+        'AAA': '#FFD700', 'AA': '#FFA500', 'A': '#FF4500',
+        'BBB': '#4169E1', 'BB': '#1E90FF', 'B': '#00BFFF'
+    }
+    grade_base = grade.rstrip('+-')
+    grade_color = grade_colors.get(grade_base, '#666666')
+    
+    return f"""
+    <div style='
+        margin-top: 15px;
+        padding: 15px;
+        background: rgba(255,255,255,0.4);
+        border-radius: 10px;
+        font-size: 0.9em;
+        box-shadow: 0 2px 10px rgba(0,0,0,0.1);'>
+        <div style='
+            display: grid;
+            grid-template-columns: repeat(2, 1fr);
+            gap: 15px;'>
+            <div style='
+                color: #333;
+                display: flex;
+                flex-direction: column;
+                gap: 5px;'>
+                <span style='font-weight: bold;'>AI Rising Rate:</span>
+                <span style='
+                    color: #FF8C00;
+                    font-size: 1.4em;
+                    letter-spacing: 2px;
+                    text-shadow: 1px 1px 2px rgba(0,0,0,0.1);'>{star_html}</span>
+            </div>
+            <div style='
+                color: #333;
+                display: flex;
+                flex-direction: column;
+                gap: 5px;'>
+                <span style='font-weight: bold;'>AI Popularity Score:</span>
+                <span style='
+                    font-size: 1.2em;
+                    font-weight: bold;
+                    color: {grade_color};
+                    text-shadow: 1px 1px 2px rgba(0,0,0,0.1);'>{grade} ({score:,})</span>
+            </div>
+        </div>
+    </div>
+    """
+
 def get_hardware_info(item: dict) -> tuple:
     """하드웨어 정보 추출"""
     try:
@@ -120,7 +238,7 @@ def get_hardware_info(item: dict) -> tuple:
     except Exception as e:
         print(f"Error parsing hardware info: {str(e)}")
         return 'Standard', 'None', 'N/A'
-    
+
 def get_card(item: dict, index: int, card_type: str = "space") -> str:
     """통합 카드 HTML 생성"""
     item_id = item.get('id', '')
@@ -141,15 +259,10 @@ def get_card(item: dict, index: int, card_type: str = "space") -> str:
     pipeline_tag = item.get('pipeline_tag', '')
     license = item.get('license', '')
     sdk = item.get('sdk', 'N/A')
-    
-    # 하드웨어 요구사항 처리
-    hardware = item.get('hardware', {})
-    cpu_info = hardware.get('cpu', 'Standard')
-    gpu_info = hardware.get('gpu', 'None')
-    if isinstance(gpu_info, dict):
-        gpu_type = gpu_info.get('type', 'N/A')
-        gpu_memory = gpu_info.get('memory', 'N/A')
-        gpu_info = f"{gpu_type} ({gpu_memory}GB)"
+
+    # AI Rating 정보 가져오기
+    rating_info = get_rating_info(item, index)
+
 
     # 카드 타입별 그라데이션 설정
     if card_type == "space":
@@ -366,28 +479,16 @@ def get_trending_spaces(progress=gr.Progress()) -> Tuple[str, str]:
     
     try:
         progress(0, desc="Fetching spaces data...")
-        # 더 자세한 정보를 포함하도록 파라미터 추가
         params = {
             'full': 'true',
-            'limit': 10,
-            'hardware': 'true',
-            'runtime': 'true'  # runtime 정보 포함
+            'limit': 10
         }
-        headers = {
-            'Accept': 'application/json'
-        }
-        
-        response = requests.get(url, params=params, headers=headers)
+        response = requests.get(url, params=params)
         response.raise_for_status()
         spaces = response.json()
         
-        # 디버그용 출력
-        print("First space full info:", spaces[0])
-        
         # 상위 10개만 선택
         top_spaces = spaces[:10]
-    
-
         
         progress(0.1, desc="Creating gallery...")
         html_content = """