ykl7 commited on
Commit
e378954
·
1 Parent(s): 4abecb0

improve retrieved document parsing

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -150,14 +150,17 @@ def retriever(query: str, selected_retriever: str):
150
  try:
151
  idx = document.lower().index("abstract:")
152
  title = document[:idx].replace('\n', ' ')
153
- retrieved_document_titles = retrieved_document_titles + title + '\n'
154
  except:
155
- retrieved_document_titles = retrieved_document_titles + document + '\n'
156
 
157
- display_retrieved_documents = f"\nRetrieved Documents: \n{retrieved_document_titles}"
158
 
159
  for chunk in display_retrieved_documents.split():
160
- text += chunk + " "
 
 
 
161
  time.sleep(0.05)
162
  # Add a blinking cursor to simulate typing
163
  placeholder.markdown(text + "▌")
 
150
  try:
151
  idx = document.lower().index("abstract:")
152
  title = document[:idx].replace('\n', ' ')
153
+ retrieved_document_titles = retrieved_document_titles + title + ' ||| '
154
  except:
155
+ retrieved_document_titles = retrieved_document_titles + document + ' ||| '
156
 
157
+ display_retrieved_documents = f" ||| Retrieved Documents: \n{retrieved_document_titles}"
158
 
159
  for chunk in display_retrieved_documents.split():
160
+ if chunk == "|||":
161
+ text += chunk + "\n"
162
+ else:
163
+ text += chunk + " "
164
  time.sleep(0.05)
165
  # Add a blinking cursor to simulate typing
166
  placeholder.markdown(text + "▌")