Update app.py

app.py

@@ -394,6 +394,37 @@ def authenticate_user_ui():
                     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.")
+                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.")
+
 # Log out user
 def logout():
     st.session_state.auth_state["user"] = None