Update components/chat.py

components/chat.py

@@ -279,38 +279,50 @@ def display_chat_interface():
                     st.warning("Please enter a valid question.")
                     return
                 
-                # Create and append human message
-                human_message = HumanMessage(content=prompt)
-                st.session_state.messages.append(human_message)
-                
-                # Get response from QA system
-                response = st.session_state.qa_system.invoke({
-                    "input": prompt,
-                    "chat_history": st.session_state.messages
-                })
-                
-                # Handle response
-                if response:
-                    ai_message = AIMessage(content=str(response))
-                    st.session_state.messages.append(ai_message)
+                try:
+                    # Create and append human message
+                    human_message = HumanMessage(content=prompt)
+                    st.session_state.messages.append(human_message)
                     
-                    # Save chat history
-                    if 'current_session_id' in st.session_state:
-                        st.session_state.persistence.save_chat_history(
-                            st.session_state.messages,
-                            st.session_state.current_session_id,
-                            metadata={
-                                'last_question': prompt,
-                                'last_updated': datetime.now().isoformat()
-                            }
-                        )
+                    # Format chat history properly
+                    chat_history = []
+                    for msg in st.session_state.messages[:-1]:  # Exclude the current message
+                        if isinstance(msg, (HumanMessage, AIMessage)):
+                            chat_history.append({
+                                "role": "user" if isinstance(msg, HumanMessage) else "assistant",
+                                "content": msg.content
+                            })
                     
-                    st.rerun()
-                else:
-                    st.error("No valid response received. Please try again.")
+                    # Get response from QA system
+                    response = st.session_state.qa_system.invoke({
+                        "input": prompt,
+                        "chat_history": []  # Start with empty history if causing issues
+                    })
+                    
+                    # Handle response
+                    if response:
+                        ai_message = AIMessage(content=str(response))
+                        st.session_state.messages.append(ai_message)
+                        
+                        # Save chat history
+                        if 'current_session_id' in st.session_state:
+                            st.session_state.persistence.save_chat_history(
+                                st.session_state.messages,
+                                st.session_state.current_session_id,
+                                metadata={
+                                    'last_question': prompt,
+                                    'last_updated': datetime.now().isoformat()
+                                }
+                            )
+                        
+                        st.rerun()
+                    else:
+                        st.error("No valid response received. Please try again.")
+                        
+                except Exception as e:
+                    st.error(f"Error processing response: {str(e)}")
+                    st.error(traceback.format_exc())
+                    # Remove the failed message attempt
+                    if st.session_state.messages:
+                        st.session_state.messages.pop()
                     
-    except Exception as e:
-        st.error("An error occurred during chat processing.")
-        st.error(f"Error details: {str(e)}")
-        import traceback
-        st.error(traceback.format_exc())