Update database.py

database.py

@@ -269,11 +269,19 @@ def create_user(email, password):
         password_hash, salt = hash_password(password)
         confirmation_token = generate_token()
         
-        # For HF Spaces demo mode, auto-confirm emails if SMTP not configured
-        auto_confirm = not all([
-            os.environ.get('SMTP_USERNAME'), 
-            os.environ.get('SMTP_PASSWORD')
-        ])
+        # For HF Spaces demo mode, auto-confirm emails
+        is_hf_spaces = os.environ.get('SPACE_ID') is not None
+        has_smtp_username = os.environ.get('SMTP_USERNAME') is not None
+        has_smtp_password = os.environ.get('SMTP_PASSWORD') is not None
+        has_smtp = has_smtp_username and has_smtp_password
+        
+        # Sempre auto-confirmar no HF Spaces ou quando SMTP não está configurado
+        auto_confirm = is_hf_spaces or not has_smtp
+        
+        # Debug logging
+        logger.info(f"Registration debug - SPACE_ID: {os.environ.get('SPACE_ID')}")
+        logger.info(f"HF_Spaces: {is_hf_spaces}, SMTP_USER: {has_smtp_username}, SMTP_PASS: {has_smtp_password}")
+        logger.info(f"SMTP configured: {has_smtp}, Auto-confirm: {auto_confirm}")
         
         # Insert user
         cursor = db.execute(
@@ -292,15 +300,17 @@ def create_user(email, password):
         db.commit()
         
         if auto_confirm:
-            logger.info(f"User created and auto-confirmed: {email} (demo mode)")
+            logger.info(f"User created and auto-confirmed: {email} (HF Spaces demo mode)")
+            message = 'Account created and ready to use! (Demo mode - no email confirmation needed)'
         else:
             logger.info(f"User created: {email}")
+            message = 'User created successfully. Please check your email for confirmation.'
             
         return {
             'success': True, 
             'user_id': user_id,
             'confirmation_token': confirmation_token,
-            'message': 'User created successfully',
+            'message': message,
             'auto_confirmed': auto_confirm
         }