template for examplars

app.py

@@ -133,7 +133,10 @@ def make_profession_table(num_clusters, prof_names, mod_name, max_cols=8):
         .to_html()
     )
 
-
+def show_examplars(num_clusters, prof_name, mod_name, cl_id):
+    # TODO: show the actual images
+    examplars_dict = clusters_dicts[num_clusters][df_models[mod_name]][prof_name]["cluster_examplars"][str(cl_id)]
+    return json.dumps(examplars_dict)
 
 with gr.Blocks() as demo:
     gr.Markdown("# 🤗 Diffusion Cluster Explorer")
@@ -183,31 +186,56 @@ with gr.Blocks() as demo:
     #        with gr.Accordion("Tag Frequencies", open=False):
 
     with gr.Tab("Profession Focus"):
-        with gr.Row():
-            num_clusters = gr.Radio(
-                [12, 24, 48],
-                value=12,
-                label="How many clusters do you want to use to represent identities?",
-            )
         with gr.Row():
             with gr.Column():
+                gr.Markdown("Select profession to visualize here:")
+                num_clusters_focus = gr.Radio(
+                    [12, 24, 48],
+                    value=12,
+                    label="How many clusters do you want to use to represent identities?",
+                )
                 profession_choice_focus = gr.Dropdown(
                     choices=professions,
                     value="social worker",
                     label="Select profession:",
                 )
+                gr.Markdown("You can show examples of profession images assigned to each cluster:")
+                model_choices_focus = gr.Dropdown(
+                    [
+                        "All Models",
+                        "Stable Diffusion 1.4",
+                        "Stable Diffusion 2",
+                        "Dall-E 2",
+                    ],
+                    value="All Models",
+                    label="Select generation model:",
+                    interactive=True,
+                )
+                cluster_id_focus = gr.Dropdown(
+                    choices=[i for i in range(num_clusters_focus.value)],
+                    value=0,
+                    label="Select cluster to visualize:",
+                )
             with gr.Column():
                 plot = gr.Plot(
                     label=f"Makeup of the cluster assignments for profession {profession_choice_focus}"
                 )
-                profession_choice_focus.change(
-                    make_profession_plot,
-                    [num_clusters, profession_choice_focus],
-                    plot,
+                for var in [num_clusters_focus, profession_choice_focus]:
+                    var.change(
+                        make_profession_plot,
+                        [num_clusters_focus, profession_choice_focus],
+                        plot,
+                        queue=False,
+                    )
+        with gr.Row():
+            examplars_plot = gr.JSON()          # TODO: turn this into a plot with the actual images
+            for var in [model_choices_focus, cluster_id_focus]:
+                var.change(
+                    show_examplars,
+                    [num_clusters_focus, profession_choice_focus, model_choices_focus, cluster_id_focus],
+                    examplars_plot,
                     queue=False,
                 )
-        with gr.Row():
-            gr.Markdown("TODO: show examplars for cluster")
 
 
 demo.launch()