Spaces:
Sleeping
Sleeping
Upload query_index.py
Browse files- query_index.py +15 -2
query_index.py
CHANGED
|
@@ -128,12 +128,25 @@ class MultimodalRAGSystem:
|
|
| 128 |
else:
|
| 129 |
# Text node
|
| 130 |
file_name = node.metadata.get("file_name", "N/A")
|
| 131 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 132 |
|
| 133 |
source_texts.append({
|
| 134 |
"text": node.text[:200] + "...",
|
| 135 |
"score": node_score.score,
|
| 136 |
-
"page":
|
| 137 |
"file": file_name,
|
| 138 |
"link": web_link
|
| 139 |
})
|
|
|
|
| 128 |
else:
|
| 129 |
# Text node
|
| 130 |
file_name = node.metadata.get("file_name", "N/A")
|
| 131 |
+
page_num = node.metadata.get("page_number", "N/A")
|
| 132 |
+
|
| 133 |
+
web_link = None
|
| 134 |
+
if file_name != "N/A":
|
| 135 |
+
# URL encode the filename to handle spaces and special chars safely
|
| 136 |
+
from urllib.parse import quote
|
| 137 |
+
safe_filename = quote(file_name)
|
| 138 |
+
web_link = f"/documents/{safe_filename}"
|
| 139 |
+
|
| 140 |
+
if page_num != "N/A":
|
| 141 |
+
web_link += f"#page={page_num}"
|
| 142 |
+
|
| 143 |
+
# DEBUG: Print link construction details
|
| 144 |
+
logger.info(f"DEBUG: File: {file_name}, Page: {page_num}, Link: {web_link}")
|
| 145 |
|
| 146 |
source_texts.append({
|
| 147 |
"text": node.text[:200] + "...",
|
| 148 |
"score": node_score.score,
|
| 149 |
+
"page": page_num,
|
| 150 |
"file": file_name,
|
| 151 |
"link": web_link
|
| 152 |
})
|