Update app.py

app.py

@@ -13,7 +13,7 @@ async def initialize(file_path, question):
     model = genai.GenerativeModel('gemini-pro')
     model = ChatGoogleGenerativeAI(model="gemini-pro", temperature=0.3)
     
-    # Prompt template for precise answers
+    # Refined prompt template to encourage precise and concise answers
     prompt_template = """Answer the question precisely and concisely using the provided context. Avoid any additional commentary or system messages.
                           If the answer is not contained in the context, respond with "answer not available in context".
                           
@@ -42,16 +42,23 @@ async def initialize(file_path, question):
         stuff_answer = await stuff_chain.ainvoke({"input_documents": pages, "question": question, "context": context})
         answer = stuff_answer.get('output_text', '').strip()
         
-        # Identify the pages that contain the answer
+        # Identify the pages that contain the answer by searching for key phrases
         relevant_pages = []
         for i, page in enumerate(pages):
-            if answer.lower() in page.page_content.lower():
+            if any(phrase.lower() in page.page_content.lower() for phrase in answer.split()):
                 relevant_pages.append(f"Page {i+1}")
 
         if relevant_pages:
             source_str = f" (Source: {', '.join(relevant_pages)})"
         else:
-            source_str = " (Source: Not found in specific page)"
+            # Additional attempt: Find any overlapping content to improve detection
+            for i, page in enumerate(pages):
+                if any(phrase in page.page_content for phrase in answer.split()[:10]):  # Checking with a subset of the answer
+                    relevant_pages.append(f"Page {i+1}")
+            if relevant_pages:
+                source_str = f" (Source: {', '.join(relevant_pages)})"
+            else:
+                source_str = " (Source: Not found in specific page)"
 
         # Create a clickable link for the document
         file_name = os.path.basename(file_path)