Update app.py

app.py

@@ -335,6 +335,21 @@ def initialize_agent_with_prompt(prompt_template):
     )
     return agent
 
+# def generate_answer(message, choice):
+#     logging.debug(f"generate_answer called with prompt_choice: {choice}")
+    
+#     if choice == "Details":
+#         agent = initialize_agent_with_prompt(QA_CHAIN_PROMPT_1)
+#     elif choice == "Conversational":
+#         agent = initialize_agent_with_prompt(QA_CHAIN_PROMPT_2)
+#     else:
+#         logging.error(f"Invalid prompt_choice: {choice}. Defaulting to 'Conversational'")
+#         agent = initialize_agent_with_prompt(QA_CHAIN_PROMPT_2)
+#     response = agent(message)
+
+#     addresses = extract_addresses(response['output'])
+#     return response['output'], addresses
+
 def generate_answer(message, choice):
     logging.debug(f"generate_answer called with prompt_choice: {choice}")
     
@@ -348,24 +363,46 @@ def generate_answer(message, choice):
     response = agent(message)
 
     addresses = extract_addresses(response['output'])
-    return response['output'], addresses
+    return [message, response['output']], addresses  # Ensure it returns a list with two elements
+
+# def bot(history, choice, tts_choice):
+#     if not history:
+#         return history
+#     response, addresses = generate_answer(history[-1][0], choice)
+#     history[-1][1] = ""
+    
+#     with concurrent.futures.ThreadPoolExecutor() as executor:
+#         if tts_choice == "Alpha":
+#             audio_future = executor.submit(generate_audio_elevenlabs, response)
+#         elif tts_choice == "Beta":
+#             audio_future = executor.submit(generate_audio_parler_tts, response)
+#         elif tts_choice == "Gamma":
+#             audio_future = executor.submit(generate_audio_mars5, response)
+        
+        
+#         for character in response:
+#             history[-1][1] += character
+#             time.sleep(0.05)
+#             yield history, None
+        
+#         audio_path = audio_future.result()
+#         yield history, audio_path
 
 def bot(history, choice, tts_choice):
     if not history:
         return history
     response, addresses = generate_answer(history[-1][0], choice)
-    history[-1][1] = ""
-    
+    history[-1][1] = response[1]  # Update bot response
+
     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[1])
         elif tts_choice == "Beta":
-            audio_future = executor.submit(generate_audio_parler_tts, response)
+            audio_future = executor.submit(generate_audio_parler_tts, response[1])
         elif tts_choice == "Gamma":
-            audio_future = executor.submit(generate_audio_mars5, response)
-        
+            audio_future = executor.submit(generate_audio_mars5, response[1])
         
-        for character in response:
+        for character in response[1]:
             history[-1][1] += character
             time.sleep(0.05)
             yield history, None