Update app.py

app.py

@@ -13,27 +13,25 @@ def get_embedding(text):
     inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True)
     with torch.no_grad():
         outputs = model(**inputs)
-    # Use .squeeze() to remove single-dimensional entries from the shape of the array.
     return outputs.pooler_output.squeeze().numpy()
 
 def compare_texts(*inputs):
     categories = ["Hobbies", "Favorite Foods", "Travel Preferences", "Music Taste", "Movie/TV Preferences"]
     similarities = []
+    results = []
     for i in range(len(categories)):
         text1, text2 = inputs[i*2], inputs[i*2 + 1]
         embedding1 = get_embedding(text1)
         embedding2 = get_embedding(text2)
-        # Ensure both embeddings are 2D by using [embedding1], [embedding2].
         similarity = cosine_similarity([embedding1], [embedding2])[0][0]
         similarities.append(similarity)
-        print(f"{categories[i]} similarity: {similarity:.2f}")
+        results.append(f"{categories[i]} similarity: {similarity:.2f}%")
+        print(f"{categories[i]} similarity: {similarity:.2f}%")
     
     average_similarity = np.mean(similarities)
     percentage_similarity = round(average_similarity * 100, 2)
-    return f"Overall Compatibility: {percentage_similarity}%"
-
-# Remainder of your Gradio application setup remains the same
-
+    results.append(f"Overall Compatibility: {percentage_similarity}%")
+    return "\n".join(results)
 
 with gr.Blocks() as demo:
     gr.Markdown("# Couple Compatibility Checker")
@@ -45,7 +43,7 @@ with gr.Blocks() as demo:
             inputs.append(gr.Textbox(label=f"Person 1's {category}"))
             inputs.append(gr.Textbox(label=f"Person 2's {category}"))
     btn_compare = gr.Button("Compare")
-    result = gr.Textbox(label="Compatibility Result", interactive=False, lines=3)
+    result = gr.Textbox(label="Compatibility Results", interactive=False, lines=10)
 
     btn_compare.click(compare_texts, inputs=inputs, outputs=result)