Update app.py

app.py

@@ -67,9 +67,8 @@ 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)
-    print(annotated_records)
+    pending_records = int(os.getenv("TARGET_RECORDS")) - annotated_records
 
     # Prepare data for the donut chart
     source = pd.DataFrame({
@@ -102,8 +101,7 @@ def kpi_chart_remaining() -> alt.Chart:
     """
 
     source_dataset, results = obtain_source_target_datasets()
-    pending_records = len(source_dataset)
-
+    pending_records = int(os.getenv("TARGET_RECORDS")) - len(results)
     # Assuming you have a DataFrame with user data, create a sample DataFrame
     data = pd.DataFrame({"Category": ["Total remaining"], "Value": [pending_records]})