Update components/chat.py

components/chat.py

@@ -1,38 +1,45 @@
-# src/components/chat.py
 import streamlit as st
-from langchain.schema import HumanMessage, AIMessage
+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]:
+    for msg in chat_history[:-1]:  # Exclude the most recent message
         if isinstance(msg, HumanMessage):
-            formatted_history.append({"role": "user", "content": msg.content})
-        else:
-            formatted_history.append({"role": "assistant", "content": msg.content})
+            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
 
 def display_chat_interface():
     """Display chat interface component"""
     st.markdown("### 💬 RFP Analysis Chat")
     
+    # Display chat history
     for message in st.session_state.chat_history:
         if isinstance(message, HumanMessage):
             st.markdown(f"🧑‍💼 **You**: {message.content}")
         else:
             st.markdown(f"🤖 **Assistant**: {message.content}")
     
+    # 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))
         
         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
                 response = st.session_state.qa_system.invoke({
                     "input": prompt,
                     "chat_history": formatted_history
                 })
             
+            # 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}")