Update components/auth.py

components/auth.py

@@ -4,38 +4,40 @@ from utils.db import sign_up_user, login_user
 import gradio as gr
 
 def register(email, password):
-    """Handler for the signup process with detailed error reporting."""
+    """
+    Handler for the signup process.
+    On success, it will now also return empty strings to clear the input fields.
+    """
     if not email or not password:
-        return "❌ Email and password cannot be empty."
+        # Return a no-op update for the textboxes on failure
+        return "❌ Email and password cannot be empty.", gr.update(), gr.update()
     
     result = sign_up_user(email, password)
 
-    if result is None:
-        return "❌ Signup failed due to a server connection error. Check logs."
-
-    if result.user:
+    if result and result.user:
+        message = "✅ Signup successful! You can now log in."
         if result.user and not result.session:
-             return "✅ Signup successful! Please check your email to confirm your account."
-        return "✅ Signup successful! You can now log in."
+             message = "✅ Signup successful! Please check your email to confirm your account."
+        # On success, clear the signup email and password fields by returning empty strings
+        return message, "", "" 
     
     if hasattr(result, 'error') and result.error:
-        return f"❌ Signup Failed: {result.error.message}"
+        # Return a no-op update for the textboxes on failure
+        return f"❌ Signup Failed: {result.error.message}", gr.update(), gr.update()
     
-    return "❌ Signup failed. An unknown error occurred."
+    # Return a no-op update for the textboxes on failure
+    return "❌ Signup failed. An unknown error occurred.", gr.update(), gr.update()
 
 
 def login(email, password, current_user_state):
-    """
-    Handler for the login process.
-    On success, returns a command to switch the UI to the 'prediction_tab'.
-    """
+    """Handler for the login process."""
     if not email or not password:
         return "❌ Email and password cannot be empty.", current_user_state, gr.update()
 
     result = login_user(email, password)
     
     if result and result.session:
-        new_user_state = {"email": email, "logged_in": True}
+        new_user_state = {"email": email, "id": result.user.id, "logged_in": True}
         return f"✅ Login successful! Redirecting...", new_user_state, gr.Tabs(selected="prediction_tab")
     
     if hasattr(result, 'error') and result.error:
@@ -47,8 +49,22 @@ def login(email, password, current_user_state):
 def logout(current_user_state):
     """
     Handler for the logout process.
-    On success, returns a command to switch the UI to the 'home_tab'.
+    Now also returns values to clear the login and prediction forms.
     """
-    new_user_state = {"email": None, "logged_in": False}
-    # On logout, redirect to the home tab.
-    return "👋 Logged out successfully.", new_user_state, gr.Tabs(selected="home_tab")
+    new_user_state = {"email": None, "id": None, "logged_in": False}
+    
+    # Message, new state, redirect, then empty values for all fields to be cleared
+    # The order of these return values MUST match the 'outputs' list in app.py
+    return (
+        "👋 Logged out successfully.",
+        new_user_state,
+        gr.Tabs(selected="home_tab"),
+        "",          # Clear login email textbox
+        "",          # Clear login password textbox
+        None,        # Clear pregnancies number input
+        None,        # Clear glucose number input
+        None,        # Clear blood pressure number input
+        None,        # Clear insulin number input
+        None,        # Clear BMI number input
+        None         # Clear age number input
+    )