Update app.py

app.py

@@ -27,7 +27,10 @@ def register_callback():
     try:
         # Create a new user in Firebase
         user = auth.create_user(email=email, password=password)
-        st.success("Registration successful! Please log in.")
+        
+        # Send email verification
+        auth.generate_email_verification_link(email)
+        st.success("Registration successful! Please check your email to verify your account.")
     except Exception as e:
         st.error(f"Registration failed: {e}")
 
@@ -38,9 +41,14 @@ def login_callback():
     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!")
+        
+        # Check if email is verified
+        if user.email_verified:
+            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!")
+        else:
+            st.error("Please verify your email before logging in.")
     except Exception as e:
         st.error(f"Login failed: {e}")