Update app.py

app.py

@@ -43,11 +43,19 @@ def pdf_changes(pdf_doc):
     )
     return "Ready"
 
+def format(query, history):
+    fhistory = []
+    for user, bot in history:
+        fhistory.append(f"User: {user}")
+        fhistory.append(f"Assistant: {bot}")
+    return fhistory
+
 def query(history, text):
-    result = qa({'question': text, 'chat_history': history})
-    history.append((text, result['answer']))
+    fhistory = format(text,history)
+    result = qa({'question': text, 'chat_history': fhistory})
+    new_history = history + [(text,result['answer'])]
     
-    return history,""
+    return new_history,""
 
 css="""
 #col-container {max-width: 700px; margin-left: auto; margin-right: auto;}