Harishkhawaja commited on
Commit
15346ff
·
verified ·
1 Parent(s): bdc2edb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -5
app.py CHANGED
@@ -70,15 +70,25 @@ def handle_input(file, clipboard, query):
70
  raw_text = extract_text(file, clipboard)
71
  if not raw_text:
72
  return "Please provide either a PDF or clipboard text."
73
-
 
 
 
 
 
 
 
74
  qa = process_text(raw_text)
75
- response = qa.invoke({"query": query if query else "Summarize the key points and risks in this policy."})
76
-
77
  result = response["result"]
78
  sources = response["source_documents"]
79
- source_preview = "\n\n📄 Sources:\n" + "\n---\n".join([doc.page_content[:300] + "..." for doc in sources[:3]])
80
-
 
 
81
  return result + source_preview
 
82
  except Exception as e:
83
  return f"❌ Error: {str(e)}"
84
 
 
70
  raw_text = extract_text(file, clipboard)
71
  if not raw_text:
72
  return "Please provide either a PDF or clipboard text."
73
+
74
+ # Your prompt for explanation + risks
75
+ default_query = (
76
+ "Explain this policy in simple terms and highlight the risks for the user. "
77
+ "Provide bullet points for risks."
78
+ )
79
+ user_query = query if query else default_query
80
+
81
  qa = process_text(raw_text)
82
+ response = qa.invoke({"query": user_query})
83
+
84
  result = response["result"]
85
  sources = response["source_documents"]
86
+ source_preview = "\n\n📄 Sources:\n" + "\n---\n".join(
87
+ [doc.page_content[:300] + "..." for doc in sources[:3]]
88
+ )
89
+
90
  return result + source_preview
91
+
92
  except Exception as e:
93
  return f"❌ Error: {str(e)}"
94