Update components/chat.py

components/chat.py

@@ -3,34 +3,39 @@ from langchain_core.messages import HumanMessage, AIMessage
 from utils.database import verify_vector_store
 
 def display_chat_interface():
+    
+    
+     # Check if vector store exists and is properly loaded
+    if 'vector_store' not in st.session_state:
+        st.warning("📚 Please upload documents first.")
+        return
+        
+    # Verify vector store has documents
+    if not verify_vector_store(st.session_state.vector_store):
+        st.warning("⌛ Documents are still being processed. Please wait...")
+        return
+    
     """Display chat interface component"""
     st.markdown("### 💬 RFP Analysis Chat")
     
-    # Initialize chat history if not exists
     if 'messages' not in st.session_state:
         st.session_state.messages = []
     
-    # Display chat history
     for message in st.session_state.messages:
         if isinstance(message, HumanMessage):
             st.markdown(f"🧑‍💼 **You**: {message.content}")
         elif isinstance(message, AIMessage):
             st.markdown(f"🤖 **Assistant**: {message.content}")
     
-    # Chat input
-    if prompt := st.chat_input("Ask about the RFPs..."):
+    if prompt := st.chat_input("Ask about the RFPs...", disabled=not st.session_state.get('vector_store')):
         try:
             with st.spinner("Analyzing documents..."):
-                # Create message objects
                 human_message = HumanMessage(content=prompt)
-                
-                # Add to chat history
                 st.session_state.messages.append(human_message)
                 
-                # Get response
                 response = st.session_state.qa_system.invoke({
                     "input": prompt,
-                    "chat_history": list(st.session_state.messages)  # Ensure it's a list
+                    "chat_history": list(st.session_state.messages)
                 })
                 
                 if response:
@@ -39,9 +44,8 @@ def display_chat_interface():
                     st.rerun()
                 else:
                     st.error("No valid response received")
-            
         except Exception as e:
             st.error(f"Error: {e}")
-            st.error(f"Error type: {type(e)}")
             import traceback
-            st.error(traceback.format_exc())
+            st.error(traceback.format_exc())
+