Update app.py

app.py

@@ -58,6 +58,10 @@ 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 "message_limit" not in st.session_state:
+    st.session_state["message_limit"] = 0
+if "used_messages" not in st.session_state:
+    st.session_state["used_messages"] = 0
 
 # Initialize logging and load environment variables
 logging.basicConfig(level=logging.INFO)