Update app.py

app.py

@@ -289,7 +289,30 @@ def generate_answer(message, choice):
     addresses = extract_addresses(response['output'])
     return response['output'], addresses
 
-def bot(history, choice, tts_choice):
+# 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, state):
     if not history:
         return history
     response, addresses = generate_answer(history[-1][0], choice)
@@ -303,7 +326,6 @@ def bot(history, choice, tts_choice):
         elif tts_choice == "Gamma":
             audio_future = executor.submit(generate_audio_mars5, response)
         
-        
         for character in response:
             history[-1][1] += character
             time.sleep(0.05)
@@ -312,8 +334,6 @@ 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)
@@ -704,8 +724,11 @@ with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
             tts_choice = gr.Radio(label="Select TTS System", choices=["Alpha", "Beta", "Gamma"], value="Alpha")
             retriver_button = gr.Button("Retriver")
             # retriver_button.click(fn=add_message, inputs=[chatbot, chat_input], outputs=[chatbot, chat_input]).then(fn=bot, inputs=[chatbot, choice, tts_choice], outputs=[chatbot],api_name="Ask_Retriever")
+            # retriver_button.click(fn=add_message, inputs=[chatbot, chat_input], outputs=[chatbot, chat_input]).then(
+            #     fn=bot, inputs=[chatbot, choice, tts_choice], outputs=[chatbot], api_name="Ask_Retriever")
             retriver_button.click(fn=add_message, inputs=[chatbot, chat_input], outputs=[chatbot, chat_input]).then(
-                fn=bot, inputs=[chatbot, choice, tts_choice], outputs=[chatbot], api_name="Ask_Retriever")
+                    fn=bot, inputs=[chatbot, choice, tts_choice, state], outputs=[chatbot], api_name="Ask_Retriever")
+
             #gr.Audio(interactive=False, autoplay=True)]
 
             # voice_response_button = gr.Button("Voice Response")