Update app.py

app.py

@@ -55,6 +55,18 @@ logging.getLogger('streamlit').setLevel(logging.ERROR)
 INITIAL_MESSAGE_LIMIT = 100
 
 
+if "wix_user_id" not in st.session_state:
+    st.session_state["wix_user_id"] = str(uuid.uuid4())  # Assign a unique ID for each session
+if "email" not in st.session_state:
+    st.session_state["email"] = "demo@example.com"  # Default email for demo
+if "message_limit" not in st.session_state:
+    st.session_state["message_limit"] = 1000
+if "used_messages" not in st.session_state:
+    st.session_state["used_messages"] = 0
+if "chat_history" not in st.session_state:
+    st.session_state["chat_history"] = []
+
+
 # Initialize logging and load environment variables
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -1460,11 +1472,7 @@ def update_message_usage(wix_user_id):
     else:
         st.error("Error updating message usage.")
 
-# Main interface
-if "wix_user_id" not in st.session_state:
-    st.session_state["wix_user_id"] = None
 
-st.session_state["wix_user_id"] = str(uuid.uuid4())
 
 initialize_user_session(st.session_state["wix_user_id"], st.session_state["email"])