Update app.py

app.py

@@ -1213,14 +1213,23 @@ def update_message_counter():
     message_counter_placeholder.markdown(f"**Messages Left**: {remaining_messages} / {st.session_state['message_limit']}")
 
 def initialize_user_data(wix_user_id):
-    response = requests.get(f"{backend_url}/check-user/{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
-        st.session_state["used_messages"] = data.get("used_messages", 0)
+    try:
+        response = requests.get(f"{backend_url}/check-user/{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
+            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()
+    except Exception as e:
+        st.error("Error initializing user data. Default message limit applied.")
+        # Set defaults if there’s a failure in fetching
+        st.session_state["message_limit"] = 100
+        st.session_state["used_messages"] = 0
         update_message_counter()
-    else:
-        st.error("Error fetching user data.")
 
 # Update message usage in Firebase after each interaction
 def update_message_usage(wix_user_id):