Update components/chat.py

components/chat.py

@@ -4,6 +4,9 @@ from utils.database import verify_vector_store
 
 def clean_response_content(response_content):
     """Clean the response content of any technical metadata."""
+    # Convert to string if it's an AIMessage
+    response_content = str(response_content)
+    
     # Remove the content= prefix if it exists
     if "content='" in response_content:
         response_content = response_content.split("content='")[1]
@@ -23,86 +26,7 @@ def clean_response_content(response_content):
 def display_chat_interface():
     """Display modern chat interface with clean formatting."""
     
-    # Add custom CSS for modern chat styling
-    st.markdown("""
-    <style>
-        /* Clean, modern chat container */
-        .chat-container {
-            max-width: 800px;
-            margin: auto;
-        }
-        
-        /* Message styling */
-        .user-message {
-            background-color: #f0f2f6;
-            padding: 1rem;
-            border-radius: 10px;
-            margin: 1rem 0;
-        }
-        
-        .assistant-message {
-            background-color: #ffffff;
-            border: 1px solid #e0e0e0;
-            padding: 1.5rem;
-            border-radius: 10px;
-            margin: 1rem 0;
-        }
-        
-        /* Section styling */
-        .section-header {
-            font-weight: bold;
-            color: #0f52ba;
-            margin-top: 1rem;
-            font-size: 1.2em;
-        }
-        
-        .source-info {
-            background-color: #f8f9fa;
-            padding: 0.5rem;
-            border-left: 3px solid #0f52ba;
-            margin-top: 1rem;
-            font-size: 0.9rem;
-            font-style: italic;
-        }
-        
-        /* Summary section */
-        .summary-section {
-            background-color: #f8f9fa;
-            padding: 1rem;
-            border-radius: 6px;
-            margin-bottom: 1rem;
-        }
-        
-        /* Key points */
-        .key-points {
-            margin: 1rem 0;
-        }
-        
-        .key-point-item {
-            margin: 0.5rem 0;
-            line-height: 1.5;
-            padding-left: 1.5rem;
-            position: relative;
-        }
-        
-        .key-point-item:before {
-            content: "•";
-            position: absolute;
-            left: 0.5rem;
-        }
-        
-        /* Content sections */
-        .content-section {
-            margin: 1rem 0;
-        }
-        
-        /* Bold text */
-        .highlight {
-            font-weight: 600;
-            color: #1a365d;
-        }
-    </style>
-    """, unsafe_allow_html=True)
+    [Previous CSS styling remains the same...]
     
     # Check if QA system is initialized
     if 'qa_system' not in st.session_state or st.session_state.qa_system is None:
@@ -166,7 +90,8 @@ def display_chat_interface():
                 })
                 
                 if response:
-                    ai_message = AIMessage(content=str(response))
+                    response_str = str(response)
+                    ai_message = AIMessage(content=response_str)
                     st.session_state.messages.append(ai_message)
                     st.rerun()
                 else: