Update app.py

app.py

@@ -443,8 +443,6 @@ def tts_response(history, response, tts_choice):
             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)
 
         # Initialize the response string if it's None
         if history[-1][1] is None:
@@ -459,8 +457,12 @@ def tts_response(history, response, tts_choice):
         # Get the path of the generated audio
         audio_path = audio_future.result()
 
-        # Return the final history and audio path
-        yield history, audio_path
+        # Check if the audio path is valid before yielding it
+        if audio_path and os.path.exists(audio_path):
+            yield history, audio_path
+        else:
+            logging.error("Audio file not generated correctly")
+            yield history, None
 
 
 
@@ -710,10 +712,16 @@ def generate_answer(message, choice, retrieval_mode, selected_model):
 
 
 
+# def add_message(history, message):
+#     history.append((message, None))
+#     return history, gr.Textbox(value="", interactive=True, show_label=False)
+
 def add_message(history, message):
-    history.append((message, None))
+    # Append a tuple with the user's message and a placeholder for the assistant's response
+    history.append((message, ""))  # Use an empty string to ensure the format is correct
     return history, gr.Textbox(value="", interactive=True, show_label=False)
 
+
 def print_like_dislike(x: gr.LikeData):
     print(x.index, x.value, x.liked)