Update app.py

app.py

@@ -191,22 +191,6 @@ def generate_audio_elevenlabs(text):
         logging.error(f"Error generating audio: {response.text}")
         return None
 
-
-# Define function to generate a streaming response
-def chat_with_bot(messages, user_message):
-    # Add user message to the chat history
-    messages.append((user_message, ""))
-    response = get_response(user_message)
-    
-    # Simulate streaming response by iterating over each character in the response
-    for character in response:
-        messages[-1] = (user_message, messages[-1][1] + character)
-        yield messages  # Stream each character
-        time.sleep(0.05)  # Adjust delay as needed for real-time effect
-
-    yield messages  # Final yield to ensure full response is displayed
-
-
 # Function to generate audio with Eleven Labs TTS from the last bot response
 def generate_audio_from_last_response(history):
     # Get the most recent bot response from the chat history
@@ -229,10 +213,32 @@ examples = [
 def insert_prompt(current_text, prompt):
     return prompt[0] if prompt else current_text
 
+# Function to add a user's message to the chat history and clear the input box
+def add_message(history, message):
+    history.append((message, None))  # Add the user's message to the chat history
+    return history, gr.Textbox(value="", interactive=True, show_label=False)  # Clear the input box
+
+
+# Define the function to handle the conversation and get the response in a streaming way
+def chat_with_bot(history, user_message):
+    # Add user message to the chat history
+    history.append((user_message, ""))
+
+    # Generate the response in a streaming manner
+    response = get_response(user_message)
+    
+    # Simulate streaming by yielding each character of the response
+    for character in response:
+        history[-1] = (user_message, history[-1][1] + character)
+        yield history  # Stream each character
+        time.sleep(0.05)  # Adjust delay as needed for real-time effect
+
+    yield history  # Final yield to ensure full response is displayed
+
 
 
 # Create the Gradio Blocks interface
-with gr.Blocks(theme="Pijush2023/pijush_theme_Everforest") as demo:
+with gr.Blocks() as demo:
     chatbot = gr.Chatbot([], elem_id="RADAR", bubble_full_width=False)
     with gr.Row():
         with gr.Column():
@@ -257,7 +263,8 @@ with gr.Blocks(theme="Pijush2023/pijush_theme_Everforest") as demo:
             gr.Examples(examples=examples, fn=insert_prompt, inputs=question_input, outputs=question_input)
 
     # Define interactions
-    get_response_btn.click(fn=chat_with_bot, inputs=[chatbot, question_input], outputs=chatbot)
+    get_response_btn.click(fn=add_message, inputs=[chatbot, question_input], outputs=[chatbot, question_input])\
+                    .then(fn=chat_with_bot, inputs=[chatbot, question_input], outputs=chatbot)
     generate_audio_btn.click(fn=generate_audio_from_last_response, inputs=chatbot, outputs=audio_output)
     clean_btn.click(fn=clear_fields, inputs=[], outputs=[chatbot, question_input, audio_output])