Update app.py

app.py

@@ -376,54 +376,28 @@ def authenticate_user_ui():
                             main()
                         else:
                             st.error("Authentication failed. Please check your email and password.")
-
+        
         elif tabs == "Sign-Up":
-            email = st.text_input("Enter Email", help="Enter your email address")
-            password = st.text_input("Enter Password", type="password", help="Enter your password")
-            confirm_password = st.text_input("Confirm Password", type="password", help="Re-enter your password for confirmation")
-            
-            if st.button("Sign-Up"):
-                if not email or not password or not confirm_password:
-                    st.error("Please fill all the fields.")
-                elif password != confirm_password:
-                    st.error("Passwords do not match.")
-                else:
-                    success, uid = create_user(email, password)
-                    if success:
-                        st.success(f"User with UID: {uid} created successfully! You can now log in.")
-                    else:
-                        st.error("User creation failed. Please try again.")
-
-    # Move the form rendering code outside the if blocks
-    # This ensures that the forms are always displayed
-    email = st.text_input("Enter Email", help="Enter your email address")
-    password = st.text_input("Enter Password", type="password", help="Enter your password")
-    confirm_password = st.text_input("Confirm Password", type="password", help="Re-enter your password for confirmation")
-
-    if st.button("Login"):
-        if not email or not password:
-            st.error("Please enter both email and password.")
-        else:
-            success, user = authenticate_user(email, password)
-            if success:
-                st.session_state.auth_state["user"] = user
-                st.session_state.auth_state["signed_in"] = True
-                st.success("Authentication successful! You can now manage your set of images and profiles.")
+            if st.session_state.auth_state["signed_in"]:
+                st.sidebar.button("Sign Out", on_click=logout)
+                st.title("Welcome!")
                 main()
             else:
-                st.error("Authentication failed. Please check your email and password.")
-
-    if st.button("Sign-Up"):
-        if not email or not password or not confirm_password:
-            st.error("Please fill all the fields.")
-        elif password != confirm_password:
-            st.error("Passwords do not match.")
-        else:
-            success, uid = create_user(email, password)
-            if success:
-                st.success(f"User with UID: {uid} created successfully! You can now log in.")
-            else:
-                st.error("User creation failed. Please try again.")
+                email = st.text_input("Enter Email", help="Enter your email address")
+                password = st.text_input("Enter Password", type="password", help="Enter your password")
+                confirm_password = st.text_input("Confirm Password", type="password", help="Re-enter your password for confirmation")
+                
+                if st.button("Sign-Up"):
+                    if not email or not password or not confirm_password:
+                        st.error("Please fill all the fields.")
+                    elif password != confirm_password:
+                        st.error("Passwords do not match.")
+                    else:
+                        success, uid = create_user(email, password)
+                        if success:
+                            st.success(f"User with UID: {uid} created successfully! You can now log in.")
+                        else:
+                            st.error("User creation failed. Please try again.")
 
 # Log out user
 def logout():