Update app.py

app.py

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