Update app.py

app.py

@@ -338,21 +338,41 @@ def generate_answer(message, choice, retrieval_mode):
 #     history.append([response, None])  # Ensure the response is added in the correct format
 
 
+# def bot(history, choice, tts_choice, retrieval_mode):
+#     if not history:
+#         return history
+    
+#     response, addresses = generate_answer(history[-1][0], choice, retrieval_mode)
+#     response = f"<div style='font-size: 14px;'>{response}</div>"  # Adjust the font size
+
+#     history[-1][1] = ""
+
+#     # Immediately update the chatbot UI with the text response
+#     for character in response:
+#         history[-1][1] += character
+#         time.sleep(0.05)
+#         yield history, None
+
+#     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)
+        
+#         audio_path = audio_future.result()
+
+#     history.append([response, None])  # Ensure the response is added in the correct format
+#     yield history, audio_path
+
 def bot(history, choice, tts_choice, retrieval_mode):
     if not history:
         return history
     
     response, addresses = generate_answer(history[-1][0], choice, retrieval_mode)
-    response = f"<div style='font-size: 14px;'>{response}</div>"  # Adjust the font size
-
     history[-1][1] = ""
 
-    # Immediately update the chatbot UI with the text response
-    for character in response:
-        history[-1][1] += character
-        time.sleep(0.05)
-        yield history, None
-
     with concurrent.futures.ThreadPoolExecutor() as executor:
         if tts_choice == "Alpha":
             audio_future = executor.submit(generate_audio_elevenlabs, response)
@@ -360,11 +380,16 @@ def bot(history, choice, tts_choice, retrieval_mode):
             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
 
     history.append([response, None])  # Ensure the response is added in the correct format
-    yield history, audio_path