Update components/chat.py

components/chat.py

@@ -2,19 +2,6 @@ import streamlit as st
 from langchain_core.messages import HumanMessage, AIMessage
 from utils.database import verify_vector_store
 
-def clean_response_content(response_content):
-    """Clean the response content of any technical metadata."""
-    # Convert the response to string first
-    content = str(response_content)
-    
-    # Clean up the content
-    if "content='" in content:
-        content = content.split("content='")[1].split("'")[0]
-    elif 'content="' in content:
-        content = content.split('content="')[1].split('"')[0]
-        
-    return content.replace('\\n', '\n')
-
 def display_chat_interface():
     """Display modern chat interface with clean formatting."""
     
@@ -88,9 +75,7 @@ def display_chat_interface():
             </div>
             """, unsafe_allow_html=True)
         elif isinstance(message, AIMessage):
-            # Get the content and clean it
-            content = clean_response_content(message.content)
-            
+            content = str(message.content)
             st.markdown(f"""
             <div class="assistant-message">
                 🤖 <strong>Assistant:</strong><br>
@@ -113,10 +98,8 @@ def display_chat_interface():
                 })
                 
                 if 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)
+                    content = str(response)
+                    ai_message = AIMessage(content=content)
                     st.session_state.messages.append(ai_message)
                     st.rerun()
                 else:
@@ -125,5 +108,4 @@ def display_chat_interface():
         except Exception as e:
             st.error(f"Error: {e}")
             import traceback
-            st.error(traceback.format_exc())
-    
+            st.error(traceback.format_exc())