fix: change plddt color on selecting new model

app.py

@@ -157,20 +157,11 @@ def load_result() -> Optional[AlphaFoldData]:
 
 @solara.component
 def Page():
-    color_data = solara.use_reactive(NO_COLOR_DATA)
-    tooltip_data = solara.use_reactive(NO_TOOLTIP_DATA)
-
     color_mode = solara.use_reactive("chain")
     dark_effective = solara.lab.use_dark_effective()
 
     def on_color_mode(value: str):
         color_mode.set(value)
-        if value == "chain":
-            color_data.set(NO_COLOR_DATA)
-            tooltip_data.set(NO_TOOLTIP_DATA)
-        else:
-            color_data.set(load_result.value.color_data)
-            tooltip_data.set(load_result.value.tooltip_data)
 
     def set_result_index(value: int):
         result_index.set(value)
@@ -248,13 +239,20 @@ def Page():
     elif load_result.finished:
         fold_data: AlphaFoldData = load_result.value
 
-        with solara.Card():
+        if color_mode.value == "chain":
+            color_data = NO_COLOR_DATA
+            tooltip_data = NO_TOOLTIP_DATA
+        else:
+            color_data = fold_data.color_data
+            tooltip_data = fold_data.tooltip_data
+
+        with solara.Card(fold_data.name):
             theme = "dark" if dark_effective else "light"
             PDBeMolstar.element(
                 height="calc(100vh - 150px)",
                 custom_data=fold_data.custom_data,
-                color_data=color_data.value,
-                tooltips=tooltip_data.value,
+                color_data=color_data,
+                tooltips=tooltip_data,
                 show_water=False,
                 theme=theme,
             ).key(f"pdbemolstar-{dark_effective}")