Update app.py

app.py

@@ -36,7 +36,7 @@ def data_ingestion_from_directory():
     index = VectorStoreIndex.from_documents(documents)
     index.storage_context.persist(persist_dir=PERSIST_DIR)
 
-def handle_query(query):
+def handle_query(query, chat_history):
     chat_text_qa_msgs = [
         (
             "user",
@@ -55,7 +55,13 @@ def handle_query(query):
     storage_context = StorageContext.from_defaults(persist_dir=PERSIST_DIR)
     index = load_index_from_storage(storage_context)
 
-    query_engine = index.as_query_engine(text_qa_template=text_qa_template)
+    # Use chat history to enhance response
+    context_str = ""
+    for past_query, response in reversed(chat_history):
+        if past_query.strip():
+            context_str += f"User asked: '{past_query}'\nBot answered: '{response}'\n"
+
+    query_engine = index.as_query_engine(text_qa_template=text_qa_template, context_str=context_str)
     answer = query_engine.query(query)
 
     if hasattr(answer, 'response'):
@@ -84,7 +90,7 @@ output_component = gr.Textbox()
 
 # Function to add chat history to output
 def chat_with_history(query):
-    response = handle_query(query)
+    response = handle_query(query, chat_history)
     chat_history.append((query, response))
     return response