Update app.py

app.py

@@ -51,22 +51,23 @@ def compare_embeddings(*texts):
     
     return fig
 
-def add_textbox(textbox_container):
-    new_textbox = gr.Textbox(label=f"Text {len(textbox_container) + 1}")
-    textbox_container.append(new_textbox)
-    return textbox_container
+def add_textbox(textboxes):
+    new_textbox = gr.Textbox(label=f"Text {len(textboxes) + 1}")
+    return textboxes + [new_textbox]
 
-def remove_textbox(textbox_container):
-    if len(textbox_container) > 2:
-        textbox_container.pop()
-    return textbox_container
+def remove_textbox(textboxes):
+    if len(textboxes) > 2:
+        return textboxes[:-1]
+    return textboxes
 
 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")])
+    
     with gr.Column() as textbox_container:
-        textboxes = [gr.Textbox(label="Text 1"), gr.Textbox(label="Text 2")]
+        gr.components.Row(textboxes.value)
     
     with gr.Row():
         add_btn = gr.Button("Add String")
@@ -75,10 +76,11 @@ 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=[textbox_container, plot_output], value="Clear")
     
-    add_btn.click(add_textbox, inputs=[textbox_container], outputs=[textbox_container])
-    remove_btn.click(remove_textbox, inputs=[textbox_container], outputs=[textbox_container])
-    submit_btn.click(compare_embeddings, inputs=textbox_container, outputs=[plot_output])
+    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])
+    submit_btn.click(compare_embeddings, inputs=textboxes, outputs=[plot_output])
 
 iface.launch()