Update app.py

app.py

@@ -51,23 +51,22 @@ def compare_embeddings(*texts):
     
     return fig
 
-def add_textbox(textboxes):
-    new_textbox = gr.Textbox(label=f"Text {len(textboxes) + 1}")
-    return textboxes + [new_textbox]
+def add_textbox(num_textboxes):
+    return gr.Textbox.update(visible=True, label=f"Text {num_textboxes + 1}"), num_textboxes + 1
 
-def remove_textbox(textboxes):
-    if len(textboxes) > 2:
-        return textboxes[:-1]
-    return textboxes
+def remove_textbox(num_textboxes):
+    return gr.Textbox.update(visible=False), max(2, num_textboxes - 1)
 
 with gr.Blocks() as iface:
     gr.Markdown("# 3D Embedding Comparison")
     gr.Markdown("Compare the embeddings of multiple strings visualized in 3D space using Mistral 7B.")
     
-    textboxes = gr.State([gr.Textbox(label="Text 1"), gr.Textbox(label="Text 2")])
+    num_textboxes = gr.State(value=2)
     
     with gr.Column() as textbox_container:
-        gr.components.Row(textboxes.value)
+        textboxes = [gr.Textbox(label=f"Text {i+1}") for i in range(10)]  # Create 10 textboxes
+        for i in range(2, 10):  # Hide textboxes 3-10 initially
+            textboxes[i].visible = False
     
     with gr.Row():
         add_btn = gr.Button("Add String")
@@ -76,11 +75,10 @@ with gr.Blocks() as iface:
     plot_output = gr.Plot()
     
     submit_btn = gr.Button("Submit")
-    clear_btn = gr.ClearButton(components=[textbox_container, plot_output], value="Clear")
+    clear_btn = gr.ClearButton(components=textboxes + [plot_output], value="Clear")
     
-    add_btn.click(add_textbox, inputs=[textboxes], outputs=[textboxes])
-    remove_btn.click(remove_textbox, inputs=[textboxes], outputs=[textboxes])
-    textboxes.change(lambda x: gr.components.Row(x), inputs=[textboxes], outputs=[textbox_container])
+    add_btn.click(add_textbox, inputs=[num_textboxes], outputs=[textboxes[-1], num_textboxes])
+    remove_btn.click(remove_textbox, inputs=[num_textboxes], outputs=[textboxes[-1], num_textboxes])
     submit_btn.click(compare_embeddings, inputs=textboxes, outputs=[plot_output])
 
 iface.launch()