Update app.py

app.py

@@ -45,17 +45,22 @@ async def initialize(file_path, question):
         # Identify key sentences or phrases
         key_phrases = answer.split(". ")  # Split answer into sentences for more precise matching
         
-        relevant_pages = set()
+        # Score each page based on the presence of key phrases
+        page_scores = [0] * len(pages)
         for i, page in enumerate(pages):
             for phrase in key_phrases:
                 if phrase.lower() in page.page_content.lower():
-                    relevant_pages.add(i+1)  # Add page number if phrase is found
-
+                    page_scores[i] += 1
+        
+        # Determine the page with the highest score
+        max_score = max(page_scores)
+        relevant_pages = [i+1 for i, score in enumerate(page_scores) if score == max_score]
+        
         if relevant_pages:
             page_numbers = ', '.join(str(p) for p in sorted(relevant_pages))
-            source_str = f"Relevant pages: {page_numbers}"
+            source_str = f"Relevant page(s): {page_numbers}"
         else:
-            source_str = "Relevant pages: Not found in specific page"
+            source_str = "Relevant page(s): Not found in specific page"
 
         # Create a clickable link for the document
         file_name = os.path.basename(file_path)