Update app.py

app.py

@@ -6,7 +6,7 @@ from langchain_core.output_parsers import StrOutputParser
 from langchain_openai import ChatOpenAI
 from langchain_community.graphs import Neo4jGraph
 from typing import List, Tuple
-from langchain_core.pydantic_v1 import BaseModel, Field
+from pydantic import BaseModel, Field
 from langchain_core.messages import AIMessage, HumanMessage
 from langchain_core.runnables import (
     RunnableBranch,
@@ -203,20 +203,19 @@ def generate_audio_elevenlabs(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, ""))
-
-    # Generate the response in a streaming manner
     response = get_response(user_message)
     
-    # Simulate streaming by yielding each character of the response
+    # 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  # Update the chat history with the latest response incrementally
-        time.sleep(0.05)  # Optional: Add a slight delay to simulate real-time streaming
+        yield messages  # Stream each character
+        time.sleep(0.05)  # Adjust delay as needed for real-time effect
 
-    yield messages  # Ensure the final response is fully displayed
+    yield messages  # Final yield to ensure full response is displayed
 
 
 # Create the Gradio Blocks interface
@@ -250,9 +249,9 @@ with gr.Blocks() as demo:
     #clean_btn.click(fn=clear_fields, inputs=[], outputs=[question_input, response_output])
 
     # Define interactions
-    get_response_btn.click(fn=chat_with_bot, inputs=[chatbot, question_input], outputs=[chatbot, response_output])
-    generate_audio_btn.click(fn=generate_audio_elevenlabs, inputs=response_output, outputs=audio_output)
-    clean_btn.click(fn=clear_fields, inputs=[], outputs=[chatbot, question_input, response_output, audio_output])
+    get_response_btn.click(fn=chat_with_bot, inputs=[chatbot, question_input], outputs=chatbot)
+    generate_audio_btn.click(fn=generate_audio_elevenlabs, inputs=question_input, outputs=audio_output)
+    clean_btn.click(fn=clear_fields, inputs=[], outputs=[chatbot, question_input, audio_output])
 
 # Launch the Gradio interface
 demo.launch(show_error=True)