revert

app.py

@@ -1,26 +1,21 @@
 import os
-from datetime import datetime, timedelta
-
-import argilla as rg
-import gradio as gr
+import datetime
 import pandas as pd
-import plotly.colors as colors
+import gradio as gr
+import argilla as rg
 import plotly.graph_objects as go
-from cachetools import TTLCache, cached
+import plotly.colors as colors
 
 client = rg.Argilla(
     api_url=os.getenv("ARGILLA_API_URL"), api_key=os.getenv("ARGILLA_API_KEY")
 )
-cache = TTLCache(maxsize=100, ttl=timedelta(minutes=10), timer=datetime.now)
 
 
 def fetch_data(dataset_name: str, workspace: str):
     return client.datasets(dataset_name, workspace=workspace)
 
 
-@cached(cache)
-def get_progress(dataset_name: str, workspace: str) -> dict:
-    dataset = fetch_data(dataset_name, workspace)
+def get_progress(dataset) -> dict:
     records = list(dataset.records)
     total_records = len(records)
     annotated_records = len(
@@ -34,9 +29,7 @@ def get_progress(dataset_name: str, workspace: str) -> dict:
     }
 
 
-@cached(cache)
-def get_leaderboard(dataset_name: str, workspace: str) -> dict:
-    dataset = fetch_data(dataset_name, workspace)
+def get_leaderboard(dataset) -> dict:
     user_annotations = {}
     for record in dataset.records:
         for response in record.responses:
@@ -156,10 +149,9 @@ def create_treemap(user_annotations, total_records):
 
 
 def update_dashboard():
-    dataset_name = os.getenv("DATASET_NAME")
-    workspace = os.getenv("WORKSPACE")
-    progress = get_progress(dataset_name, workspace)
-    user_annotations = get_leaderboard(dataset_name, workspace)
+    dataset = fetch_data(os.getenv("DATASET_NAME"), os.getenv("WORKSPACE"))
+    progress = get_progress(dataset)
+    user_annotations = get_leaderboard(dataset)
 
     gauge_chart = create_gauge_chart(progress)
     treemap = create_treemap(user_annotations, progress["total"])
@@ -175,10 +167,7 @@ def update_dashboard():
 
 
 with gr.Blocks() as demo:
-    gr.Markdown("# Argilla Dataset Annotation Progress Dashboard")
-    gr.Markdown(
-        "This dashboard shows the progress of the dataset annotation and the contributions of the annotators for a dataset on [Argilla](https://davanstrien-my-argilla.hf.space/dataset/1100a091-7f3f-4a6e-ad51-4e859abab58f/annotation-mode?page=2&status=pending)."
-    )
+    gr.Markdown("# Argilla Dataset Dashboard")
 
     with gr.Row():
         gauge_output = gr.Plot(label="Overall Progress")