map colors to categories

app.py

@@ -108,7 +108,7 @@ def donut_chart_total() -> alt.Chart:
         {
             "values": [annotated_records, pending_records],
             "category": [ANNOTATED, PENDING],
-            "colors": ["#27AE60", "#F1C40F"],  # Green for Completed, Grey for Remaining
+            "colors": ["#27AE60", "#F1C40F"],  # Green for Completed, Yellow for Remaining
         }
     )
 
@@ -117,7 +117,11 @@ def donut_chart_total() -> alt.Chart:
         radius=alt.Radius(
             "values", scale=alt.Scale(type="sqrt", zero=True, rangeMin=20)
         ),
-        color=alt.Color("category:N", range=source['colors'], legend=alt.Legend(title=CATEGORY)),
+
+        color=alt.Color(
+            "category:N",
+            scale=alt.Scale(domain=source['category'], range=source['colors']),  # Map colors to categories
+            legend=alt.Legend(title=CATEGORY)
     )
 
     c1 = base.mark_arc(innerRadius=20, stroke="#fff")