Update app.py

app.py

@@ -29,7 +29,7 @@ def reduce_to_3d(embedding):
 
 @spaces.GPU
 def compare_embeddings(*texts):
-    embeddings = [get_embedding(text) for text in texts]
+    embeddings = [get_embedding(text) for text in texts if text.strip()]
     embeddings_3d = [reduce_to_3d(emb) for emb in embeddings]
     
     fig = go.Figure()
@@ -45,34 +45,31 @@ def compare_embeddings(*texts):
 def generate_text_boxes(n):
     return [gr.Textbox(label=f"Text {i+1}") for i in range(n)]
 
-def update_interface(n):
-    new_inputs = generate_text_boxes(n)
-    return new_inputs, gr.Plot()
-
 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.Row():
-        num_texts = gr.Slider(minimum=2, maximum=10, step=1, value=2, label="Number of texts to compare")
-        update_button = gr.Button("Update")
+    num_texts = gr.Slider(minimum=2, maximum=10, step=1, value=2, label="Number of texts to compare")
     
     with gr.Column() as input_column:
-        inputs = generate_text_boxes(2)
+        text_boxes = gr.Group(generate_text_boxes(2))
     
     output = gr.Plot()
     
     compare_button = gr.Button("Compare Embeddings")
     
-    update_button.click(
+    def update_interface(n):
+        return gr.Group.update(visible=False), gr.Group.update(visible=True, components=generate_text_boxes(n))
+
+    num_texts.change(
         update_interface,
         inputs=[num_texts],
-        outputs=[input_column, output]
+        outputs=[text_boxes, text_boxes]
     )
     
     compare_button.click(
         compare_embeddings,
-        inputs=inputs,
+        inputs=text_boxes,
         outputs=output
     )