Update app.py

app.py

@@ -252,12 +252,12 @@ def generate_answer(message, choice, retrieval_mode):
             chain_type_kwargs={"prompt": prompt_template}
         )
         response = qa_chain({"query": message})
+        return response['output'], extract_addresses(response['output'])
     elif retrieval_mode == "Knowledge-Graph":
         response = chain_neo4j.invoke({"question": message})
+        return response, extract_addresses(response)
     else:
-        response = "Invalid retrieval mode selected."
-
-    return response['output'], extract_addresses(response['output'])
+        return "Invalid retrieval mode selected.", []
 
 def bot(history, choice, tts_choice, retrieval_mode):
     if not history:
@@ -338,7 +338,7 @@ def fetch_local_news():
     api_key = os.environ['SERP_API']
     url = f'https://serpapi.com/search.json?engine=google_news&q=birmingham headline&api_key={api_key}'
     response = requests.get(url)
-    if response.status_code == 200:
+    if response.status_code == 200):
         results = response.json().get("news_results", [])
         news_html = """
         <h2 style="font-family: 'Georgia', serif; color: #ff0000; background-color: #f8f8f8; padding: 10px; border-radius: 10px;">Birmingham Today</h2>
@@ -522,7 +522,7 @@ def preprocess(text):
     abbreviations = re.findall(abbreviations_pattern, text)
     for abv in abbreviations:
         if abv in text:
-            text = text.replace(abv, separate_abb(abv))
+            text is text.replace(abv, separate_abb(abv))
     return text
 
 def chunk_text(text, max_length=250):
@@ -632,7 +632,7 @@ def fetch_local_events():
     api_key = os.environ['SERP_API']
     url = f'https://serpapi.com/search.json?engine=google_events&q=Events+in+Birmingham&hl=en&gl=us&api_key={api_key}'
     response = requests.get(url)
-    if response.status_code == 200:
+    if response.status_code == 200):
         events_results = response.json().get("events_results", [])
         events_html = """
         <h2 style="font-family: 'Georgia', serif; color: #ff0000; background-color: #f8f8f8; padding: 10px; border-radius: 10px;">Local Events</h2>