Update app.py

app.py

@@ -64,42 +64,49 @@ def get_user_annotations_dictionary(
 
 
 
-def donut_chart() -> alt.Chart:
-    """
-    This function returns a donut chart with the number of annotated and pending records.
-
-    Returns:
-        An altair chart with the donut chart.
-    """
+import altair as alt
+import pandas as pd
+import os
 
+def donut_chart() -> alt.Chart:
+    # Load your data
     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],
-            "category": ["Submitted", "Pending"],  # Add a new column for categories
-        }
-    )
+    pending_records = len(source_dataset)
+    annotated_records = int(os.getenv("TARGET_RECORDS")) - annotated_records
+
+    # Prepare data for the donut chart
+    source = pd.DataFrame({
+        "values": [annotated_records, pending_records],
+        "category": ["Submitted", "Pending"],
+        "color": ["#28a745", "#dcdcdc"]  # Green for submitted, grey for pending
+    })
 
+    # Create the base of the donut chart
     base = alt.Chart(source).encode(
-        theta=alt.Theta("values:Q", stack=True),
-        radius=alt.Radius(
-            "values", scale=alt.Scale(type="sqrt", zero=True, rangeMin=20)
-        ),
-        color=alt.Color("category:N", legend=alt.Legend(title="Category")),
+        theta=alt.Theta(field="values", type="quantitative", stack=True),
+        color=alt.Color(field="color", type="nominal", legend=None),  # Use the colors defined in the DataFrame
+        tooltip=["category", "values"]
     )
 
-    c1 = base.mark_arc(innerRadius=20, stroke="#fff")
+    # Create the arcs for the donut chart
+    arcs = base.mark_arc(innerRadius=100, outerRadius=120, stroke="#fff")
 
-    c2 = base.mark_text(radiusOffset=10).encode(text="values:Q")
+    # Add text labels in the middle of the donut chart
+    text = base.mark_text(radius=140, size=20, align='center').encode(
+        text=alt.Text("values:Q"),
+        angle=alt.Angle(field="values", type="quantitative", aggregate='sum', stack=True)  # Calculate the mid-angle position for the labels
+    )
+
+    # Combine the arcs and text
+    chart = arcs + text
 
-    chart = c1 + c2
+    # Set the chart background to transparent
+    chart = chart.configure_view(strokeOpacity=0)
 
     return chart
 
 
+
 def kpi_chart() -> alt.Chart:
     """
     This function returns a KPI chart with the total amount of annotators.