Update components/chat.py

components/chat.py

@@ -1,6 +1,7 @@
 # components/chat.py
 import streamlit as st
 from langchain_core.messages import HumanMessage, AIMessage
+import asyncio
 
 def display_chat_interface():
     """Display chat interface component"""
@@ -21,17 +22,26 @@ def display_chat_interface():
     if prompt := st.chat_input("Ask about the RFPs..."):
         try:
             with st.spinner("Analyzing documents..."):
-                # Add user message to history
-                st.session_state.chat_history.append(HumanMessage(content=prompt))
-                
-                # Run the agent
-                response = st.session_state.qa_system.run(prompt)
-                
-                # Add AI response to history
-                st.session_state.chat_history.append(AIMessage(content=response))
+                # 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())
                 
-                # Force refresh to show new messages
-                st.rerun()
+                if response and 'output' in 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']))
+                    
+                    # Force refresh to show new messages
+                    st.rerun()
+                else:
+                    st.error("No valid response received from the agent")
             
         except Exception as e:
             st.error(f"Error generating response: {e}")