Update components/chat.py

components/chat.py

@@ -1,6 +1,7 @@
 # components/chat.py
 import streamlit as st
-from langchain_core.messages import HumanMessage, AIMessage
+from langchain_core.messages import HumanMessage, AIMessage, SystemMessage
+from utils.database import format_chat_history  # Import the formatting function
 
 def display_chat_interface():
     """Display chat interface component"""
@@ -8,30 +9,41 @@ def display_chat_interface():
     
     # Initialize chat history if not exists
     if 'chat_history' not in st.session_state:
-        st.session_state.chat_history = []
+        st.session_state.chat_history = [
+            SystemMessage(content="Chat started. You can ask questions about the RFP documents.")
+        ]
     
     # Display chat history
     for message in st.session_state.chat_history:
         if isinstance(message, HumanMessage):
             st.markdown(f"🧑‍💼 **You**: {message.content}")
-        else:
+        elif isinstance(message, AIMessage):
             st.markdown(f"🤖 **Assistant**: {message.content}")
+        # Don't display system messages in the UI
     
     # Chat input
     if prompt := st.chat_input("Ask about the RFPs..."):
         try:
             with st.spinner("Analyzing documents..."):
-                # Use the correct input format
+                # Add the new message to chat history
+                current_msg = HumanMessage(content=prompt)
+                st.session_state.chat_history.append(current_msg)
+                
+                # Format the chat history properly
+                formatted_history = format_chat_history(st.session_state.chat_history[:-1])
+                
+                # Make the query with properly formatted input
                 result = st.session_state.qa_system({
-                    "input": prompt,  # Changed from "question" to "input"
+                    "input": prompt,
+                    "chat_history": formatted_history
                 })
                 
                 if result and 'answer' in result:
-                    # Add messages to chat history
-                    st.session_state.chat_history.append(HumanMessage(content=prompt))
-                    st.session_state.chat_history.append(AIMessage(content=result['answer']))
+                    # Add AI response to chat history
+                    ai_msg = AIMessage(content=result['answer'])
+                    st.session_state.chat_history.append(ai_msg)
                     
-                    # Force refresh to show new messages
+                    # Rerun to update the display
                     st.rerun()
                 else:
                     st.error("No valid response received from the QA system")