Update app.py

app.py

@@ -53,16 +53,22 @@ async def initialize(file_path, question):
 
         if relevant_pages:
             page_numbers = ', '.join(str(p) for p in sorted(relevant_pages))
-            return f"Relevant pages: {page_numbers}"
+            source_str = f"Relevant pages: {page_numbers}"
         else:
-            return "No relevant pages found in the document."
+            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: {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="Relevant Pages")
+output_text = gr.Textbox(label="Answer and Relevant Pages")
 
 async def pdf_qa(file, question):
     if file is None: