Update components/chat.py

components/chat.py

@@ -5,51 +5,108 @@ from utils.database import verify_vector_store
 
 
 def display_chat_interface():
-    """Display chat interface component"""
-    st.markdown("### 💬 RFP Analysis Chat")
+    """Display modern chat interface with clean formatting."""
     
-    # Initialize chat history if not exists
+    # 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;
+        }
+        
+        .source-info {
+            background-color: #f8f9fa;
+            padding: 0.5rem;
+            border-left: 3px solid #0f52ba;
+            margin-top: 1rem;
+            font-size: 0.9rem;
+        }
+        
+        /* Headers and content */
+        .content-section {
+            margin: 0.5rem 0;
+        }
+        
+        /* Clean bullets */
+        .clean-bullet {
+            margin: 0.3rem 0;
+        }
+    </style>
+    """, unsafe_allow_html=True)
+    
+    # Initialize chat history
     if 'messages' not in st.session_state:
         st.session_state.messages = []
     
     # Display chat history
     for message in st.session_state.messages:
         if isinstance(message, HumanMessage):
-            st.markdown(f"🧑‍💼 **You**: {message.content}")
+            st.markdown(f"""
+            <div class="user-message">
+                🧑‍💼 <strong>You:</strong><br>{message.content}
+            </div>
+            """, unsafe_allow_html=True)
         elif isinstance(message, AIMessage):
-            st.markdown(f"🤖 **Assistant**: {message.content}")
+            # Clean and structure the response
+            response_content = message.content
+            if hasattr(message, 'additional_kwargs'):
+                response_content = response_content.replace('additional_kwargs={}', '').replace('response_metadata={}', '')
+            
+            st.markdown(f"""
+            <div class="assistant-message">
+                🤖 <strong>Assistant:</strong><br>
+                {response_content}
+            </div>
+            """, unsafe_allow_html=True)
     
-    # Chat input
-    if prompt := st.chat_input("Ask about the RFPs..."):
+    # Chat input at the bottom
+    if prompt := st.chat_input("Ask about your documents..."):
         try:
-            with st.spinner("Analyzing documents..."):
+            with st.spinner("Analyzing..."):
                 # Create message objects
                 human_message = HumanMessage(content=prompt)
-                
-                # Add to chat history
                 st.session_state.messages.append(human_message)
                 
-                # Get response from QA system
+                # Get response
                 response = st.session_state.qa_system.invoke({
                     "input": prompt,
                     "chat_history": st.session_state.messages
                 })
                 
                 if response:
-                    # Create AI message
                     ai_message = AIMessage(content=str(response))
-                    
-                    # Add to chat history
                     st.session_state.messages.append(ai_message)
-                    
-                    # Force refresh to show new messages
                     st.rerun()
                 else:
                     st.error("No valid response received")
-            
+                    
         except Exception as e:
             st.error(f"Error: {e}")
-            st.error(f"Error type: {type(e)}")
             import traceback
             st.error(traceback.format_exc())