Update app.py

app.py

@@ -48,7 +48,7 @@ import asyncio
 import re 
 # Set up logging to suppress Streamlit warnings about experimental functions
 logging.getLogger('streamlit').setLevel(logging.ERROR)
-
+INITIAL_MESSAGE_LIMIT = 100
 
 
 
@@ -1208,10 +1208,31 @@ def display_save_confirmation(type_saved):
 # Function to update the message counter in a static location
 message_counter_placeholder = st.sidebar.empty()
 
+def initialize_message_limit(user_id):
+    user_data = db.child("users").child(user_id).get().val()
+    if user_data:
+        st.session_state["message_limit"] = user_data.get("message_limit", INITIAL_MESSAGE_LIMIT)
+        st.session_state["used_messages"] = user_data.get("used_messages", 0)
+    else:
+        # Set default limits for new users and store them in Firebase
+        st.session_state["message_limit"] = INITIAL_MESSAGE_LIMIT
+        st.session_state["used_messages"] = 0
+        db.child("users").child(user_id).update({
+            "message_limit": INITIAL_MESSAGE_LIMIT,
+            "used_messages": 0
+        })
+
+def update_message_usage(user_id):
+    st.session_state["used_messages"] += 1
+    db.child("users").child(user_id).update({
+        "used_messages": st.session_state["used_messages"]
+    })
+    update_message_counter()
 def update_message_counter():
     remaining_messages = st.session_state["message_limit"] - st.session_state["used_messages"]
     message_counter_placeholder.markdown(f"**Messages Left**: {remaining_messages} / {st.session_state['message_limit']}")
 
+    
 if st.session_state.get("wix_user_id") and st.session_state.get("email"):
     retrieve_user_data(st.session_state["wix_user_id"])
     display_saved_trustbuilders()
@@ -1221,24 +1242,7 @@ if st.session_state.get("wix_user_id") and st.session_state.get("email"):
     welcome_placeholder.info(f"**Hello, {user_name}!**")
     time.sleep(3)
     welcome_placeholder.empty()
-    def check_user_subscription(wix_user_id):
-        response = requests.get(f"{backend_url}/check-subscription/{wix_user_id}")
-        if response.status_code == 200:
-            data = response.json()
-            st.session_state["message_limit"] = data.get("message_limit", 0)
-            st.session_state["used_messages"] = data.get("used_messages", 0)
-            update_message_counter()
-        else:
-            st.error("Error fetching user subscription details.")
-
-    def update_message_usage(wix_user_id):
-        response = requests.post(f"{backend_url}/update-message-usage/{st.session_state['wix_user_id']}")
-        if response.status_code == 200:
-            st.session_state["used_messages"] += 1
-            update_message_counter()
-        else:
-            st.error("Error updating message usage.")
-    check_user_subscription(st.session_state["wix_user_id"])
+  
 
     # Input for AI interaction
     prompt = st.chat_input("")