Changing to pie charts

app.py

@@ -27,13 +27,11 @@ def show_cluster(cl_id, num_clusters):
         img_path = "/".join([st.replace("/", "") for st in cl_dct['img_path_list'][i].split("//")][3:])
         images.append((Image.open(os.path.join("identities-images", img_path)), "_".join([img_path.split("/")[0], img_path.split("/")[-1]]).replace('Photo_portrait_of_an_','').replace('Photo_portrait_of_a_','').replace('SD_v2_random_seeds_identity_','(SD v.2) ').replace('dataset-identities-dalle2_','(Dall-E 2) ').replace('SD_v1.4_random_seeds_identity_','(SD v.1.4) ').replace('_',' ')))
     model_fig = go.Figure()
-    model_fig.add_trace(go.Bar(x=list(dict(cl_dct["labels_model"]).keys()), 
-                               y=list(dict(cl_dct["labels_model"]).values()),
-                               marker_color=px.colors.qualitative.G10))
+    model_fig.add_trace(go.Pie(labels=list(dict(cl_dct["labels_model"]).keys()), 
+                               values=list(dict(cl_dct["labels_model"]).values())))
     gender_fig = go.Figure()
-    gender_fig.add_trace(go.Bar(x=list(dict(cl_dct["labels_gender"]).keys()), 
-                                y=list(dict(cl_dct["labels_gender"]).values()),
-                                marker_color=px.colors.qualitative.G10))
+    gender_fig.add_trace(go.Pie(labels=list(dict(cl_dct["labels_gender"]).keys()), 
+                                values=list(dict(cl_dct["labels_gender"]).values())))
     ethnicity_fig = go.Figure()
     ethnicity_fig.add_trace(go.Bar(x=list(dict(cl_dct["labels_ethnicity"]).keys()), 
                                    y=list(dict(cl_dct["labels_ethnicity"]).values()), 
@@ -57,12 +55,15 @@ with gr.Blocks(title=TITLE) as demo:
             cluster_id = gr.Slider(minimum=0, maximum=num_clusters.value-1, step=1, value=0, label="Click to move between clusters")
             a = gr.Text(label="Number of images")
     with gr.Row():
-            c = gr.Plot(label="Model makeup of cluster")
-            b = gr.Plot(label="Gender label makeup of cluster")
-            d = gr.Plot(label="Ethnicity label makeup of cluster")
+            with gr.Column(scale=1):
+                c = gr.Plot(label="Model makeup of cluster")
+            with gr.Column(scale=1):
+                b = gr.Plot(label="Gender label makeup of cluster")
+            with gr.Column(scale=2):
+                d = gr.Plot(label="Ethnicity label makeup of cluster")
     demo.load(fn=show_cluster, inputs=[cluster_id, num_clusters], outputs=[a,b,c,d, gallery])
     num_clusters.change(fn=show_cluster, inputs=[cluster_id, num_clusters], outputs=[a,b,c,d, gallery])
     cluster_id.change(fn=show_cluster, inputs=[cluster_id, num_clusters], outputs=[a,b,c,d, gallery])
 
 if __name__ == "__main__":
-    demo.queue().launch(debug=True)
+    demo.queue().launch(debug=True)