Update app.py

app.py

@@ -141,6 +141,31 @@ highlighted_clustered_sentences = []
 for cluster in clustered_sentences:
     highlighted_clustered_sentences.extend(highlight_words_within_cluster(cluster, model, exclude_words))
     
+
+def calculate_similarity_score(sentences):
+    # Encode all sentences to get their embeddings
+    model = SentenceTransformer('all-MiniLM-L6-v2')
+    embeddings = model.encode(sentences)
+
+    # Calculate average cosine similarity
+    total_similarity = 0
+    comparisons = 0
+    for i in range(len(embeddings)):
+        for j in range(i+1, len(embeddings)):
+            # Cosine similarity between embeddings
+            similarity = 1 - cosine(embeddings[i], embeddings[j])
+            total_similarity += similarity
+            comparisons += 1
+
+    # Average similarity
+    average_similarity = total_similarity / comparisons if comparisons > 0 else 0
+
+    # Scale from [-1, 1] to [0, 100]
+    score_out_of_100 = (average_similarity + 1) / 2 * 100
+
+
+    return score_out_of_100
+
     
 text_list = []
 
@@ -178,7 +203,12 @@ def setTextVisibility(cbg, model_name_input):
     for idx, sentence in enumerate(highlighted_clustered_sentences):
         result.append("<p><strong>"+ cbg[idx] +"</strong></p><p>"+ sentence +"</p><br/>")
 
-    return result
+    score = round(calculate_similarity_score(sentences))
+
+    final_html = f"""<div>{result}<div style="text-align: center; font-size: 24px; font-weight: bold;">Similarity Score: {score}</div></div>"""
+
+
+    return final_html
     
 
     # update_show = [gr.Textbox(visible=True, label=text, value=answer_question(text, model_name_input)) for text in cbg]
@@ -224,7 +254,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         #btnExec.click(setTextVisibility, inputs=[cbg, model_name_input], outputs=text_list)
         btnExec.click(setTextVisibility, inputs=[cbg, model_name_input], outputs=html_result)
         gr.HTML("""
-        <div style="text-align: center; font-size: 24px; font-weight: bold;">Similarity Score: 76%</div>
+        <div style="text-align: center; font-size: 24px; font-weight: bold;">Similarity Score: </div>
                 """)
 
         clear = gr.ClearButton(link = "http://127.0.0.1:7865")