Update app.py

app.py

@@ -33,7 +33,7 @@ def generate_random_color():
 
 @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()
@@ -61,7 +61,8 @@ class DynamicInterface:
             gr.Markdown("# 3D Embedding Comparison")
             gr.Markdown("Compare the embeddings of strings visualized in 3D space using Mistral 7B.")
             
-            self.string_boxes = [gr.Textbox(label=f"Text {i+1}") for i in range(self.num_strings)]
+            with gr.Column() as self.string_column:
+                self.string_boxes = [gr.Textbox(label=f"Text {i+1}") for i in range(self.num_strings)]
             
             with gr.Row():
                 add_btn = gr.Button("Add String")
@@ -73,30 +74,28 @@ class DynamicInterface:
             
             output = gr.Plot()
             
-            add_btn.click(self.add_string, inputs=[], outputs=[interface])
-            remove_btn.click(self.remove_string, inputs=[], outputs=[interface])
-            clear_btn.click(self.clear_inputs, inputs=self.string_boxes, outputs=self.string_boxes)
+            add_btn.click(self.add_string, outputs=[self.string_column])
+            remove_btn.click(self.remove_string, outputs=[self.string_column])
+            clear_btn.click(self.clear_inputs, outputs=self.string_boxes)
             submit_btn.click(compare_embeddings, inputs=self.string_boxes, outputs=output)
         
         return interface
 
     def add_string(self):
         self.num_strings += 1
-        self.string_boxes.append(gr.Textbox(label=f"Text {self.num_strings}"))
-        return self.rebuild_interface()
+        new_textbox = gr.Textbox(label=f"Text {self.num_strings}")
+        self.string_boxes.append(new_textbox)
+        return gr.Column.update(visible=True, children=self.string_boxes)
 
     def remove_string(self):
         if self.num_strings > 2:
             self.num_strings -= 1
             self.string_boxes.pop()
-            return self.rebuild_interface()
-        return None
+            return gr.Column.update(visible=True, children=self.string_boxes)
+        return gr.Column.update()
 
-    def clear_inputs(self, *inputs):
-        return ["" for _ in inputs]
-
-    def rebuild_interface(self):
-        return gr.update(visible=False), self.build_interface()
+    def clear_inputs(self):
+        return ["" for _ in self.string_boxes]
 
     def launch(self):
         self.interface.launch()