Update components/chat.py

components/chat.py

@@ -9,12 +9,6 @@ def display_chat_interface():
     if 'messages' not in st.session_state:
         st.session_state.messages = []
     
-    # DEBUG: Print current state of messages
-    st.write("DEBUG - Current messages in session state:")
-    for msg in st.session_state.messages:
-        st.write(f"Message type: {type(msg)}")
-        st.write(f"Message content: {msg.content}")
-    
     # Display chat history
     for message in st.session_state.messages:
         if isinstance(message, HumanMessage):
@@ -26,50 +20,27 @@ def display_chat_interface():
     if prompt := st.chat_input("Ask about the RFPs..."):
         try:
             with st.spinner("Analyzing documents..."):
-                # Add human message to chat history
+                # Create message objects
                 human_message = HumanMessage(content=prompt)
-                st.session_state.messages.append(human_message)
                 
-                # DEBUG: Print message object details
-                st.write("DEBUG - New human message:")
-                st.write(f"Type: {type(human_message)}")
-                st.write(f"Content: {human_message.content}")
-                st.write("DEBUG - Current chat history types:")
-                st.write([type(msg) for msg in st.session_state.messages])
-                
-                # Get response from QA system
-                st.write("DEBUG - About to invoke QA system with these inputs:")
-                st.write({
-                    "input": prompt,
-                    "chat_history": [str(msg) for msg in st.session_state.messages]
-                })
+                # Add to chat history
+                st.session_state.messages.append(human_message)
                 
+                # Get response
                 response = st.session_state.qa_system.invoke({
                     "input": prompt,
-                    "chat_history": st.session_state.messages
+                    "chat_history": list(st.session_state.messages)  # Ensure it's a list
                 })
                 
                 if response:
-                    # DEBUG: Print response details
-                    st.write("DEBUG - Response received:")
-                    st.write(f"Response type: {type(response)}")
-                    st.write(f"Response content: {response}")
-                    
-                    # Add AI message to chat history
                     ai_message = AIMessage(content=str(response))
                     st.session_state.messages.append(ai_message)
-                    
-                    # Force refresh to show new messages
                     st.rerun()
                 else:
                     st.error("No valid response received")
             
         except Exception as e:
-            st.error(f"Error generating response: {e}")
+            st.error(f"Error: {e}")
+            st.error(f"Error type: {type(e)}")
             import traceback
-            st.error(f"Detailed error: {traceback.format_exc()}")
-            
-            # DEBUG: Print the state when error occurs
-            st.write("DEBUG - State at error:")
-            st.write("Messages types:", [type(msg) for msg in st.session_state.messages])
-            st.write("QA system type:", type(st.session_state.qa_system))
+            st.error(traceback.format_exc())