Update app.py

app.py

@@ -25,8 +25,12 @@ def register_callback():
     password = st.session_state.reg_password
     try:
         # Create a new user in Firebase
-        user = auth.create_user(email=email, password=password)
-        st.success("Registration successful! Please log in.")
+        user = auth.create_user_with_email_and_password(email, password)
+        st.success("Registration successful! Please check your email for verification.")
+
+        # Send verification email
+        auth.send_email_verification(user['idToken'])
+        st.info("A verification email has been sent to your email address.")
     except Exception as e:
         st.error(f"Registration failed: {e}")
 
@@ -36,10 +40,19 @@ def login_callback():
     password = st.session_state.login_password
     try:
         # Authenticate user with Firebase
-        user = auth.get_user_by_email(email)
-        st.session_state.logged_in = True
-        st.session_state.current_user = user.uid  # Store the user UID in session state
-        st.success("Logged in successfully!")
+        user = auth.sign_in_with_email_and_password(email, password)
+        
+        # Check if email is verified
+        user_info = auth.get_account_info(user['idToken'])
+        if user_info['users'][0]['emailVerified']:
+            st.session_state.logged_in = True
+            st.session_state.current_user = user['localId']
+            st.success("Logged in successfully!")
+        else:
+            st.error("Please verify your email address before logging in.")
+            # Resend verification email if needed
+            auth.send_email_verification(user['idToken'])
+            st.info("A new verification email has been sent to your email address.")
     except Exception as e:
         st.error(f"Login failed: {e}")
 
@@ -86,4 +99,4 @@ if st.session_state.logged_in:
     main_app()
 else:
     registration_form()
-    login_form()
+    login_form()