Update app.py

app.py

@@ -137,6 +137,19 @@ def process_upload(api_key, files, lang):
     else:
         return f"No valid documents were indexed. Errors: {'; '.join(error_messages)}", None
 
+# Define the calculate_similarity function
+def calculate_similarity(response, ground_truth):
+    response_embedding = sentence_model.encode(response, convert_to_tensor=True)
+    truth_embedding = sentence_model.encode(ground_truth, convert_to_tensor=True)
+    
+    # Normalize the embeddings
+    response_embedding = response_embedding / response_embedding.norm(p=2)
+    truth_embedding = truth_embedding / truth_embedding.norm(p=2)
+    
+    # Calculate cosine similarity using sklearn's cosine_similarity function
+    similarity = cosine_similarity(response_embedding.reshape(1, -1), truth_embedding.reshape(1, -1))[0][0]
+    return similarity * 100  # Convert to percentage
+
 # This is the missing query_app function that needs to be defined
 def query_app(query, model_name, use_similarity_check, openai_api_key):
     global vector_index, query_log