only fetching from the API once at load

app.py

@@ -30,10 +30,10 @@ def obtain_source_target_datasets() -> (
     filtered_source_dataset = source_dataset.filter_by(response_status=["pending"])
 
     # Obtain a list of users from the private workspace
-    #target_dataset = rg.FeedbackDataset.from_argilla(
+    # target_dataset = rg.FeedbackDataset.from_argilla(
     #    os.getenv("RESULTS_DATASET"), workspace=os.getenv("RESULTS_WORKSPACE")
-    #)
-    
+    # )
+
     target_dataset = source_dataset.filter_by(response_status=["submitted"])
 
     return filtered_source_dataset, target_dataset
@@ -64,18 +64,21 @@ def get_user_annotations_dictionary(
 
     return output
 
+
 def donut_chart() -> alt.Chart:
+
     # Load your data
-    source_dataset, results = obtain_source_target_datasets()
-    annotated_records = len(results)
+    annotated_records = len(target_dataset)
     pending_records = int(os.getenv("TARGET_RECORDS")) - annotated_records
 
     # Prepare data for the donut chart
-    source = pd.DataFrame({
-        "values": [annotated_records, pending_records],
-        "category": ["Completed", "Remaining"],
-        "colors": ["#4CAF50", "#757575"]  # Green for Completed, Grey for Remaining
-    })
+    source = pd.DataFrame(
+        {
+            "values": [annotated_records, pending_records],
+            "category": ["Completed", "Remaining"],
+            "colors": ["#4CAF50", "#757575"],  # Green for Completed, Grey for Remaining
+        }
+    )
 
     base = alt.Chart(source).encode(
         theta=alt.Theta("values:Q", stack=True),
@@ -93,6 +96,7 @@ def donut_chart() -> alt.Chart:
 
     return chart
 
+
 def kpi_chart_remaining() -> alt.Chart:
     """
     This function returns a KPI chart with the total amount of annotators.
@@ -100,8 +104,7 @@ def kpi_chart_remaining() -> alt.Chart:
         An altair chart with the KPI chart.
     """
 
-    source_dataset, results = obtain_source_target_datasets()
-    pending_records = int(os.getenv("TARGET_RECORDS")) - len(results)
+    pending_records = int(os.getenv("TARGET_RECORDS")) - len(target_dataset)
     # Assuming you have a DataFrame with user data, create a sample DataFrame
     data = pd.DataFrame({"Category": ["Total remaining"], "Value": [pending_records]})
 
@@ -115,6 +118,7 @@ def kpi_chart_remaining() -> alt.Chart:
 
     return chart
 
+
 def kpi_chart_submitted() -> alt.Chart:
     """
     This function returns a KPI chart with the total amount of annotators.
@@ -122,9 +126,6 @@ def kpi_chart_submitted() -> alt.Chart:
         An altair chart with the KPI chart.
     """
 
-    # Obtain the total amount of annotators
-    _, target_dataset = obtain_source_target_datasets()
-
     total = len(target_dataset)
 
     # Assuming you have a DataFrame with user data, create a sample DataFrame
@@ -150,12 +151,12 @@ def kpi_chart() -> alt.Chart:
     """
 
     # Obtain the total amount of annotators
-    _, target_dataset = obtain_source_target_datasets()
-    user_ids_annotations = get_user_annotations_dictionary(target_dataset)
     total_annotators = len(user_ids_annotations)
 
     # Assuming you have a DataFrame with user data, create a sample DataFrame
-    data = pd.DataFrame({"Category": ["Total Contributors"], "Value": [total_annotators]})
+    data = pd.DataFrame(
+        {"Category": ["Total Contributors"], "Value": [total_annotators]}
+    )
 
     # Create Altair chart
     chart = (
@@ -195,15 +196,17 @@ def main() -> None:
         extra_headers={"Authorization": f"Bearer {os.getenv('HF_TOKEN')}"},
     )
 
+    global source_dataset, target_dataset, user_ids_annotations
     source_dataset, target_dataset = obtain_source_target_datasets()
     user_ids_annotations = get_user_annotations_dictionary(target_dataset)
 
     top5_dataframe = obtain_top_5_users(user_ids_annotations)
 
-    annotated = len(target_dataset) 
+    annotated = len(target_dataset)
     remaining = int(os.getenv("TARGET_RECORDS")) - annotated
-    percentage_completed = round((annotated / int(os.getenv("TARGET_RECORDS"))) * 100,1)
-
+    percentage_completed = round(
+        (annotated / int(os.getenv("TARGET_RECORDS"))) * 100, 1
+    )
 
     with gr.Blocks() as demo:
         gr.Markdown(
@@ -236,7 +239,6 @@ def main() -> None:
                 outputs=[plot],
             )
 
-
             plot2 = gr.Plot(label="Plot")
             demo.load(
                 donut_chart,
@@ -244,7 +246,6 @@ def main() -> None:
                 outputs=[plot2],
             )
 
-
         gr.Markdown(
             """
             ## 👾 Contributors Hall of Fame