Update app.py

app.py

@@ -73,6 +73,9 @@ def predict(message, history):
     #For now, we only query the vector database once, at the start.
     if len(history) == 0:
         source_text = vector_search(message)
+    else:
+        source_text = re.sub("^.+?\n\n### Source ###\n", "", history[0][0])
+        print(source_text)
     
     history_transformer_format = history + [[message, ""]]