Update final_book_retriever.py

final_book_retriever.py

@@ -38,14 +38,14 @@ def answer(query, min_similarity=0.7):
     # Retrieve document details for documents with similarity score greater than min_similarity
     ranked_documents = []
     for index in ranked_indices:
-        similarity_score = similarities[0][index]
+        similarity_score = round(similarities[0][index], 2)  # Round similarity score to two decimal places
         if similarity_score > min_similarity:
             document_details = {
                 "Book": data['Book Title'].iloc[index],
                 "Author": data['Author'].iloc[index],
                 "Edition": data['Edition'].iloc[index],
-                "File Name": data['File_name'].iloc[index]
-                #"Similarity Score": similarity_score
+                "File Name": data['File_name'].iloc[index],
+                "Similarity Score": similarity_score
             }
             ranked_documents.append(document_details)
         else:
@@ -54,7 +54,7 @@ def answer(query, min_similarity=0.7):
 
     if not ranked_documents:
         print("No similar books found")
-    
+
     return ranked_documents
 
 # Example usage