Update app.py

app.py

@@ -42,23 +42,19 @@ 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 by searching for key phrases
-        relevant_pages = []
+        # Identify key sentences or phrases
+        key_phrases = answer.split(". ")  # Split answer into sentences for more precise matching
+        
+        relevant_pages = set()
         for i, page in enumerate(pages):
-            if any(phrase.lower() in page.page_content.lower() for phrase in answer.split()):
-                relevant_pages.append(f"Page {i+1}")
+            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
 
         if relevant_pages:
-            source_str = f" (Source: {', '.join(relevant_pages)})"
+            source_str = f" (Source: {', '.join(f'Page {p}' for p in sorted(relevant_pages))})"
         else:
-            # 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)"
+            source_str = " (Source: Not found in specific page)"
 
         # Create a clickable link for the document
         file_name = os.path.basename(file_path)
@@ -73,6 +69,9 @@ input_question = gr.Textbox(label="Ask about the document")
 output_text = gr.Textbox(label="Answer - GeminiPro")
 
 async def pdf_qa(file, question):
+    if file is None:
+        return "Error: No file uploaded. Please upload a PDF document."
+    
     answer = await initialize(file.name, question)
     return answer