Update app.py

app.py

@@ -72,19 +72,25 @@ runnable_sequence = RunnableSequence(prompt | llm)
 
 # Initialize memory
 
-# Define your get_text_response function
 def get_text_response(user_message, history):
-    # Add user message to the memory (for example purposes, assume `add_user_message` is the correct method)
+    # Ensure user_message is a string
+    if not isinstance(user_message, str):
+        raise ValueError("user_message must be a string")
+    
+    # Save the user message to the memory
     memory.save_context({"user_message": user_message}, None)
     
     # Use the RunnableSequence to generate a response
     response = runnable_sequence.run(user_message=user_message)
     
-    # Add LLM response to the memory (for example purposes, assume `add_ai_message` is the correct method)
+    # Ensure response is a string
+    if not isinstance(response, str):
+        raise ValueError("Response must be a string")
+    
+    # Save the LLM response to the memory
     memory.save_context(None, {"chat_history": response})
     
     return response
-
 # Example usage with Gradio
 theme = "default"  # or your custom theme