Update app.py

app.py

@@ -13,35 +13,41 @@ if firebase_creds:
 else:
     st.error("Firebase credentials not found. Please check your secrets.")
 
-# Initialize Firebase (only once)
-if not firebase_admin._apps and firebase_creds:
-    cred = credentials.Certificate(firebase_creds)
-    firebase_admin.initialize_app(cred)
-    st.success("Firebase initialized successfully!")
-else:
-    st.error("Failed to initialize Firebase.")
+# Initialize session state
+if "logged_in" not in st.session_state:
+    st.session_state.logged_in = False
+if "current_user" not in st.session_state:
+    st.session_state.current_user = None
 
-# Function to send verification email
-def send_verification_email(email):
+# Callback for registration
+def register_callback():
+    email = st.session_state.reg_email
+    password = st.session_state.reg_password
     try:
-        # Generate email verification link
-        verification_link = auth.generate_email_verification_link(email)
-        
-        # Send email using SMTP (replace with your email service)
-        msg = MIMEText(f"Please verify your email by clicking this link: {verification_link}")
-        msg["Subject"] = "Verify Your Email"
-        msg["From"] = "[email protected]"
-        msg["To"] = email
+        # Create a new user in Firebase
+        user = auth.create_user(email=email, password=password)
+        st.success("Registration successful! Please log in.")
+    except Exception as e:
+        st.error(f"Registration failed: {e}")
 
-        # Replace with your SMTP server details
-        with smtplib.SMTP("smtp.example.com", 587) as server:
-            server.starttls()
-            server.login("your-email@example.com", "your-email-password")
-            server.sendmail("[email protected]", [email], msg.as_string())
-        
-        st.success("Verification email sent successfully!")
+# Callback for login
+def login_callback():
+    email = st.session_state.login_email
+    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!")
     except Exception as e:
-        st.error(f"Failed to send verification email: {e}")
+        st.error(f"Login failed: {e}")
+
+# Callback for logout
+def logout_callback():
+    st.session_state.logged_in = False
+    st.session_state.current_user = None
+    st.info("Logged out successfully!")
 
 # Registration form
 def registration_form():
@@ -49,18 +55,35 @@ def registration_form():
         st.subheader("Register")
         email = st.text_input("Email", key="reg_email")
         password = st.text_input("Password (min 6 characters)", type="password", key="reg_password")
-        submit_button = st.form_submit_button("Register")
+        submit_button = st.form_submit_button("Register", on_click=register_callback)
 
-        if submit_button:
-            try:
-                # Create a new user in Firebase
-                user = auth.create_user(email=email, password=password)
-                st.success("Registration successful! Please check your email for verification.")
+# Login form
+def login_form():
+    with st.form("Login"):
+        st.subheader("Login")
+        email = st.text_input("Email", key="login_email")
+        password = st.text_input("Password", type="password", key="login_password")
+        submit_button = st.form_submit_button("Login", on_click=login_callback)
 
-                # Send verification email
-                send_verification_email(email)
-            except Exception as e:
-                st.error(f"Registration failed: {e}")
+# Main app screen (after login)
+def main_app():
+    st.subheader(f"Welcome, {st.session_state.current_user}!")
+    st.write("Enter a name below to get a greeting.")
 
-# Render the registration form
-registration_form()
+    # Input field for name
+    name = st.text_input("Enter a name", key="name_input")
+
+    # Display greeting in real-time
+    if name:
+        st.write(f"Hello {name}!")
+
+    # Logout button
+    if st.button("Logout", on_click=logout_callback):
+        pass
+
+# Render the appropriate screen based on login status
+if st.session_state.logged_in:
+    main_app()
+else:
+    registration_form()
+    login_form()