plot

app.py

@@ -17,6 +17,7 @@ import gspread
 import numpy as np
 import pandas as pd
 import requests
+import plotly.graph_objects as go
 
 from requests import HTTPError
 from apscheduler.executors.pool import ThreadPoolExecutor
@@ -28,7 +29,7 @@ from gspread_dataframe import get_as_dataframe, set_with_dataframe
 from gspread_formatting.dataframe import format_with_dataframe
 from huggingface_hub import HfApi, list_liked_repos, list_metrics, list_models
 from tabulate import tabulate
-from datetime import datetime
+from datetime import datetime, timedelta
 
 
 DISCORD_TOKEN = os.environ.get("DISCORD_TOKEN", None)
@@ -701,7 +702,133 @@ def get_data2():
         display_df = pd.DataFrame(display_data) 
         return display_df
     except Exception as e:
-        print(f"get_data2 Error: {e}")        
+        print(f"get_data2 Error: {e}")      
+
+
+#-------------------------------------------------------------------------------------------------------------------------------
+colors = ['white', 'lightgreen', 'green', 'darkgreen']
+
+# year->get all dates
+def get_dates_in_year(year):
+    start_date = datetime(year, 1, 1)
+    end_date = datetime(year + 1, 1, 1)
+    delta = end_date - start_date
+    return [start_date + timedelta(days=i) for i in range(delta.days)]
+
+# HF API->get activity
+def get_user_likes(username):
+    url = f"https://huggingface.co/api/users/{username}/likes"
+    response = requests.get(url)
+    if response.status_code == 200:
+        return response.json()
+    else:
+        print(f"Failed to fetch data: {response.status_code}")
+        return []
+
+# data->get dates
+def parse_activity_dates(activity_data):
+    activity_dates = []
+    for item in activity_data:
+        timestamp = item.get('createdAt')
+        if timestamp:
+            date = datetime.fromisoformat(timestamp[:-1]).date()
+            activity_dates.append(date)
+    return activity_dates
+
+def create_plot(username, year):
+    dates_in_year = get_dates_in_year(year)
+
+    dates_in_year = [d.date() for d in dates_in_year]
+
+    activity_data = get_user_likes(username)
+    activity_dates = parse_activity_dates(activity_data)
+
+    activity_count = {date: 0 for date in dates_in_year}
+    for date in activity_dates:
+        if date in activity_count:
+            activity_count[date] += 1
+
+    # activity->color
+    def get_color(activity_count):
+        if activity_count == 0:
+            return 'white'
+        elif activity_count < 5:
+            return 'lightgreen'
+        elif activity_count < 10:
+            return 'green'
+        else:
+            return 'darkgreen'
+
+    # data->Plotly
+    num_weeks = len(dates_in_year) // 7 + (1 if len(dates_in_year) % 7 != 0 else 0)
+
+    z = [[None for _ in range(num_weeks)] for _ in range(7)] # Was 0 
+    hover_texts = [["" for _ in range(num_weeks)] for _ in range(7)]
+
+    for idx, date in enumerate(dates_in_year):
+        week = idx // 7
+        day = idx % 7
+        z[day][week] = activity_count[date]
+        hover_texts[day][week] = f"{date}: {activity_count[date]} activities"
+
+    # heatmap
+    fig = go.Figure(data=go.Heatmap(
+        z=z,
+        x=list(range(num_weeks)),
+        y=list(range(7)),
+        hoverinfo='text',
+        text=hover_texts,
+        showscale=False,
+        colorscale=[
+            [0, 'white'],
+            [0.25, 'lightgreen'],
+            [0.5, 'green'],
+            [1, 'darkgreen']
+        ],
+        zmin=0,
+        zmax=20
+    ))
+
+    # lock the plot size and ensure cells are square
+    cell_size = 20  
+    fig.update_layout(
+        title=f'{username} Activity in {year}',
+        xaxis_nticks=52,
+        yaxis_nticks=7,
+        xaxis_title=' ',
+        yaxis_title=' ',
+        yaxis=dict(
+            tickmode='array',
+            tickvals=list(range(7)),
+            ticktext=['Monday', 'Tuesday', 'Wednesday', 'Thursday', 'Friday', 'Saturday', 'Sunday'],
+            scaleanchor="x",  # This ensures that the cells are square
+            scaleratio=1
+        ),
+        xaxis=dict(
+            scaleanchor="y",
+            scaleratio=1,
+            fixedrange=True,  # disable zoom and pan
+            showticklabels=False  # hide the x-axis labels
+        ),
+        autosize=False,
+        width=num_weeks * cell_size,
+        height=14 * cell_size,  # halve height of each cell
+        margin=dict(
+            l=20,
+            r=20,
+            b=20,
+            t=50,
+            pad=2
+        )
+    )
+
+    fig.update_layout(
+        xaxis=dict(fixedrange=True),  # disable zoom and pan
+        yaxis=dict(fixedrange=True)   # disable zoom and pan
+    )
+
+    return fig
+#-------------------------------------------------------------------------------------------------------------------------------
 
 
 demo = gr.Blocks()
@@ -740,7 +867,21 @@ with demo:
                     gr.Markdown("# 📈 How to earn Experience!")
                 with gr.Row():
                     gr.DataFrame(get_data2, every=5, interactive=False)
-            #with gr.TabItem("📈 Members of the Week", elem_id="week-table", id=1):
+            #with gr.TabItem("📈 Activity Heatmap", elem_id="activity-heatmap", id=1):
+                with gr.Row():
+                    username_input = gr.Textbox(label="Enter Username")
+                with gr.Row():
+                    year_buttons = [gr.Button(str(year)) for year in range(2021, 2025)]
+                with gr.Row():
+                    plot = gr.Plot(label="Activity Heatmap")
+            
+                # buttons for updating the plot
+                for button in year_buttons:
+                    button.click(
+                        fn=create_plot,
+                        inputs=[username_input, gr.State(int(button.value))],
+                        outputs=plot
+                    )            
          
             #with gr.TabItem("📈 Hub-only leaderboard", elem_id="hub-table", id=2):
     except Exception as e: