Update ui.py

ui.py

@@ -20,7 +20,7 @@ def InitSession():
      #   st.session_state.loggedin=False
         
 
-InitSession()
+#InitSession()
 
 
 
@@ -180,11 +180,21 @@ def APP():
 
 
 
-        
-        if st.button("Log in"):
-             st.login("auth0")
+
+    
+    
         if st.user.is_logged_in:
-            st.logout()
+            
+            if st.button("🚪 Logout"):
+                st.logout()
+                st.rerun()
+    
+            #Then show the main UI
             SHOWTABS()
-       
+        else:
+            #Show login when NOT logged in
+            st.info("Please log in to access the Bio Lab")
+            if st.button("Log in"):
+                st.login("auth0")
+                st.stop()