Update app.py

app.py

@@ -274,9 +274,27 @@ def initialize_agent_with_prompt(prompt_template):
     return agent
 
 
+# def generate_answer(message, choice):
+#     logging.debug(f"generate_answer called with prompt_choice: {choice}")
+    
+#     if choice == "Details":
+#         agent = initialize_agent_with_prompt(QA_CHAIN_PROMPT_1)
+#     elif choice == "Conversational":
+#         agent = initialize_agent_with_prompt(QA_CHAIN_PROMPT_2)
+#     else:
+#         logging.error(f"Invalid prompt_choice: {choice}. Defaulting to 'Conversational'")
+#         agent = initialize_agent_with_prompt(QA_CHAIN_PROMPT_2)
+#     response = agent(message)
+
+#     addresses = extract_addresses(response['output'])
+#     return response['output'], addresses
+
 def generate_answer(message, choice):
     logging.debug(f"generate_answer called with prompt_choice: {choice}")
     
+    # Reset conversational memory
+    conversational_memory.clear()
+    
     if choice == "Details":
         agent = initialize_agent_with_prompt(QA_CHAIN_PROMPT_1)
     elif choice == "Conversational":