v2p2

app.py

@@ -33,7 +33,7 @@ def generate_random_color():
 
 @spaces.GPU
 def compare_embeddings(strings):
-    embeddings = [get_embedding(text) for text in strings]
+    embeddings = [get_embedding(text) for text in strings if text.strip()]
     embeddings_3d = [reduce_to_3d(emb) for emb in embeddings]
     
     fig = go.Figure()
@@ -53,14 +53,14 @@ def compare_embeddings(strings):
     
     return fig
 
-def add_string(strings):
-    strings.append("")
-    return strings, gr.Textbox.update(visible=True)
+def add_string(string_list):
+    string_list.append("")
+    return string_list
 
-def remove_string(strings):
-    if len(strings) > 1:
-        strings.pop()
-    return strings, gr.Textbox.update(visible=len(strings) > 1)
+def remove_string(string_list):
+    if len(string_list) > 1:
+        string_list.pop()
+    return string_list
 
 with gr.Blocks(title="3D Embedding Comparison") as iface:
     gr.Markdown("# 3D Embedding Comparison")
@@ -69,7 +69,9 @@ with gr.Blocks(title="3D Embedding Comparison") as iface:
     with gr.Row():
         with gr.Column(scale=1):
             string_inputs = gr.State([""])
-            textboxes = gr.Textbox(label="Text 1", elem_id="textbox_1")
+            textboxes = gr.Textbox.update(visible=True, label="Text 1")
+            with gr.Column(variant="panel") as textbox_container:
+                pass
             add_button = gr.Button("Add String")
             remove_button = gr.Button("Remove String")
         
@@ -78,18 +80,41 @@ with gr.Blocks(title="3D Embedding Comparison") as iface:
     
     submit_button = gr.Button("Compare Embeddings")
 
-    def update_textboxes(strings):
-        return [gr.Textbox.update(value=s, label=f"Text {i+1}", visible=True) for i, s in enumerate(strings)]
+    def update_textboxes(string_list):
+        components = []
+        for i, s in enumerate(string_list):
+            components.append(gr.Textbox(value=s, label=f"Text {i+1}"))
+        return gr.Column.update(visible=True, children=components)
 
-    add_button.click(add_string, inputs=[string_inputs], outputs=[string_inputs, textboxes])
-    remove_button.click(remove_string, inputs=[string_inputs], outputs=[string_inputs, textboxes])
+    add_button.click(
+        add_string,
+        inputs=[string_inputs],
+        outputs=[string_inputs],
+        then=lambda: update_textboxes(string_inputs.value)
+    ).then(
+        lambda: gr.Column.update(visible=True),
+        None,
+        textbox_container
+    )
+
+    remove_button.click(
+        remove_string,
+        inputs=[string_inputs],
+        outputs=[string_inputs],
+        then=lambda: update_textboxes(string_inputs.value)
+    ).then(
+        lambda: gr.Column.update(visible=True),
+        None,
+        textbox_container
+    )
     
     submit_button.click(
-        compare_embeddings,
+        lambda x: [t.strip() for t in x if t.strip()],
+        inputs=[string_inputs],
+        outputs=[string_inputs],
+        then=lambda x: compare_embeddings(x),
         inputs=[string_inputs],
         outputs=[plot_output]
     )
 
-    string_inputs.change(update_textboxes, inputs=[string_inputs], outputs=[textboxes])
-
 iface.launch()