Update app.py

app.py

@@ -310,6 +310,34 @@ def generate_answer(message, choice, retrieval_mode):
 
 
 
+# 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] = ""
+
+#     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
+
+#     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
@@ -319,6 +347,12 @@ def bot(history, choice, tts_choice, retrieval_mode):
 
     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)
@@ -326,17 +360,11 @@ 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