Update app.py

app.py

@@ -105,30 +105,13 @@ def register_callback():
     except Exception as e:
         st.error(f"Registration failed: {e}")
 
-# Callback for login
-def login_callback():
-    login_identifier = st.session_state.login_identifier
-    password = st.session_state.login_password
-    try:
-        # Try to get the user by email first
-        try:
-          user = auth.get_user_by_email(login_identifier)
-        except:
-            # If getting by email fails, try to get the user by display name
-            user_list = auth.list_users()
-            for user_info in user_list.users:
-                if user_info.display_name == login_identifier:
-                    user = user_info
-                    break
-            else:
-                raise Exception("User not found")
-       
-        st.session_state.logged_in = True
-        st.session_state.current_user = user.uid
-        st.session_state.display_name = user.display_name # Store the display name
-        st.success("Logged in successfully!")
-    except Exception as 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.session_state.display_name = None
+    st.session_state.selected_image = None
+    st.info("Logged out successfully!")
 
 # Callback for login
 def login_callback():