Update app.py

app.py

@@ -1212,24 +1212,24 @@ 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']}")
 
-def initialize_user_data(wix_user_id):
+def initialize_wix_only_user(wix_user_id, email):
     try:
-        response = requests.get(f"{backend_url}/check-user/{wix_user_id}")
+        response = requests.post(f"{backend_url}/wix_user_login_no_chargebee", json={
+            "email": email,
+            "wixUserId": wix_user_id
+        })
+        
         if response.status_code == 200:
-            data = response.json()
-            st.session_state["message_limit"] = data.get("message_limit", 100)  # Default limit is 100 if none is found
+            data = response.json().get("user_data", {})
+            st.session_state["message_limit"] = data.get("message_limit", 100)
             st.session_state["used_messages"] = data.get("used_messages", 0)
         else:
-            # Set default values if there’s an error fetching from Firebase
-            st.session_state["message_limit"] = 100
-            st.session_state["used_messages"] = 0
-        update_message_counter()
+            st.error("Error logging in as a Wix-only user.")
     except Exception as e:
-        st.error("Error initializing user data. Default message limit applied.")
-        # Set defaults if there’s a failure in fetching
+        st.error("Error initializing user data for Wix-only login.")
         st.session_state["message_limit"] = 100
         st.session_state["used_messages"] = 0
-        update_message_counter()
+    update_message_counter()
 
 # Update message usage in Firebase after each interaction
 def update_message_usage(wix_user_id):
@@ -1252,7 +1252,7 @@ if st.session_state.get("wix_user_id") and st.session_state.get("email"):
     welcome_placeholder.empty()
 
     # Initialize the user's message limit and usage from Firebase
-    initialize_user_data(st.session_state["wix_user_id"])
+    initialize_wix_only_user(st.session_state["wix_user_id"])
 
     # Input field for chatbot interaction
     prompt = st.chat_input("")