Update ui.py

ui.py

@@ -1,6 +1,13 @@
 import streamlit as st 
 import requests
 import os 
+from login import LOGIN 
+from register import REGISTER 
+
+
+
+
+
 
 tok=os.getenv("TOK")
 
@@ -15,6 +22,9 @@ def InitSession():
     if "projectname" not in st.session_state:
         st.session_state.projectname=""
 
+    if "loggedin" not in st.session_state:
+        st.session_state.loggedin=False
+        
 
 InitSession()
 
@@ -34,7 +44,7 @@ def APP():
     
         tab1, tab2, tab3 = st.tabs(["PROTEIN ENGINEERING LAB", "EXECUTED OPERATIONS", "LAB OUTPUT"])
     
-    
+        
         def SHOWTABS():
             
             
@@ -168,5 +178,33 @@ def APP():
                     ie=response.json()
                     st.json(ie)
 
+
+
+
+        base_option_map = {
+                    
+                    0: ":material/login:",
+                    1: ":material/personadd:",
+    
+                    }
+
+        selection = st.pills(
+                "ACCESS",
+                options=option_map.keys(),
+                format_func=lambda option: option_map[option],
+                selection_mode="single",
+                )
+        if selection==0:
+            logged_in_object=LOGIN()
             
-        SHOWTABS()
+            if logged_in_object.get("loggedin")==True :
+                
+        
+                SHOWTABS()
+
+            elif logged_in_object.get("loggedin")==False:
+                 st.write("login failed , try again")
+
+        if selection==1:
+            REGISTER()
+