Update ui.py

ui.py

@@ -1,11 +1,7 @@
 import streamlit as st 
 import requests
 import os 
-from login import LOGIN 
-from register import REGISTER 
-
-
-
+from auth import authenticator
 
 
 
@@ -22,12 +18,14 @@ 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
+    #if "loggedin" not in st.session_state:
+     #   st.session_state.loggedin=False
         
 
 InitSession()
 
+name, auth_status, username = authenticator.login('Login', 'main')
+
 
 def APP():
 
@@ -195,16 +193,20 @@ def APP():
                 selection_mode="single",
                 )
         if selection==0:
-            logged_in_object=LOGIN()
             
-            if logged_in_object.get("loggedin")==True :
+            
+            if auth_status:
                 
         
                 SHOWTABS()
 
-            elif logged_in_object.get("loggedin")==False:
+            elif auth_status==False:
                  st.write("login failed , try again")
 
         if selection==1:
-            REGISTER()
+            try:
+                if authenticator.register_user('Register'):
+                    st.success("User registered successfully!")
+            except Exception as e:
+                    st.error(e)