Update database.py

database.py

@@ -269,11 +269,17 @@ 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')
+        ])
+        
         # Insert user
         cursor = db.execute(
-            '''INSERT INTO users (email, password_hash, salt, confirmation_token) 
-               VALUES (?, ?, ?, ?)''',
-            (email, password_hash, salt, confirmation_token)
+            '''INSERT INTO users (email, password_hash, salt, confirmation_token, is_confirmed) 
+               VALUES (?, ?, ?, ?, ?)''',
+            (email, password_hash, salt, confirmation_token, auto_confirm)
         )
         user_id = cursor.lastrowid
         
@@ -285,12 +291,17 @@ def create_user(email, password):
         
         db.commit()
         
-        logger.info(f"User created: {email}")
+        if auto_confirm:
+            logger.info(f"User created and auto-confirmed: {email} (demo mode)")
+        else:
+            logger.info(f"User created: {email}")
+            
         return {
             'success': True, 
             'user_id': user_id,
             'confirmation_token': confirmation_token,
-            'message': 'User created successfully'
+            'message': 'User created successfully',
+            'auto_confirmed': auto_confirm
         }
         
     except Exception as e:
@@ -480,7 +491,7 @@ def get_current_user():
 
 # Email functionality (for Hugging Face Spaces)
 def send_confirmation_email(email, token):
-    """Send confirmation email (simplified for HF Spaces)"""
+    """Send confirmation email (simplified for HF Spaces with timeout)"""
     try:
         # For Hugging Face Spaces, we'll use environment variables for SMTP
         smtp_server = os.environ.get('SMTP_SERVER', 'smtp.gmail.com')
@@ -489,7 +500,7 @@ def send_confirmation_email(email, token):
         smtp_password = os.environ.get('SMTP_PASSWORD')
         
         if not all([smtp_username, smtp_password]):
-            logger.warning("SMTP credentials not configured")
+            logger.warning("SMTP credentials not configured - skipping email")
             return False
         
         # Create confirmation URL (will be updated with actual domain)
@@ -518,7 +529,12 @@ def send_confirmation_email(email, token):
         
         msg.attach(MIMEText(body, 'html'))
         
-        # Send email
+        # Send email with timeout
+        import socket
+        
+        # Set socket timeout to prevent hanging
+        socket.setdefaulttimeout(10)
+        
         server = smtplib.SMTP(smtp_server, smtp_port)
         server.starttls()
         server.login(smtp_username, smtp_password)
@@ -526,11 +542,20 @@ def send_confirmation_email(email, token):
         server.sendmail(smtp_username, email, text)
         server.quit()
         
+        # Reset socket timeout
+        socket.setdefaulttimeout(None)
+        
         logger.info(f"Confirmation email sent to {email}")
         return True
         
     except Exception as e:
         logger.error(f"Error sending confirmation email: {e}")
+        # Reset socket timeout on error
+        try:
+            import socket
+            socket.setdefaulttimeout(None)
+        except:
+            pass
         return False
 
 # --- CONTENT CURATION FUNCTIONS ---