Update app.py

app.py

@@ -54,10 +54,12 @@ import re
 # Set up logging to suppress Streamlit warnings about experimental functions
 logging.getLogger('streamlit').setLevel(logging.ERROR)
 
-if "chat_history" not in st.session_state:
-    st.session_state["chat_history"] = []
 if "documents" not in st.session_state:
     st.session_state["documents"] = {}
+    
+if "chat_history" not in st.session_state:
+    st.session_state["chat_history"] = []
+
 if "message_limit" not in st.session_state:
     st.session_state["message_limit"] = 0
 if "used_messages" not in st.session_state:
@@ -1960,7 +1962,16 @@ def handle_prompt(prompt):
                     # Prevent duplicate messages in chat history
                     if not any(msg["content"] == prompt for msg in st.session_state["chat_history"]):
                         st.session_state.chat_history.append({"role": "user", "content": prompt})
+                    document_response = handle_document_query(prompt)
+                    if document_response:
+                        with st.chat_message("assistant"):
+                            st.markdown(document_response)
+                        st.session_state["handled"] = True  # Mark as handled
+                        return
 
+                    
+
+                    
         
                     # Handle missing Trust Bucket if needed
                     if st.session_state.get("missing_trustbucket_content") and not st.session_state["handled"]: