dh-mc commited on
Commit
8c37fda
1 Parent(s): 4da2dc4

refine UI for sources

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -117,7 +117,8 @@ def qa(chatbot):
117
  titles = []
118
  for doc in ret["source_documents"]:
119
  url = f"{doc.metadata['url']}#page={doc.metadata['page'] + 1}"
120
- title = url.split("/")[-1]
 
121
  if title not in titles:
122
  titles.append(title)
123
  chatbot[-1][1] += f"1. [{title}]({url})\n"
 
117
  titles = []
118
  for doc in ret["source_documents"]:
119
  url = f"{doc.metadata['url']}#page={doc.metadata['page'] + 1}"
120
+ file_name = doc.metadata["source"].split("/")[-1]
121
+ title = f"{file_name} Page: {doc.metadata['page']}"
122
  if title not in titles:
123
  titles.append(title)
124
  chatbot[-1][1] += f"1. [{title}]({url})\n"