Update app.py

app.py

@@ -161,10 +161,10 @@ def main(query: str, client: QdrantClient, collection_name: str, llm, dense_mode
       ], stop=["</s>"], temperature=0.7, max_tokens=3000)
     
     answer = output['choices'][0]['message']['content']
-    final_response = f"{answer}\n\n\n\nSources :\n\n{result_metadatas}"
+    answer_with_metadatas = f"{answer}\n\n\n\nSources :\n\n{result_metadatas}"
     
     print(f'OUTPUT: {output}')
-    return final_response
+    return answer, answer_with_metadatas
 
 @st.cache_resource
 def load_models_and_documents():
@@ -366,15 +366,15 @@ if __name__ == '__main__':
         st.chat_message("user").markdown(prompt)
         st.session_state.messages.append({"role": "user", "content": prompt})
 
-        ai_response = main(prompt, client, collection_name, llm, dense_model, sparse_model)
+        ai_response, ai_response_with_metadatas = main(prompt, client, collection_name, llm, dense_model, sparse_model)
         with st.chat_message("assistant"):
             message_placeholder = st.empty()
             full_response = ""
-            for chunk in re.split(r'(\s+)', ai_response):
+            for chunk in re.split(r'(\s+)', ai_response_with_metadatas):
                 full_response += chunk + " "
                 time.sleep(0.01)
                 message_placeholder.markdown(full_response + "▌")
-        st.session_state.messages.append({"role": "assistant", "content": full_response})
+        st.session_state.messages.append({"role": "assistant", "content": ai_response})
 
     st.sidebar.title("Upload your file")
     uploaded_files = st.sidebar.file_uploader("Choose a file", accept_multiple_files=True, type=['docx', 'doc', 'odt', 'pptx', 'ppt', 'xlsx', 'csv', 'tsv', 'eml', 'msg', 'rtf', 'epub', 'html', 'xml', 'pdf', 'png', 'jpg', 'heic','txt'])