Update components/chat.py

components/chat.py

@@ -271,6 +271,7 @@ def display_chat_interface():
                 </div>
                 """, unsafe_allow_html=True)
         
+        # Chat input
         # Chat input
         if prompt := st.chat_input("Ask about your documents..."):
             with st.spinner("Analyzing..."):
@@ -284,22 +285,12 @@ def display_chat_interface():
                     human_message = HumanMessage(content=prompt)
                     st.session_state.messages.append(human_message)
                     
-                    # 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
-                            })
-                    
                     # Get response from QA system
                     response = st.session_state.qa_system.invoke({
                         "input": prompt,
-                        "chat_history": []  # Start with empty history if causing issues
+                        "chat_history": []
                     })
                     
-                    # Handle response
                     if response:
                         ai_message = AIMessage(content=str(response))
                         st.session_state.messages.append(ai_message)
@@ -318,7 +309,7 @@ def display_chat_interface():
                         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())