Update app.py

app.py

@@ -1,11 +1,8 @@
 import gradio as gr
 import os
-import glob
-import base64
-import mimetypes
-import random
 from datetime import datetime, timezone
 from time import sleep
+import math
 
 from huggingface_hub import HfApi
 import threading
@@ -16,6 +13,17 @@ api = HfApi()
 screenshots = []
 screenshots_folder = "screenshots"
 
+def calculate_trending_score(total_likes: int, age_seconds: int) -> float:
+    age_in_hours = age_seconds / 3600.0
+    DECAY_EXPONENT = 1.8
+    BASE_TIME_HOURS = 2.0
+
+    effective_age_for_decay = max(1.0, age_in_hours + BASE_TIME_HOURS)
+    likes_score = math.log10(max(1, total_likes + 1))
+    trending_score = likes_score / (effective_age_for_decay ** DECAY_EXPONENT)
+
+    return trending_score
+
 def load_screenshots():
     global screenshots
     for filename in os.listdir(screenshots_folder):
@@ -73,7 +81,7 @@ def update_screenshot_info():
                 "user_avatar": avatar_url,
                 "user_link": user_link,
                 "color": color_list[i % len(color_list)],
-                "rate": space_info.likes / (datetime.now(timezone.utc) - space_info.created_at).total_seconds()
+                "rate": calculate_trending_score(space_info.likes, (datetime.now(timezone.utc) - space_info.created_at).total_seconds())
             })
             print("Updated:", new_screenshots[i]["user"], new_screenshots[i]["user_link"])
         except Exception as e: