Update app.py

app.py

@@ -1233,7 +1233,7 @@ def neo4j_retrieval_conversational(question: str):
     structured_data = structured_retriever(question)
     logging.debug(f"Structured data (Conversational): {structured_data}")
     prompt = QA_CHAIN_PROMPT_NEO4J_CONVERSATIONAL.format(context=structured_data, question=question)
-    response = chat_model(prompt)
+    response = chat_model(prompt.to_messages())
     return response, []
 
 # Neo4j Retrieval chain for detailed mode
@@ -1241,7 +1241,7 @@ def neo4j_retrieval_details(question: str):
     structured_data = structured_retriever(question)
     logging.debug(f"Structured data (Details): {structured_data}")
     prompt = QA_CHAIN_PROMPT_NEO4J_DETAILS.format(context=structured_data, question=question)
-    response = chat_model(prompt)
+    response = chat_model(prompt.to_messages())
     return response, extract_addresses(response)
 
 # Update the generate_answer function to include Neo4j retrieval modes
@@ -1270,24 +1270,26 @@ def generate_answer(message, choice, retrieval_mode):
     else:
         return "Invalid retrieval mode selected.", []
 
-
+# Rest of the code remains the same
 
 def bot(history, choice, tts_choice, retrieval_mode):
     if not history:
         return history
     
     response, addresses = generate_answer(history[-1][0], choice, retrieval_mode)
+    response_text = response if isinstance(response, str) else response[0]
+
     history[-1][1] = ""
 
     with concurrent.futures.ThreadPoolExecutor() as executor:
         if tts_choice == "Alpha":
-            audio_future = executor.submit(generate_audio_elevenlabs, response)
+            audio_future = executor.submit(generate_audio_elevenlabs, response_text)
         elif tts_choice == "Beta":
-            audio_future = executor.submit(generate_audio_parler_tts, response)
+            audio_future = executor.submit(generate_audio_parler_tts, response_text)
         elif tts_choice == "Gamma":
-            audio_future = executor.submit(generate_audio_mars5, response)
+            audio_future = executor.submit(generate_audio_mars5, response_text)
 
-        for character in response:
+        for character in response_text:
             history[-1][1] += character
             time.sleep(0.05)
             yield history, None
@@ -1295,7 +1297,7 @@ def bot(history, choice, tts_choice, retrieval_mode):
         audio_path = audio_future.result()
         yield history, audio_path
 
-    history.append([response, None])  # Ensure the response is added in the correct format
+    history.append([response_text, None])  # Ensure the response is added in the correct format
 
 def add_message(history, message):
     history.append((message, None))
@@ -1351,7 +1353,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>
@@ -1469,7 +1471,7 @@ def show_map_if_details(history, choice):
     if choice in ["Details", "Conversational"]:
         return gr.update(visible=True), update_map_with_response(history)
     else:
-        return gr.update(visible=False), ""
+        return gr.update(visible(False), "")
 
 def generate_audio_elevenlabs(text):
     XI_API_KEY = os.environ['ELEVENLABS_API']
@@ -1644,7 +1646,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>