Refactor functions for updating enabled embeddings in images module

tabs/images/events.py

@@ -390,12 +390,16 @@ def add_to_embeddings(custom_embedding, enabled_embeddings):
         )
 
 
-def remove_from_embeddings(enabled_embeddings, index):
-    enabled_embeddings.pop(index)
-    return (
-        gr.update(
-            value=enabled_embeddings
-        )
+def update_enabled_embeddings_list(enabled_embeddings):
+    return gr.update( # enabled_embeddings_list
+        value=enabled_embeddings,
+        choices=enabled_embeddings
+    )
+
+
+def update_enabled_embeddings(enabled_embeddings_list):
+    return gr.update( # enabled_embeddings
+        value=enabled_embeddings_list
     )
 
 

tabs/images/ui.py

@@ -155,12 +155,8 @@ def image_tab():
     # Embeddings
     custom_embedding.change(update_custom_embedding, custom_embedding, [custom_embedding_info])
     add_embedding.click(add_to_embeddings, [custom_embedding, enabled_embeddings], [custom_embedding, custom_embedding_info, enabled_embeddings])
-    for i in range(6):
-        globals()[f"embedding_remove_{i}"].click(
-            lambda enabled_embeddings, index=i: remove_from_embeddings(enabled_embeddings, index),
-            [enabled_embeddings],
-            [enabled_embeddings]
-        )
+    enabled_embeddings.change(update_enabled_embeddings_list, enabled_embeddings, [enabled_embeddings_list]) # type: ignore
+    enabled_embeddings_list.change(update_enabled_embeddings, enabled_embeddings_list, [enabled_embeddings]) # type: ignore
     
     # Generate Image
     generate.click(