Update app.py

app.py

@@ -861,6 +861,7 @@ def check_password(password, hashed):
     return bcrypt.checkpw(password.encode(), hashed)
 
 # User signup and login functions
+
 def signup_user(email, password):
     conn = sqlite3.connect('user_data.db')
     c = conn.cursor()
@@ -873,23 +874,29 @@ def signup_user(email, password):
     finally:
         conn.close()
 
-def login_user(email, password):
+def login_user(email, password, state):
     conn = sqlite3.connect('user_data.db')
     c = conn.cursor()
     c.execute("SELECT password FROM users WHERE email = ?", (email,))
     result = c.fetchone()
     conn.close()
     if result and check_password(password, result[0]):
-        return "Login successful!"
+        state['authenticated'] = True
+        return "Login successful!", state
     else:
-        return "Invalid email or password!"
+        return "Invalid email or password!", state
+
+def logout(state):
+    message, state = logout_user(state)
+    return message, state, "Login"
 
-# Functions for Gradio interface
+# Define the necessary functions
 def signup(email, password):
     return signup_user(email, password)
 
-def login(email, password):
-    return login_user(email, password)
+def login(email, password, state):
+    message, state = login_user(email, password, state)
+    return message, state, "Logout" if state['authenticated'] else "Login"
 
 # Initialize database
 initialize_database()
@@ -899,24 +906,27 @@ initialize_database()
 
 
 
-# Gradio interface
-with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
+ith gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
+    state = gr.State({'authenticated': False})
+
     with gr.Row():
         with gr.Column():
             gr.Markdown("<h1>Welcome to the App</h1>")
+            
             with gr.Tab("Sign Up"):
-                signup_email = gr.Textbox(placeholder="Email", label="Email")
+                signup_email = gr.Textbox(placeholder="Email", label="Email", type="email")
                 signup_password = gr.Textbox(placeholder="Password", label="Password", type="password")
                 signup_button = gr.Button("Sign Up")
                 signup_output = gr.Textbox(interactive=False, label="Output")
                 signup_button.click(signup, inputs=[signup_email, signup_password], outputs=signup_output)
 
             with gr.Tab("Login"):
-                login_email = gr.Textbox(placeholder="Email", label="Email")
+                login_email = gr.Textbox(placeholder="Email", label="Email", type="email")
                 login_password = gr.Textbox(placeholder="Password", label="Password", type="password")
                 login_button = gr.Button("Login")
                 login_output = gr.Textbox(interactive=False, label="Output")
-                login_button.click(login, inputs=[login_email, login_password], outputs=login_output)
+                login_button.click(login, inputs=[login_email, login_password, state], outputs=[login_output, state, login_button])
+                login_button.click(logout, inputs=[state], outputs=[login_output, state, login_button], show_progress=False)
             
             state = gr.State()
             chatbot = gr.Chatbot([], elem_id="RADAR:Channel 94.1", bubble_full_width=False)