Update app.py

app.py

@@ -1438,39 +1438,36 @@ if "message_limit" not in st.session_state:
 if "used_messages" not in st.session_state:
     st.session_state["used_messages"] = 0
 
-def initialize_user_session(session_user_id, email):
+def initialize_user_session():
     try:
-        # Simulate backend initialization
-        response = requests.post(f"{backend_url}/initialize_session_user", json={
+        user_id = st.session_state["wix_user_id"]
+        email = st.session_state["email"]
+        user_data = {
+            "user_name": "Demo",
             "email": email,
-            "sessionUserId": session_user_id,
-            "userName": "Demo",
-            "messageLimit": 1000
-        })
+            "message_limit": 1000,
+            "used_messages": 0,
+        }
 
-        if response.status_code == 200:
-            data = response.json()
-            user_data = data.get("user_data", {})
-            st.session_state["message_limit"] = user_data.get("message_limit", 1000)
-            st.session_state["used_messages"] = user_data.get("used_messages", 0)
-        else:
-            st.error(f"Failed to initialize session user. Status Code: {response.status_code}")
-            st.error(f"Response: {response.text}")
-    except Exception as e:
-        st.error(f"Error initializing session user: {str(e)}")
-        st.session_state["message_limit"] = 1000
-        st.session_state["used_messages"] = 0
+        # Save user session to Firebase
+        db.child("users").child(user_id).set(user_data)
 
+        # Update session state
+        st.session_state["user_name"] = user_data["user_name"]
+        st.session_state["message_limit"] = user_data["message_limit"]
+        st.session_state["used_messages"] = user_data["used_messages"]
+
+        st.success("User session initialized successfully.")
+    except Exception as e:
+        st.error(f"Error initializing user session: {e}")
    
 
 # Update message usage in Firebase after each interaction
 def update_message_usage(wix_user_id):
-    response = requests.post(f"{backend_url}/update-message-usage/{wix_user_id}")
-    if response.status_code == 200:
-        st.session_state["used_messages"] += 1
-        update_message_counter()
-    else:
-        st.error("Error updating message usage.")
+  
+    st.session_state["used_messages"] += 1
+    update_message_counter()
+