Update app.py

app.py

@@ -103,6 +103,11 @@ def generate_answer(message, choice):
         logging.error(f"Invalid prompt_choice: {choice}. Defaulting to 'Conversational'")
         agent = initialize_agent_with_prompt(QA_CHAIN_PROMPT_2)
     response = agent(message)
+
+    # Extract addresses for mapping regardless of the choice
+    addresses = extract_addresses(response['output'])
+    if addresses:
+        generate_map(addresses)
     return response['output']
 
 def bot(history, choice):