Update components/chat.py

components/chat.py

@@ -1,15 +1,5 @@
 import streamlit as st
-from langchain.schema import HumanMessage, AIMessage, SystemMessage
-
-def format_chat_history(chat_history):
-    """Format chat history for QA system"""
-    formatted_history = []
-    for msg in chat_history[:-1]:  # Exclude the most recent message
-        if isinstance(msg, HumanMessage):
-            formatted_history.append(SystemMessage(content=f"Human: {msg.content}"))
-        elif isinstance(msg, AIMessage):
-            formatted_history.append(SystemMessage(content=f"Assistant: {msg.content}"))
-    return formatted_history
+from langchain.schema import HumanMessage, AIMessage
 
 def display_chat_interface():
     """Display chat interface component"""
@@ -25,21 +15,22 @@ def display_chat_interface():
     # Chat input
     if prompt := st.chat_input("Ask about the RFPs..."):
         # Add user message to chat history
-        st.session_state.chat_history.append(HumanMessage(content=prompt))
+        current_msg = HumanMessage(content=prompt)
+        st.session_state.chat_history.append(current_msg)
         
         try:
             with st.spinner("Analyzing documents..."):
-                # Create formatted history that Langchain expects
-                formatted_history = format_chat_history(st.session_state.chat_history)
-                
-                # Make the query with formatted history
+                # Pass only the messages list without formatting
                 response = st.session_state.qa_system.invoke({
-                    "input": prompt,
-                    "chat_history": formatted_history
+                    "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"]))
             
         except Exception as e:
-            st.error(f"Error generating response: {e}")
+            st.error(f"Error generating response: {e}")
+            # Print more detailed error information
+            import traceback
+            st.error(f"Detailed error: {traceback.format_exc()}")