Update components/chat.py

components/chat.py

@@ -113,8 +113,9 @@ def display_chat_interface():
                 })
                 
                 if response:
-                    # Clean the response before creating the message
-                    cleaned_response = clean_response_content(response)
+                    # Extract the content and then clean the response
+                    response_content = response.content
+                    cleaned_response = clean_response_content(response_content)
                     ai_message = AIMessage(content=cleaned_response)
                     st.session_state.messages.append(ai_message)
                     st.rerun()