avakanski commited on
Commit
cb9efc0
·
verified ·
1 Parent(s): 3d02e3f

Upload query_index.py

Browse files
Files changed (1) hide show
  1. 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
- web_link = f"/documents/{file_name}" if file_name != "N/A" else None
 
 
 
 
 
 
 
 
 
 
 
 
 
132
 
133
  source_texts.append({
134
  "text": node.text[:200] + "...",
135
  "score": node_score.score,
136
- "page": node.metadata.get("page_number", "N/A"),
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
  })