Update app.py

app.py

@@ -34,6 +34,8 @@ os.makedirs(PERSIST_DIR, exist_ok=True)
 
 # Function to save chat history to cookies
 def save_chat_history_to_cookies(chat_id, query, response, cookies):
+    if cookies is None:
+        cookies = {}
     history = cookies.get('chat_history', '[]')
     history_list = eval(history)
     history_list.append({
@@ -44,7 +46,7 @@ def save_chat_history_to_cookies(chat_id, query, response, cookies):
     })
     cookies['chat_history'] = str(history_list)
 
-def handle_query(query, cookies):
+def handle_query(query, cookies=None):
     chat_text_qa_msgs = [
         (
             "user",
@@ -64,9 +66,12 @@ def handle_query(query, cookies):
 
     # Use chat history to enhance response
     context_str = ""
-    for past_query, response in reversed(cookies.get('chat_history', [])):
-        if past_query.strip():
-            context_str += f"User asked: '{past_query}'\nBot answered: '{response}'\n"
+    if cookies:
+        history = cookies.get('chat_history', '[]')
+        history_list = eval(history)
+        for entry in reversed(history_list):
+            if entry["query"].strip():
+                context_str += f"User asked: '{entry['query']}'\nBot answered: '{entry['response']}'\n"
 
     query_engine = index.as_query_engine(text_qa_template=text_qa_template, context_str=context_str)
     answer = query_engine.query(query)
@@ -85,7 +90,8 @@ def handle_query(query, cookies):
     return response
 
 # Define your Gradio chat interface function
-def chat_interface(message, history, cookies):
+def chat_interface(message, history):
+    cookies = {}  # You might need to get cookies from the request in a real implementation
     try:
         # Process the user message and generate a response
         response = handle_query(message, cookies)