Update components/chat.py

components/chat.py

@@ -1,7 +1,6 @@
 # components/chat.py
 import streamlit as st
 from langchain_core.messages import HumanMessage, AIMessage
-import asyncio
 
 def display_chat_interface():
     """Display chat interface component"""
@@ -22,21 +21,15 @@ def display_chat_interface():
     if prompt := st.chat_input("Ask about the RFPs..."):
         try:
             with st.spinner("Analyzing documents..."):
-                # Use asyncio to handle the async call
-                async def get_response():
-                    response = await st.session_state.qa_system.acall({
-                        "input": prompt,
-                        "chat_history": st.session_state.chat_history
-                    })
-                    return response
-
-                # Run the async function
-                response = asyncio.run(get_response())
+                # Pass the question through the chain
+                response = st.session_state.qa_system.invoke({
+                    "question": prompt  # Changed from "input" to "question"
+                })
                 
-                if response and 'output' in response:
+                if response:
                     # Add messages to chat history
                     st.session_state.chat_history.append(HumanMessage(content=prompt))
-                    st.session_state.chat_history.append(AIMessage(content=response['output']))
+                    st.session_state.chat_history.append(AIMessage(content=response))
                     
                     # Force refresh to show new messages
                     st.rerun()