Update app.py

app.py

@@ -63,51 +63,33 @@ def get_user_annotations_dictionary(
     return output
 
 
-
-import altair as alt
-import pandas as pd
-import os
-
-import altair as alt
-import pandas as pd
-import os
-
 def donut_chart() -> alt.Chart:
     # Load your data
     source_dataset, results = obtain_source_target_datasets()
     pending_records = len(source_dataset)
     annotated_records = len(results)
+    
+    source = pd.DataFrame(
+        {
+            "values": [annotated_records, pending_records],
+            "category": ["Annotated", "Pending"],  # Add a new column for categories
+        }
+    )
 
-    # 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(field="values", type="quantitative"),
-        color=alt.Color(field="color", type="nominal", legend=None),  # Use the colors defined in the DataFrame
-        tooltip=["category", "values"]
+        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")),
     )
 
-    # Create the arcs for the donut chart
-    arcs = base.mark_arc(innerRadius=100, outerRadius=120, stroke="#fff")
+    c1 = base.mark_arc(innerRadius=20, stroke="#fff")
 
-    # Add text labels
-    text = base.mark_text(radiusOffset=20, size=20).encode(
-        text=alt.Text("values:Q"),
-        color=alt.value('black')  # Setting text color to black for better readability
-    )
-
-    # Combine the arcs and text
-    chart = arcs + text
+    c2 = base.mark_text(radiusOffset=10).encode(text="values:Q")
 
-    # Set the chart background to transparent
-    chart = chart.configure_view(strokeOpacity=0)
+    chart = c1 + c2
 
-    return chart