Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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":
|
| 76 |
-
|
| 77 |
result = response["result"]
|
| 78 |
sources = response["source_documents"]
|
| 79 |
-
source_preview = "\n\n📄 Sources:\n" + "\n---\n".join(
|
| 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 |
|