Update app.py

app.py

@@ -127,6 +127,12 @@ def generate_answer(message, choice):
     addresses = extract_addresses(response['output'])
     return response['output'], addresses
 
+
+def extract_location_from_response(response):
+    agent = initialize_agent_with_prompt(ADDRESS_PROMPT)
+    address_response = agent(response)
+    return address_response['output']
+
 # def bot(history, choice):
 #     if not history:
 #         return history