Replace with pie chart

app.py

@@ -83,22 +83,30 @@ def compute_class_areas(predictions, image_shape):
     return class_areas
 
 
-def generate_coverage_graph(class_areas):
+def generate_coverage_pie_chart(class_areas):
     """
-    Generates a graph for class coverage percentages.
+    Generates a pie chart for class coverage percentages.
     """
+    total_percentage = sum(class_areas.values())
+    other_percentage = max(0, 100 - total_percentage)
+
+    labels = list(class_areas.keys()) + (["Other"] if other_percentage > 0 else [])
+    sizes = list(class_areas.values()) + ([other_percentage] if other_percentage > 0 else [])
+
     plt.figure(figsize=(8, 6))
-    classes = list(class_areas.keys())
-    coverage = list(class_areas.values())
-    plt.bar(classes, coverage, color="skyblue")
-    plt.xlabel("Class")
-    plt.ylabel("Coverage Percentage")
-    plt.title("Class Coverage as % of Total Image")
-    plt.ylim(0, 100)  # Always show the Y-axis up to 100%
-    graph_path = "class_coverage.png"
-    plt.savefig(graph_path)
+    plt.pie(
+        sizes,
+        labels=[f"{label} ({size:.1f}%)" for label, size in zip(labels, sizes)],
+        autopct="%1.1f%%",
+        startangle=90,
+        colors=plt.cm.tab20.colors
+    )
+    plt.axis("equal")
+    plt.title("Class Coverage Distribution")
+    chart_path = "class_coverage_pie.png"
+    plt.savefig(chart_path)
     plt.close()
-    return graph_path
+    return chart_path
 
 
 def coral_ai_inference(image, model_name):
@@ -115,12 +123,12 @@ def coral_ai_inference(image, model_name):
 
     # Generate class coverage
     class_areas = compute_class_areas(results[0], image.shape[:2])
-    graph_path = generate_coverage_graph(class_areas)
+    pie_chart_path = generate_coverage_pie_chart(class_areas)
 
     # Render the prediction
     rendered_image = results[0].plot()
 
-    return rendered_image, graph_path
+    return rendered_image, pie_chart_path
 
 
 # Dynamically generate Gradio examples
@@ -143,7 +151,7 @@ inputs = [
 
 outputs = [
     gr.Image(type="numpy", label="Segmented Image"),
-    gr.Image(type="filepath", label="Class Coverage Graph"),
+    gr.Image(type="filepath", label="Class Coverage Pie Chart"),
 ]
 
 # Icons with links