Update components/chat.py

components/chat.py

@@ -1,3 +1,4 @@
+# components/chat.py
 import streamlit as st
 from langchain.schema import HumanMessage, AIMessage
 
@@ -14,23 +15,16 @@ def display_chat_interface():
     
     # Chat input
     if prompt := st.chat_input("Ask about the RFPs..."):
-        # Add user message to chat history
-        current_msg = HumanMessage(content=prompt)
-        st.session_state.chat_history.append(current_msg)
-        
         try:
             with st.spinner("Analyzing documents..."):
-                # Pass only the messages list without formatting
-                response = st.session_state.qa_system.invoke({
-                    "input": current_msg.content,
-                    "chat_history": st.session_state.chat_history[:-1]  # Exclude current message
-                })
-            
-            # Add AI response to chat history
-            st.session_state.chat_history.append(AIMessage(content=response["output"]))
+                # Get response from QA system
+                response = st.session_state.qa_system({"question": prompt})
+                
+                # Add messages to chat history
+                st.session_state.chat_history.append(HumanMessage(content=prompt))
+                st.session_state.chat_history.append(AIMessage(content=response['answer']))
             
         except Exception as e:
             st.error(f"Error generating response: {e}")
-            # Print more detailed error information
             import traceback
             st.error(f"Detailed error: {traceback.format_exc()}")