Update app.py

app.py

@@ -353,40 +353,34 @@ def authenticate_user_ui():
 
     with c32:
         st.title("Insta's EYE")
-        # Use session_state to keep track of the selected tab
-        if "auth_tab" not in st.session_state:
-            st.session_state.auth_tab = None  # Default to None
-        
-        tabs = st.radio("Select Option", ["Login", "Sign-Up"], st.session_state.auth_tab)
-        st.session_state.auth_tab = tabs
-        
-        if tabs == "Login":
-            if st.session_state.auth_state["signed_in"]:
-                st.sidebar.button("Sign Out", on_click=logout)
-                st.title("Welcome!")
-                main()
-            else:
-                email = st.text_input("Enter Email", help="Enter your email address")
-                password = st.text_input("Enter Password", type="password", help="Enter your password")
-                
-                if st.button("Login"):
-                    if not email or not password:
-                        st.error("Please enter both email and password.")
+        st.sidebar.title("Options")
+
+    if st.session_state.auth_state["signed_in"]:
+        st.sidebar.button("Sign Out", on_click=logout)
+        st.title("Welcome!")
+        main()
+    else:
+        option = st.sidebar.radio("Select Option", ["Login", "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")
+
+        if option == "Login":
+            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:
-                        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.")
-
-        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")
+                        st.error("Authentication failed. Please check your email and password.")
+
+        elif option == "Sign-Up":
             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.")
@@ -398,7 +392,7 @@ def authenticate_user_ui():
                         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