feat: Add update colours buttom

app.py

@@ -333,6 +333,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(
@@ -442,6 +447,21 @@ def main() -> None:
             ],
             outputs=plot,
         )
+        update_colours_button.click(
+            fn=partial(update_colour_mapping, from_scratch=True),
+            inputs=model_ids_dropdown
+        ).then(
+            fn=partial(produce_radial_plot, results_dfs=results_dfs),
+            inputs=[
+                model_ids_dropdown,
+                language_names_dropdown,
+                use_rank_score_checkbox,
+                show_scale_checkbox,
+                plot_width_slider,
+                plot_height_slider,
+            ],
+            outputs=plot,
+        )
 
     demo.launch()
 
@@ -782,12 +802,15 @@ def fetch_results() -> dict[Language, pd.DataFrame]:
     return results_dfs
 
 
-def update_colour_mapping(model_ids: list[str]) -> None:
+def update_colour_mapping(model_ids: list[str], from_scratch: bool = False) -> None:
     """Get a mapping from model ids to RGB triplets.
 
     Args:
         model_ids:
-            The model ids to update the colour
+            The model ids to update the colour.
+        from_scratch:
+            Whether to reset the existing colour mapping and build a new one from
+            scratch. Defaults to False.
     """
     if not model_ids:
         return
@@ -796,6 +819,9 @@ def update_colour_mapping(model_ids: list[str]) -> None:
     global seed
     seed += 1
 
+    if from_scratch:
+        colour_mapping = dict()
+
     for i in it.count():
         min_colour_distance = MIN_COLOUR_DISTANCE_BETWEEN_MODELS - i
         retries_left = 10 * len(model_ids)