Update app.py

app.py

@@ -365,44 +365,22 @@ 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, 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] = response[1]  # Update bot response
-
+    history[-1][1] = ""
+    
     with concurrent.futures.ThreadPoolExecutor() as executor:
         if tts_choice == "Alpha":
-            audio_future = executor.submit(generate_audio_elevenlabs, response[1])
+            audio_future = executor.submit(generate_audio_elevenlabs, response)
         elif tts_choice == "Beta":
-            audio_future = executor.submit(generate_audio_parler_tts, response[1])
+            audio_future = executor.submit(generate_audio_parler_tts, response)
         elif tts_choice == "Gamma":
-            audio_future = executor.submit(generate_audio_mars5, response[1])
+            audio_future = executor.submit(generate_audio_mars5, response)
+        
         
-        for character in response[1]:
+        for character in response:
             history[-1][1] += character
             time.sleep(0.05)
             yield history, None
@@ -410,6 +388,8 @@ def bot(history, choice, tts_choice):
         audio_path = audio_future.result()
         yield history, audio_path
 
+
+
 # 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)