Update app.py

app.py

@@ -64,11 +64,16 @@ def get_user_annotations_dictionary(
 
 
 def donut_chart() -> alt.Chart:
-    # Load your data
-    source_dataset, results = obtain_source_target_datasets()
-    pending_records = len(source_dataset)
-    annotated_records = len(results)
-    
+    """
+    This function returns a donut chart with the number of annotated and pending records.
+    Returns:
+        An altair chart with the donut chart.
+    """
+
+    source_dataset, _ = obtain_source_target_datasets()
+    annotated_records = len(source_dataset)
+    pending_records = int(os.getenv("TARGET_RECORDS")) - annotated_records
+
     source = pd.DataFrame(
         {
             "values": [annotated_records, pending_records],
@@ -90,6 +95,7 @@ def donut_chart() -> alt.Chart:
 
     chart = c1 + c2
 
+    return chart