Update app.py

app.py

@@ -62,6 +62,48 @@ def get_user_annotations_dictionary(
 
     return output
 
+import altair as alt
+import pandas as pd
+import os
+
+def progress_bar_chart() -> alt.Chart:
+    source_dataset, _ = obtain_source_target_datasets()
+    annotated_records = len(source_dataset)
+    total_records = int(os.getenv("TARGET_RECORDS"))
+    percentage_complete = annotated_records / total_records * 100
+
+    # Data for the progress bar
+    progress_data = pd.DataFrame({
+        'category': ['Progress'],
+        'percentage': [percentage_complete]
+    })
+
+    # Data for the total label
+    total_label = pd.DataFrame({
+        'category': ['Total'],
+        'percentage': [100],
+        'label': [f'{annotated_records} / {total_records}']
+    })
+
+    # Progress bar
+    progress_bar = alt.Chart(progress_data).mark_bar(color='lightgreen', size=50).encode(
+        x=alt.X('percentage:Q', axis=alt.Axis(title='Completion Percentage', format='%'), scale=alt.Scale(domain=(0, 100)))
+    )
+
+    # Total label
+    total_text = alt.Chart(total_label).mark_text(dx=5, dy=-5, align='left', fontSize=15, fontWeight='bold').encode(
+        x=alt.X('percentage:Q'),
+        text=alt.Text('label:N')
+    )
+
+    # Combine the bar and the label
+    chart = progress_bar + total_text
+
+    # Optionally, add a title
+    chart = chart.properties(title='Progress Towards Goal')
+
+    return chart
+
 
 def donut_chart() -> alt.Chart:
     """
@@ -176,7 +218,7 @@ def main() -> None:
         )
         plot = gr.Plot(label="Plot")
         demo.load(
-            donut_chart,
+            progress_bar_chart,
             inputs=[],
             outputs=[plot],
         )