Update components/auth.py

components/auth.py

@@ -1,20 +1,41 @@
-from utils.db import sign_up_user, login_user
+# components/auth.py
 
-user_session = {"email": None}
+from utils.db import sign_up_user, login_user
+import gradio as gr
 
 def register(email, password):
+    """Handler for the signup process."""
+    if not email or not password:
+        return "❌ Email and password cannot be empty."
+    
     result = sign_up_user(email, password)
-    if result.user:
-        return "✅ Signup successful!"
-    return "❌ Signup failed."
 
-def login(email, password):
+    if result and result.user:
+        # Check for the common case where the user is created but needs to confirm email
+        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."
+    
+    return "❌ Signup failed. The user may already exist or the password is too weak."
+
+
+def login(email, password, current_user_state):
+    """Handler for the login process."""
+    if not email or not password:
+        return "❌ Email and password cannot be empty.", current_user_state
+
     result = login_user(email, password)
-    if result.session:
-        user_session["email"] = email
-        return f"✅ Logged in as {email}"
-    return "❌ Login failed. Check your credentials."
-
-def logout():
-    user_session["email"] = None
-    return "👋 Logged out."
+    
+    if result and result.session:
+        # Update the user state
+        new_user_state = {"email": email, "logged_in": True}
+        return f"✅ Logged in as {email}", new_user_state
+    
+    return "❌ Login failed. Check your credentials or confirm your email.", current_user_state
+
+
+def logout(current_user_state):
+    """Handler for the logout process."""
+    # Reset the user state
+    new_user_state = {"email": None, "logged_in": False}
+    return "👋 Logged out successfully.", new_user_state