Update app.py

app.py

@@ -28,31 +28,52 @@ def reduce_to_3d(embedding):
     return embedding[:3]
 
 @spaces.GPU
-def compare_embeddings(text1, text2):
-    emb1 = get_embedding(text1)
-    emb2 = get_embedding(text2)
+def compare_embeddings(*texts):
+    embeddings = [get_embedding(text) for text in texts]
+    embeddings_3d = [reduce_to_3d(emb) for emb in embeddings]
     
-    emb1_3d = reduce_to_3d(emb1)
-    emb2_3d = reduce_to_3d(emb2)
+    fig = go.Figure()
     
-    fig = go.Figure(data=[
-        go.Scatter3d(x=[0, emb1_3d[0]], y=[0, emb1_3d[1]], z=[0, emb1_3d[2]], mode='lines+markers', name='Text 1'),
-        go.Scatter3d(x=[0, emb2_3d[0]], y=[0, emb2_3d[1]], z=[0, emb2_3d[2]], mode='lines+markers', name='Text 2')
-    ])
+    for i, emb in enumerate(embeddings_3d):
+        fig.add_trace(go.Scatter3d(x=[0, emb[0]], y=[0, emb[1]], z=[0, emb[2]], 
+                                   mode='lines+markers', name=f'Text {i+1}'))
     
     fig.update_layout(scene=dict(xaxis_title='X', yaxis_title='Y', zaxis_title='Z'))
     
     return fig
 
-iface = gr.Interface(
-    fn=compare_embeddings,
-    inputs=[
-        gr.Textbox(label="Text 1"),
-        gr.Textbox(label="Text 2")
-    ],
-    outputs=gr.Plot(),
-    title="3D Embedding Comparison",
-    description="Compare the embeddings of two strings visualized in 3D space using Mistral 7B."
-)
+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")
+    
+    with gr.Column() as input_column:
+        inputs = generate_text_boxes(2)
+    
+    output = gr.Plot()
+    
+    compare_button = gr.Button("Compare Embeddings")
+    
+    update_button.click(
+        update_interface,
+        inputs=[num_texts],
+        outputs=[input_column, output]
+    )
+    
+    compare_button.click(
+        compare_embeddings,
+        inputs=inputs,
+        outputs=output
+    )
 
 iface.launch()