ykl7 commited on
Commit
5fa5d29
·
1 Parent(s): 0d66203

improve display of retrieved documents

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -144,7 +144,12 @@ def retriever(query: str, selected_retriever: str):
144
 
145
  corpus = '\n\n'.join(topk_documents)
146
 
147
- display_retrieved_documents = f"Retrieved Documents: \n{corpus}"
 
 
 
 
 
148
 
149
  for chunk in display_retrieved_documents.split():
150
  text += chunk + " "
 
144
 
145
  corpus = '\n\n'.join(topk_documents)
146
 
147
+ retrieved_document_titles = ""
148
+ for document in topk_documents:
149
+ title = document[:document.lower().index("abstract: ")].replace('\n', ' ')
150
+ retrieved_document_titles = f"{retrieved_document_titles}\n"
151
+
152
+ display_retrieved_documents = f"\nRetrieved Documents: \n{retrieved_document_titles}"
153
 
154
  for chunk in display_retrieved_documents.split():
155
  text += chunk + " "