Update app.py

app.py

@@ -63,6 +63,47 @@ def get_user_annotations_dictionary(
     return output
 
 
+def donut_chart2() -> alt.Chart:
+    # Load your data
+    source_dataset, results = obtain_source_target_datasets()
+    pending_records = len(source_dataset)
+    annotated_records = len(results)
+
+    # Prepare data for the donut chart
+    source = pd.DataFrame({
+        "values": [annotated_records, pending_records],
+        "category": ["Completed", "Remaining"],
+        "colors": ["#4CAF50", "#757575"]  # Green for Completed, Grey for Remaining
+    })
+
+    # Base chart for donut segments
+    base = alt.Chart(source).encode(
+        theta=alt.Theta(field="values", type="quantitative", stack=True),  # The angle encoding
+        color=alt.Color(field="colors", type="nominal", legend=alt.Legend(title="Category"), scale=None),  # Direct color specification
+        tooltip=['category', 'values']  # Tooltips for interactivity
+    )
+
+    # Arc marks for donut segments
+    arcs = base.mark_arc(innerRadius=100, outerRadius=150, stroke="#fff")  # Increased inner radius for a thicker donut
+
+    # Text marks for labels, adjusted to be placed within the donut segments
+    text = base.mark_text(radiusOffset=-70, size=16, color='white').encode(
+        text=alt.Text(field="values", type="quantitative"),  # The text to display
+    )
+
+    # Combine the arcs and text
+    chart = arcs + text
+
+    # Configure the view to have a transparent background
+    chart = chart.configure_view(
+        strokeWidth=0,  # Remove border around chart
+        fill=None  # Ensure background is transparent
+    ).configure_axis(
+        grid=False  # Turn off the grid
+    )
+
+    return chart
+
 def donut_chart() -> alt.Chart:
     """
     This function returns a donut chart with the number of annotated and pending records.
@@ -175,7 +216,7 @@ def main() -> None:
         )
         plot = gr.Plot(label="Plot")
         demo.load(
-            donut_chart,
+            donut_chart2,
             inputs=[],
             outputs=[plot],
         )