Update app.py

app.py

@@ -52,21 +52,23 @@ async def initialize(file_path, question):
                     relevant_pages.add(i+1)  # Add page number if phrase is found
 
         if relevant_pages:
-            source_str = f" (Source: {', '.join(f'Page {p}' for p in sorted(relevant_pages))})"
+            page_numbers = ', '.join(str(p) for p in sorted(relevant_pages))
+            source_str = f"Relevant pages: {page_numbers}"
         else:
-            source_str = " (Source: Not found in specific page)"
+            source_str = "Relevant pages: Not found in specific page"
 
         # Create a clickable link for the document
         file_name = os.path.basename(file_path)
         source_link = f"[{file_name}](file://{os.path.abspath(file_path)})"
-        return f"{answer} {source_str} - [Document: {source_link}]"
+        
+        return f"Answer: {answer}\n{source_str}\n[Document: {source_link}]"
     else:
         return "Error: Unable to process the document. Please ensure the PDF file is valid."
 
 # Define Gradio Interface
 input_file = gr.File(label="Upload PDF File")
 input_question = gr.Textbox(label="Ask about the document")
-output_text = gr.Textbox(label="Answer - GeminiPro")
+output_text = gr.Textbox(label="Answer and Relevant Pages")
 
 async def pdf_qa(file, question):
     if file is None: