Update components/chat.py

components/chat.py

@@ -1,7 +1,6 @@
 # components/chat.py
 import streamlit as st
-from langchain_core.messages import HumanMessage, AIMessage, SystemMessage
-from utils.database import format_chat_history  # Import the formatting function
+from langchain_core.messages import HumanMessage, AIMessage
 
 def display_chat_interface():
     """Display chat interface component"""
@@ -9,9 +8,7 @@ def display_chat_interface():
     
     # Initialize chat history if not exists
     if 'chat_history' not in st.session_state:
-        st.session_state.chat_history = [
-            SystemMessage(content="Chat started. You can ask questions about the RFP documents.")
-        ]
+        st.session_state.chat_history = []
     
     # Display chat history
     for message in st.session_state.chat_history:
@@ -19,34 +16,22 @@ def display_chat_interface():
             st.markdown(f"🧑‍💼 **You**: {message.content}")
         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..."):
-                # Add the new message to chat history
-                current_msg = HumanMessage(content=prompt)
-                st.session_state.chat_history.append(current_msg)
+                # Add user message to history
+                st.session_state.chat_history.append(HumanMessage(content=prompt))
                 
-                # Format the chat history properly
-                formatted_history = format_chat_history(st.session_state.chat_history[:-1])
+                # Run the agent
+                response = st.session_state.qa_system.run(prompt)
                 
-                # Make the query with properly formatted input
-                result = st.session_state.qa_system({
-                    "input": prompt,
-                    "chat_history": formatted_history
-                })
+                # Add AI response to history
+                st.session_state.chat_history.append(AIMessage(content=response))
                 
-                if result and 'answer' in result:
-                    # Add AI response to chat history
-                    ai_msg = AIMessage(content=result['answer'])
-                    st.session_state.chat_history.append(ai_msg)
-                    
-                    # Rerun to update the display
-                    st.rerun()
-                else:
-                    st.error("No valid response received from the QA system")
+                # Force refresh to show new messages
+                st.rerun()
             
         except Exception as e:
             st.error(f"Error generating response: {e}")