feat: Add update colours button

app.py

@@ -232,40 +232,30 @@ DATASETS = [
 ]
 
 
-def main() -> None:
-    """Produce a radial plot."""
+def update_colour_mapping(results_dfs: dict[Language, pd.DataFrame]) -> None:
+    """Get a mapping from model ids to RGB triplets.
 
-    global last_fetch
-    results_dfs = fetch_results()
-    last_fetch = dt.datetime.now()
+    Args:
+        results_dfs:
+            The results dataframes for each language.
+    """
+    global colour_mapping
+    global seed
+    seed += 1
 
-    all_languages = sorted(
-        [language.name for language in ALL_LANGUAGES.values()],
-        key=lambda language_name: language_name.lower(),
-    )
-    danish_models = sorted(
-        list({model_id for model_id in results_dfs[DANISH].index}),
-        key=lambda model_id: model_id.lower(),
-    )
+    gr.Info(f"Updating colour mapping...")
 
     # Get distinct RGB values for all models
     all_models = list(
         {model_id for df in results_dfs.values() for model_id in df.index}
     )
-    colour_mapping: dict[str, tuple[int, int, int]] = dict()
+    colour_mapping = dict()
 
     for i in it.count():
         min_colour_distance = MIN_COLOUR_DISTANCE_BETWEEN_MODELS - i
-
-        if i > 0:
-            logger.info(
-                f"All retries failed. Trying again with min colour distance "
-                f"{min_colour_distance}."
-            )
-
         retries_left = 10 * len(all_models)
         for model_id in all_models:
-            random.seed(hash(model_id) + i)
+            random.seed(hash(model_id) + i + seed)
             r, g, b = 0, 0, 0
             too_bright, similar_to_other_model = True, True
             while (too_bright or similar_to_other_model) and retries_left > 0:
@@ -287,6 +277,28 @@ def main() -> None:
             )
             break
 
+
+def main() -> None:
+    """Produce a radial plot."""
+
+    global last_fetch
+    results_dfs = fetch_results()
+    last_fetch = dt.datetime.now()
+
+    all_languages = sorted(
+        [language.name for language in ALL_LANGUAGES.values()],
+        key=lambda language_name: language_name.lower(),
+    )
+    danish_models = sorted(
+        list({model_id for model_id in results_dfs[DANISH].index}),
+        key=lambda model_id: model_id.lower(),
+    )
+
+    global colour_mapping
+    global seed
+    seed = 4242
+    update_colour_mapping(results_dfs=results_dfs)
+
     with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
         gr.Markdown(INTRO_MARKDOWN)
 
@@ -340,6 +352,11 @@ def main() -> None:
                         interactive=True,
                         scale=1,
                     )
+                    update_colours_button = gr.Button(
+                        value="Update colours",
+                        interactive=True,
+                        scale=1,
+                    )
                 with gr.Row():
                     plot = gr.Plot(
                         value=produce_radial_plot(
@@ -349,7 +366,6 @@ def main() -> None:
                             show_scale=show_scale_checkbox.value,
                             plot_width=plot_width_slider.value,
                             plot_height=plot_height_slider.value,
-                            colour_mapping=colour_mapping,
                             results_dfs=results_dfs,
                         ),
                     )
@@ -371,7 +387,6 @@ def main() -> None:
         update_plot_kwargs = dict(
             fn=partial(
                 produce_radial_plot,
-                colour_mapping=colour_mapping,
                 results_dfs=results_dfs,
             ),
             inputs=[
@@ -391,6 +406,11 @@ def main() -> None:
         plot_width_slider.change(**update_plot_kwargs)
         plot_height_slider.change(**update_plot_kwargs)
 
+        # Update colours when the button is clicked
+        update_colours_button.click(
+            fn=partial(update_colour_mapping, results_dfs=results_dfs),
+        ).then(**update_plot_kwargs)
+
     demo.launch()
 
 
@@ -483,7 +503,6 @@ def produce_radial_plot(
     show_scale: bool,
     plot_width: int,
     plot_height: int,
-    colour_mapping: dict[str, tuple[int, int, int]],
     results_dfs: dict[Language, pd.DataFrame] | None,
 ) -> go.Figure:
     """Produce a radial plot as a plotly figure.
@@ -501,8 +520,6 @@ def produce_radial_plot(
             The width of the plot.
         plot_height:
             The height of the plot.
-        colour_mapping:
-            A mapping from model ids to RGB triplets.
         results_dfs:
             The results dataframes for each language.