Update app.py

app.py

@@ -66,14 +66,19 @@ def process_text(input_text):
 
 # === Core RAG Handler ===
 def handle_input(file, clipboard, query):
-    raw_text = extract_text(file, clipboard)
-    if not raw_text:
-        return "Please provide either a PDF or clipboard text."
-    
-    qa = process_text(raw_text)
-    response = qa.invoke({"query": query if query else "Summarize the key points and risks in this policy."})
-    result = response["result"]
-    return result
+    try:
+        raw_text = extract_text(file, clipboard)
+        if not raw_text:
+            return "Please provide either a PDF or clipboard text."
+        
+        qa = process_text(raw_text)
+        response = qa.invoke({"query": query if query else "Summarize the key points and risks in this policy."})
+        
+        result = response["result"]
+        sources = response["source_documents"]
+        source_preview = "\n\n📄 Sources:\n" + "\n---\n".join([doc.page_content[:300] + "..." for doc in sources[:3]])
+        
+        return result + source_preview
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
@@ -87,7 +92,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     query_input = gr.Textbox(label="Ask a Question (optional)", placeholder="e.g., What risks am I agreeing to?")
     submit_btn = gr.Button("🔍 Analyze")
-    output = gr.Textbox(label="Output", lines=15)
+    output = gr.Textbox(label="Output", lines=20)
 
     submit_btn.click(fn=handle_input, inputs=[file_input, clipboard_input, query_input], outputs=output)