Update app.py

app.py

@@ -151,15 +151,22 @@ def label_visualization(clause2labels):
     print(aspect_dict, genericity_dict, boundedness_dict)
     fig, axs = plt.subplots(1, 3, figsize=(10, 6,))
     fig.tight_layout(pad=5.0)
-    axs[0].pie([float(v / total_clauses) for v in aspect_dict.values()], colors = sns.color_palette('pastel')[0:3], labels=aspect_dict.keys(), autopct='%.0f%%', normalize=True )
+    dict_aspect = {k : float(v / total_clauses) for k, v in aspect_dict.items() if v != 0}
+    dict_genericity = {k : float(v / total_clauses) for k, v in genericity_dict.items() if v != 0}
+    dict_boundedness = {k : float(v / total_clauses) for k, v in boundedness_dict.items() if v != 0}
+    print(dict_aspect)
+    print(list(dict_aspect.values()), len(dict_aspect.keys()), list(dict_aspect.keys()))
+    axs[0].pie(list(dict_aspect.values()), colors = sns.color_palette('pastel')[0:len(dict_aspect.keys())],
+                labels=dict_aspect.keys(), autopct='%.0f%%', normalize=True )
     axs[0].set_title("Aspect")
-    axs[1].pie([float(v / total_clauses) for v in genericity_dict.values()], colors = sns.color_palette('pastel')[3:6], labels=genericity_dict.keys(), autopct='%.0f%%', normalize=True)
+    axs[1].pie(list(dict_genericity.values()), colors = sns.color_palette('pastel')[3: 3 + len(dict_genericity.keys())],
+                labels=dict_genericity.keys(), autopct='%.0f%%', normalize=True)
     axs[1].set_title("Genericity")
-    axs[2].pie([float(v / total_clauses) for v in boundedness_dict.values()], colors = sns.color_palette('pastel')[6:10], labels=boundedness_dict.keys(), autopct='%.0f%%', normalize=True)
+    axs[2].pie(list(dict_boundedness.values()), colors = sns.color_palette('pastel')[6: 6 + len(dict_boundedness.keys())],
+                labels=dict_boundedness.keys(), autopct='%.0f%%', normalize=True)
     axs[2].set_title("Boundedness")
     return fig
 
-
 def run_pipeline(text):
     snippets = auto_split(text)
     print(snippets)