add defaults and

app.py

@@ -1,11 +1,13 @@
 import requests
 import gradio as gr
 from urllib.parse import urlencode
-
 import os
+from datetime import datetime
 
 # Load environment variables
 
+DEFAULT_IMAGE = "https://argilla.imglab-cdn.net/dibt/dibt_v2.png?width=1200&text=%3Cspan+size%3D%2212pt%22+weight%3D%22bold%22%3EHugging+Face++%E2%9D%A4%EF%B8%8F+bartowski+in+2024%3C%2Fspan%3E%0A%0A%3Cspan+weight%3D%22bold%22%3E3%2C057%2C452%3C%2Fspan%3E+model+downloads%0A%3Cspan+weight%3D%22bold%22%3E5%2C404%3C%2Fspan%3E+model+likes%0A%3Cspan+weight%3D%22bold%22%3E0%3C%2Fspan%3E+dataset+downloads%0A%3Cspan+weight%3D%22bold%22%3E0%3C%2Fspan%3E+dataset+likes%0A%0A%3Cspan+size%3D%2210pt%22%3EMost+Popular+Contributions%3A%3C%2Fspan%3E%0AModel%3A+%3Cspan+weight%3D%22bold%22%3Ebartowski%2Fgemma-2-9b-it-GGUF%3C%2Fspan%3E%0A++%2844%2C256+downloads%2C+196+likes%29%0ADataset%3A+%3Cspan+weight%3D%22bold%22%3ENone%3C%2Fspan%3E%0A++%280+downloads%2C+0+likes%29%0ASpace%3A+%3Cspan+weight%3D%22bold%22%3Ebartowski%2Fgguf-metadata-updater%3C%2Fspan%3E%0A++%287+likes%29&text-width=800&text-height=600&text-padding=60&text-color=39%2C71%2C111&text-x=460&text-y=40&format=png&dpr=2"
+
 
 def create_image(stats, username):
     url = "https://argilla.imglab-cdn.net/dibt/dibt_v2.png"
@@ -44,6 +46,13 @@ Space: <span weight="bold">{top_items['Top Space']['name']}</span>
     return f"{url}?{urlencode(params)}"
 
 
+def is_from_2024(created_at_str):
+    if not created_at_str:
+        return False
+    created_at = datetime.strptime(created_at_str, "%Y-%m-%dT%H:%M:%S.%fZ")
+    return created_at.year == 2024
+
+
 def get_user_stats(username):
     headers = {"Authorization": f"Bearer {os.getenv('HF_TOKEN')}"}
 
@@ -53,7 +62,12 @@ def get_user_stats(username):
         params={"author": username, "full": "True"},
         headers=headers,
     )
-    models = models_response.json()
+    # Filter for 2024 models only
+    models = [
+        model
+        for model in models_response.json()
+        if is_from_2024(model.get("createdAt"))
+    ]
 
     # Get datasets stats
     datasets_response = requests.get(
@@ -61,7 +75,12 @@ def get_user_stats(username):
         params={"author": username, "full": "True"},
         headers=headers,
     )
-    datasets = datasets_response.json()
+    # Filter for 2024 datasets only
+    datasets = [
+        dataset
+        for dataset in datasets_response.json()
+        if is_from_2024(dataset.get("createdAt"))
+    ]
 
     # Get spaces stats
     spaces_response = requests.get(
@@ -69,16 +88,21 @@ def get_user_stats(username):
         params={"author": username, "full": "True"},
         headers=headers,
     )
-    spaces = spaces_response.json()
-
-    # Calculate totals
+    # Filter for 2024 spaces only
+    spaces = [
+        space
+        for space in spaces_response.json()
+        if is_from_2024(space.get("createdAt"))
+    ]
+
+    # Calculate totals for 2024 items only
     total_model_downloads = sum(model.get("downloads", 0) for model in models)
     total_model_likes = sum(model.get("likes", 0) for model in models)
     total_dataset_downloads = sum(dataset.get("downloads", 0) for dataset in datasets)
     total_dataset_likes = sum(dataset.get("likes", 0) for dataset in datasets)
     total_space_likes = sum(space.get("likes", 0) for space in spaces)
 
-    # Find most liked items
+    # Find most liked items from 2024
     most_liked_model = max(models, key=lambda x: x.get("likes", 0), default=None)
     most_liked_dataset = max(datasets, key=lambda x: x.get("likes", 0), default=None)
     most_liked_space = max(spaces, key=lambda x: x.get("likes", 0), default=None)
@@ -135,13 +159,18 @@ with gr.Blocks(title="Hugging Face Community Stats") as demo:
 
     with gr.Row():
         username_input = gr.Textbox(
-            label=None, placeholder="Enter Hugging Face username...", scale=4
+            label="Hub username",
+            placeholder="Enter Hugging Face username...",
+            scale=6,
+            value="bartowski",
         )
-        submit_btn = gr.Button("Get Stats", scale=1)
+        submit_btn = gr.Button("Get Stats", scale=6)
 
     with gr.Row():
         with gr.Column():
-            stats_image = gr.Markdown()
+            stats_image = gr.Markdown(
+                f"![Hugging Face Stats]({DEFAULT_IMAGE})"
+            )
 
     # Add example usernames
     gr.Examples(