Update utils.py

utils.py

@@ -31,6 +31,7 @@ def get_answer(query):
     # Prepare input and decoder input
     similar_docs = get_similar_docs(query, 10, documents, embeddings)
     chat_history.append(Document(page_content=query))
+    
     combined_docs = chat_history + similar_docs
 
     input_text = ""
@@ -46,5 +47,5 @@ def get_answer(query):
 
     # Decode the generated text
     generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    chat_history.append(Document(page_content=generated_text))
+    chat_history.append(Document(page_content=generated_text.split(query)[1].capitalize()))
     return generated_text.split(query)[1].capitalize()