make arrays

app.py

@@ -108,9 +108,12 @@ def donut_chart_total() -> alt.Chart:
         {
             "values": [annotated_records, pending_records],
             "category": [ANNOTATED, PENDING],
-            "colors": ["#27AE60", "#F1C40F"],  # Green for Completed, yellow for Remaining
+            "colors": ["#27AE60", "#F1C40F"],  # Green for Completed, Yellow for Remaining
         }
     )
+    
+    domain = source['category'].tolist()
+    range_ = source['colors'].tolist()
 
     base = alt.Chart(source).encode(
         theta=alt.Theta("values:Q", stack=True),
@@ -118,11 +121,8 @@ def donut_chart_total() -> alt.Chart:
             "values", scale=alt.Scale(type="sqrt", zero=True, rangeMin=20)
         ),
 
-        color=alt.Color(
-            "category:N",
-            scale=alt.Scale(domain=source['category'], range=source['colors']),  # Map colors to categories
-            legend=alt.Legend(title=CATEGORY)
-        ),
+
+    color=alt.Color(field="category", type="nominal", scale=alt.Scale(domain=domain, range=range_), legend=alt.Legend(title=CATEGORY)),
     )
 
     c1 = base.mark_arc(innerRadius=20, stroke="#fff")