Update app.py

app.py

@@ -341,13 +341,21 @@ def generate_answer(message, choice):
     return response['output'], addresses
 
 
+# def extract_location_from_response(response):
+#     agent = initialize_agent_with_prompt(QA_CHAIN_PROMPT_1)
+#     address_response = agent(response)
+#     return address_response['output']
+
 def extract_location_from_response(response):
-    agent = initialize_agent_with_prompt(QA_CHAIN_PROMPT_1)
+    address_prompt = PromptTemplate(
+        input_variables=["context", "question"],
+        template="Extract the address from the following response:\n{context}\nResponse: {question}"
+    )
+    agent = initialize_agent_with_prompt(address_prompt)
     address_response = agent(response)
     return address_response['output']
 
 
-
 def bot(history, choice):
     if not history:
         return history