Update app.py

app.py

@@ -365,22 +365,43 @@ def generate_answer(message, choice):
     addresses = extract_addresses(response['output'])
     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_pair, addresses = generate_answer(history[-1][0], choice)
+#     history[-1] = response_pair  # Update bot response correctly
+
+#     with concurrent.futures.ThreadPoolExecutor() as executor:
+#         if tts_choice == "Alpha":
+#             audio_future = executor.submit(generate_audio_elevenlabs, response_pair[1])
+#         elif tts_choice == "Beta":
+#             audio_future = executor.submit(generate_audio_parler_tts, response_pair[1])
+#         elif tts_choice == "Gamma":
+#             audio_future = executor.submit(generate_audio_mars5, response_pair[1])
+        
+#         for character in response_pair[1]:
+#             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] = ""
-    
+    response_pair, addresses = generate_answer(history[-1][0], choice)
+    history[-1] = response_pair  # Update bot response correctly
+
     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_pair[1])
         elif tts_choice == "Beta":
-            audio_future = executor.submit(generate_audio_parler_tts, response)
+            audio_future = executor.submit(generate_audio_parler_tts, response_pair[1])
         elif tts_choice == "Gamma":
-            audio_future = executor.submit(generate_audio_mars5, response)
+            audio_future = executor.submit(generate_audio_mars5, response_pair[1])
         
-        
-        for character in response:
+        for character in response_pair[1]:
             history[-1][1] += character
             time.sleep(0.05)
             yield history, None
@@ -389,15 +410,18 @@ def bot(history, choice, tts_choice):
         yield history, audio_path
 
 
-
+def add_message(history, message):
+    history = history or []
+    history.append([message, ""])  # Ensure it is a list with two elements: message and empty response
+    return history, gr.Textbox(value="", interactive=True, placeholder="Enter message or upload file...", show_label=False)
 # def add_message(history, message):
 #     history.append((message, None))
 #     return history, gr.Textbox(value="", interactive=True, placeholder="Enter message or upload file...", show_label=False)
 
-def add_message(history, message):
-    history = history or []
-    history.append((message, None))
-    return history, gr.Textbox(value="", interactive=True, placeholder="Enter message or upload file...", show_label=False)
+# def add_message(history, message):
+#     history = history or []
+#     history.append((message, None))
+#     return history, gr.Textbox(value="", interactive=True, placeholder="Enter message or upload file...", show_label=False)
 
 # def generate_voice_response(history, tts_choice):
 #     if not history: