Update app.py

app.py

@@ -6,7 +6,7 @@ import json
 
 # Load Firebase credentials from Hugging Face Secrets
 firebase_creds = os.getenv("FIREBASE_CREDENTIALS")
-
+FIREBASE_API_KEY = os.getenv("FIREBASE_API_KEY")
 if firebase_creds:
     firebase_creds = json.loads(firebase_creds)
 else:
@@ -25,14 +25,53 @@ if "logged_in" not in st.session_state:
 if "current_user" not in st.session_state:
     st.session_state.current_user = None
 
+def send_verification_email(id_token):
+    url = f'https://identitytoolkit.googleapis.com/v1/accounts:sendOobCode?key={FIREBASE_API_KEY}'
+    headers = {'Content-Type': 'application/json'}
+    data = {
+        'requestType': 'VERIFY_EMAIL',
+        'idToken': id_token
+    }
+    
+    response = requests.post(url, headers=headers, json=data)
+    result = response.json()
+    
+    if 'error' in result:
+        return {'status': 'error', 'message': result['error']['message']}
+    else:
+        return {'status': 'success', 'email': result['email']}
+        
 # Callback for registration
 def register_callback():
     email = st.session_state.reg_email
     password = st.session_state.reg_password
     try:
-        # Create a new user in Firebase
+        # Step 1: Create a new user in Firebase
         user = auth.create_user(email=email, password=password)
-        st.success("Registration successful! Please log in.")
+        st.success("Registration successful! Sending verification email...")
+
+        # Step 2: Sign in the user programmatically to get the id_token
+        url = f'https://identitytoolkit.googleapis.com/v1/accounts:signInWithPassword?key={FIREBASE_API_KEY}'
+        data = {
+            'email': email,
+            'password': password,
+            'returnSecureToken': True
+        }
+        response = requests.post(url, json=data)
+        result = response.json()
+
+        if 'idToken' in result:
+            id_token = result['idToken']  # Retrieve the id_token
+            st.session_state.id_token = id_token  # Store the id_token in session state
+
+            # Step 3: Send the verification email
+            verification_result = send_verification_email(id_token)
+            if verification_result['status'] == 'success':
+                st.success(f"Verification email sent to {email}.")
+            else:
+                st.error(f"Failed to send verification email: {verification_result['message']}")
+        else:
+            st.error(f"Failed to retrieve id_token: {result['error']['message']}")
     except Exception as e:
         st.error(f"Registration failed: {e}")