implement power

app.py

@@ -21,6 +21,7 @@ max_tabs = 10
 input_images = [None for i in range(max_tabs)]
 input_prompts = [None for i in range(max_tabs)]
 embedding_plots = [None for i in range(max_tabs)]
+embedding_powers = [1. for i in range(max_tabs)]
 # global embedding_base64s
 embedding_base64s = [None for i in range(max_tabs)]
 # embedding_base64s = gr.State(value=[None for i in range(max_tabs)])
@@ -136,19 +137,14 @@ def on_prompt_change_update_embeddings(prompt):
     embeddings_b64 = embedding_to_base64(embeddings)
     return gr.Text.update(embeddings_b64)
 
-# def on_embeddings_changed_update_average_embeddings(last_embedding_base64):
-# def on_embeddings_changed_update_average_embeddings(embedding_base64s):
-def on_embeddings_changed_update_average_embeddings(embedding_base64s_state, embedding_base64, idx):
-    # global embedding_base64s
+def update_average_embeddings(embedding_base64s_state, embedding_powers):
     final_embedding = None
     num_embeddings = 0
-    embedding_base64s_state[idx] = embedding_base64 if embedding_base64 != '' else None
-    # for textbox in embedding_base64s:
-    #     embedding_base64 = textbox.value
-    for embedding_base64 in embedding_base64s_state:
+    for i, embedding_base64 in enumerate(embedding_base64s_state):
         if embedding_base64 is None or embedding_base64 == "":
             continue
         embedding = base64_to_embedding(embedding_base64)
+        embedding = embedding * embedding_powers[i]
         if final_embedding is None:
             final_embedding = embedding
         else:
@@ -161,6 +157,16 @@ def on_embeddings_changed_update_average_embeddings(embedding_base64s_state, emb
         return gr.Text.update('')
     final_embedding = final_embedding / num_embeddings
     embeddings_b64 = embedding_to_base64(final_embedding)
+    return embeddings_b64
+
+def on_power_change_update_average_embeddings(embedding_base64s_state, embedding_power_state, power, idx):
+    embedding_power_state[idx] = power
+    embeddings_b64 = update_average_embeddings(embedding_base64s_state, embedding_power_state)
+    return gr.Text.update(embeddings_b64)
+
+def on_embeddings_changed_update_average_embeddings(embedding_base64s_state, embedding_power_state, embedding_base64, idx):
+    embedding_base64s_state[idx] = embedding_base64 if embedding_base64 != '' else None
+    embeddings_b64 = update_average_embeddings(embedding_base64s_state, embedding_power_state)
     return gr.Text.update(embeddings_b64)    
 
 def on_embeddings_changed_update_plot(embeddings_b64):
@@ -328,8 +334,13 @@ Try uploading a few images and/or add some text prompts and click generate image
                     with gr.Column(scale=2, min_width=240):
                         input_prompts[i] = gr.Textbox(label="Text Prompt", show_label=True)
                     with gr.Column(scale=3, min_width=600):
-                        with gr.Accordion(f"Embeddings (base64)", open=False):
-                            embedding_base64s[i] = gr.Textbox(show_label=False)
+                        with gr.Row():
+                            # with gr.Slider(min=-5, max=5, value=1, label="Power", show_label=True):
+                            #     embedding_powers[i] = gr.Slider.value
+                            embedding_powers[i] = gr.Slider(minimum=-3, maximum=3, value=1, label="Power", show_label=True, interactive=True)
+                        with gr.Row():
+                            with gr.Accordion(f"Embeddings (base64)", open=False):
+                                embedding_base64s[i] = gr.Textbox(show_label=False)
                 for idx, (tab_title, examples) in enumerate(tabbed_examples.items()):
                     with gr.Tab(tab_title):
                         with gr.Row():
@@ -372,15 +383,14 @@ Try uploading a few images and/or add some text prompts and click generate image
         output = gr.Gallery(label="Generated variations")
 
     embedding_base64s_state = gr.State(value=[None for i in range(max_tabs)])
+    embedding_power_state = gr.State(value=[1. for i in range(max_tabs)])
     for i in range(max_tabs):
         input_images[i].change(on_image_load_update_embeddings, input_images[i], [embedding_base64s[i]])
-        # input_prompts[i].submit(on_prompt_change_update_embeddings, input_prompts[i], [embedding_base64s[i]])
         input_prompts[i].change(on_prompt_change_update_embeddings, input_prompts[i], [embedding_base64s[i]])
         embedding_base64s[i].change(on_embeddings_changed_update_plot, embedding_base64s[i], [embedding_plots[i]])
-        # embedding_plots[i].change(on_plot_changed, embedding_base64s[i], average_embedding_base64)
-        # embedding_plots[i].change(on_embeddings_changed_update_average_embeddings, embedding_base64s[i], average_embedding_base64)
         idx_state = gr.State(value=i)
-        embedding_base64s[i].change(on_embeddings_changed_update_average_embeddings, [embedding_base64s_state, embedding_base64s[i], idx_state], average_embedding_base64)
+        embedding_base64s[i].change(on_embeddings_changed_update_average_embeddings, [embedding_base64s_state, embedding_power_state, embedding_base64s[i], idx_state], average_embedding_base64)
+        embedding_powers[i].change(on_power_change_update_average_embeddings, [embedding_base64s_state, embedding_power_state, embedding_powers[i], idx_state], average_embedding_base64)
 
     average_embedding_base64.change(on_embeddings_changed_update_plot, average_embedding_base64, average_embedding_plot)