Dear Dr. {user.full_name},
Congratulations! Your NeuroSight account has been successfully created and verified.
We're excited to have you join our community of healthcare professionals using AI-powered brain disease detection technology.
š Your Registration Details
Full Name:
{user.full_name}
Email:
{user.email}
Medical Reg. No:
{user.medical_registration_no or 'N/A'}
Role:
{user.specialization or 'N/A'}
Years of Experience:
{user.years_of_experience or 'N/A'} years
Hospital/Clinic:
{user.hospital or 'N/A'}
Department:
{user.department or 'N/A'}
Registration Date:
{user.created_at.strftime('%B %d, %Y at %I:%M %p')}
š What's Next?
- Upload Brain Scans: Start analyzing MRI scans for various brain diseases
- View Analysis History: Track all your previous analyses
- Generate Reports: Download professional PDF reports for your patients
- Manage Profile: Update your professional details anytime
Go to Dashboard
š” Quick Tips
- Ensure MRI scans are in supported formats (JPG, PNG, JPEG)
- For best results, use high-quality brain scan images
- Review the confidence scores provided with each analysis
- Keep your professional credentials up to date
Need Help? If you have any questions or need assistance, please don't hesitate to contact our support team.
Thank you for choosing NeuroSight. We're committed to supporting you in providing the best possible care for your patients.
Best regards,
The NeuroSight Team
"""
# Plain text version as fallback
msg.body = f"""
Welcome to NeuroSight!
Dear Dr. {user.full_name},
Congratulations! Your NeuroSight account has been successfully created and verified.
Your Registration Details:
- Full Name: {user.full_name}
- Email: {user.email}
- Medical Registration No: {user.medical_registration_no or 'N/A'}
- Role: {user.specialization or 'N/A'}
- Years of Experience: {user.years_of_experience or 'N/A'} years
- Hospital/Clinic: {user.hospital or 'N/A'}
- Department: {user.department or 'N/A'}
- Registration Date: {user.created_at.strftime('%B %d, %Y at %I:%M %p')}
What's Next?
- Upload Brain Scans for analysis
- View your analysis history
- Generate professional PDF reports
- Manage your profile
Visit your dashboard: http://localhost:5000/dashboard
Thank you for choosing NeuroSight!
Best regards,
The NeuroSight Team
"""
mail.send(msg)
print(f"ā Welcome email sent to {user.email}")
return True
except Exception as e:
print(f"ā Failed to send welcome email to {user.email}: {str(e)}")
return False
def send_otp_email(user, otp_code):
"""Send OTP email for email verification"""
try:
msg = Message(
subject='NeuroSight - Email Verification Code',
recipients=[user.email]
)
# Create HTML email body
msg.html = f"""
Hello {user.full_name},
Thank you for registering with NeuroSight! To complete your registration, please verify your email address using the code below:
YOUR VERIFICATION CODE
{otp_code}
Valid for 10 minutes
How to verify:
- Return to the registration page
- Enter the 6-digit code above
- Click "Verify Email"
ā ļø Security Notice:
- Never share this code with anyone
- NeuroSight will never ask for this code via phone or email
- This code expires in 10 minutes
- You have 5 attempts to enter the correct code
If you didn't request this code, please ignore this email or contact our support team if you have concerns.
Best regards,
The NeuroSight Team
"""
# Plain text version as fallback
msg.body = f"""
NeuroSight - Email Verification
Hello {user.full_name},
Thank you for registering with NeuroSight!
Your verification code is: {otp_code}
This code is valid for 10 minutes.
How to verify:
1. Return to the registration page
2. Enter the 6-digit code
3. Click "Verify Email"
Security Notice:
- Never share this code with anyone
- This code expires in 10 minutes
- You have 5 attempts to enter the correct code
If you didn't request this code, please ignore this email.
Best regards,
The NeuroSight Team
"""
mail.send(msg)
print(f"ā OTP email sent to {user.email}")
return True
except Exception as e:
print(f"ā Failed to send OTP email to {user.email}: {str(e)}")
return False
# Configuration
UPLOAD_FOLDER = os.path.join(app.static_folder, "uploads")
REPORTS_FOLDER = os.path.join(app.static_folder, "reports")
os.makedirs(UPLOAD_FOLDER, exist_ok=True)
os.makedirs(REPORTS_FOLDER, exist_ok=True)
# Load feature extractor
print("Loading ViT feature extractor...")
feature_extractor = ViTFeatureExtractor.from_pretrained('google/vit-base-patch16-224-in21k')
# Disease configurations
DISEASE_CONFIG = {
'ms': {
'name': 'Multiple Sclerosis',
'model_path': 'multiple_sclerosis.pth',
'class_mapping': {0: 'Control-Axial', 1: 'Control-Sagittal', 2: 'MS-Axial', 3: 'MS-Sagittal'}
},
'alzheimer': {
'name': "Alzheimer's Disease",
'model_path': 'alzhimermodel.pth',
'class_mapping': {0: 'Mild-alzhimer', 1: 'Moderate-alzhimer', 2: 'Non-alzhimer', 3: 'VeryMild-alzhimer'}
},
'dementia': {
'name': 'Dementia',
'model_path': 'dementia_detection_model_2.h5',
'class_mapping': {0: 'Non-Demented', 1: 'Very-Mild-Demented', 2: 'Mild-Demented', 3: 'Moderate-Demented'}
},
'stroke': {
'name': 'Stroke',
'model_path': 'stroke.pth',
'class_mapping': {0: 'Normal š', 1: 'Stroke š'}
}
}
def map_convnext_keys(state_dict):
new_dict = {}
for k, v in state_dict.items():
new_k = k
# Map stem
if k.startswith('stem.0'):
new_k = k.replace('stem.0', 'convnext.embeddings.patch_embeddings')
elif k.startswith('stem.1'):
new_k = k.replace('stem.1', 'convnext.embeddings.layernorm')
# Map stages
elif k.startswith('stages'):
# stages.0.blocks.0 -> convnext.encoder.stages.0.layers.0
parts = k.split('.')
stage_idx = parts[1]
block_idx = parts[3]
rest = '.'.join(parts[4:])
prefix = f'convnext.encoder.stages.{stage_idx}.layers.{block_idx}'
if 'gamma' in rest:
new_k = f'{prefix}.layer_scale_parameter'
elif 'conv_dw' in rest:
new_k = f'{prefix}.dwconv.{rest.replace("conv_dw.", "")}'
elif 'norm' in rest:
new_k = f'{prefix}.layernorm.{rest.replace("norm.", "")}'
elif 'mlp.fc1' in rest:
new_k = f'{prefix}.pwconv1.{rest.replace("mlp.fc1.", "")}'
elif 'mlp.fc2' in rest:
new_k = f'{prefix}.pwconv2.{rest.replace("mlp.fc2.", "")}'
elif 'downsample' in k:
# stages.0.downsample.0 -> convnext.encoder.stages.0.downsampling_layer.0
ds_idx = parts[3]
rest_ds = '.'.join(parts[4:])
new_k = f'convnext.encoder.stages.{stage_idx}.downsampling_layer.{ds_idx}.{rest_ds}'
# Map head
elif k.startswith('head'):
if 'fc' in k:
new_k = k.replace('head.fc', 'classifier')
elif 'norm' in k:
new_k = k.replace('head.norm', 'convnext.layernorm')
new_dict[new_k] = v
return new_dict
def load_model(model_path, num_labels=4):
"""Load PyTorch (ViT/ConvNeXt) or TensorFlow/Keras (EfficientNet) model"""
try:
full_path = os.path.join(os.getcwd(), model_path)
# Check if this is a Keras/TensorFlow model (.h5)
if model_path.lower().endswith('.h5'):
print(f" Loading Keras model for {model_path}...")
model = keras.models.load_model(full_path)
model.model_type = 'keras' # Tag for later use
print(f"ā Loaded Keras model: {model_path}")
return model
# PyTorch models
if 'stroke' in model_path:
# Stroke model is a ConvNeXt (Base) with 1 output (binary)
print(f" Loading ConvNeXt for {model_path}...")
try:
model = ConvNextForImageClassification.from_pretrained(
'facebook/convnext-base-224-22k-1k',
num_labels=num_labels,
ignore_mismatched_sizes=True
)
# Load and map weights
state_dict = torch.load(full_path, map_location=torch.device('cpu'))
new_state_dict = map_convnext_keys(state_dict)
model.load_state_dict(new_state_dict)
model.model_type = 'pytorch'
except Exception as e:
print(f" Warning: Could not load ConvNeXt config: {e}")
return None
else:
# ViT models
model = ViTForImageClassification.from_pretrained('google/vit-base-patch16-224-in21k', num_labels=num_labels)
state_dict = torch.load(full_path, map_location=torch.device('cpu'))
model.load_state_dict(state_dict)
model.model_type = 'pytorch'
model.eval()
print(f"ā Loaded model: {model_path}")
return model
except Exception as e:
print(f"ā Could not load model {model_path}: {str(e)}")
return None
# Load all available models
print("\nLoading disease detection models...")
models = {}
for disease_key, config in DISEASE_CONFIG.items():
# Stroke model has 1 output node (binary), others have 4
num_labels = 1 if disease_key == 'stroke' else 4
model = load_model(config['model_path'], num_labels=num_labels)
if model:
models[disease_key] = model
print(f"\nLoaded {len(models)} out of {len(DISEASE_CONFIG)} models successfully.\n")
# ============ HELPER FUNCTIONS ============
def get_google_provider_cfg():
"""Fetch Google's OAuth 2.0 provider configuration"""
try:
return requests.get(app.config['GOOGLE_DISCOVERY_URL']).json()
except:
return None
def send_reset_email(user):
"""Send password reset email to user"""
try:
# Generate reset token (expires in 1 hour)
token = serializer.dumps(user.email, salt='password-reset-salt')
# Create reset URL
reset_url = url_for('reset_password', token=token, _external=True)
# Create email message
msg = Message(
subject='NeuroSight - Password Reset Request',
recipients=[user.email],
sender=app.config['MAIL_DEFAULT_SENDER']
)
# Email body
msg.html = f"""
š§ NeuroSight
AI-Powered Brain Disease Detection
Password Reset Request
Hello {user.full_name},
We received a request to reset your password for your NeuroSight account.
Click the button below to reset your password:
Or copy and paste this link into your browser:
{reset_url}
This link will expire in 1 hour.
If you didn't request this password reset, please ignore this email.
NeuroSight - Rajalakshmi Engineering College
Contact: asuproject0112@gmail.com
"""
mail.send(msg)
return True
except Exception as e:
print(f"Error sending email: {e}")
return False
def verify_reset_token(token, expiration=3600):
"""Verify password reset token (default 1 hour expiration)"""
try:
email = serializer.loads(token, salt='password-reset-salt', max_age=expiration)
return email
except (SignatureExpired, BadSignature):
return None
# ============ AUTHENTICATION ROUTES ============
@app.route('/register', methods=['GET', 'POST'])
def register():
"""User registration"""
if current_user.is_authenticated:
return redirect(url_for('dashboard'))
if request.method == 'POST':
email = request.form.get('email', '').strip()
password = request.form.get('password', '')
confirm_password = request.form.get('confirm_password', '')
full_name = request.form.get('full_name', '').strip()
role = request.form.get('role', '').strip()
hospital = request.form.get('hospital', '').strip()
license_number = request.form.get('license_number', '').strip()
phone = request.form.get('phone', '').strip()
# Validation
if not all([email, password, full_name, role]):
flash('Please fill in all required fields.', 'danger')
return render_template('register.html')
if not validate_email(email):
flash('Invalid email address.', 'danger')
return render_template('register.html')
if password != confirm_password:
flash('Passwords do not match.', 'danger')
return render_template('register.html')
is_valid, message = validate_password(password)
if not is_valid:
flash(message, 'danger')
return render_template('register.html')
# Check if user already exists
if User.query.filter_by(email=email).first():
flash('Email already registered. Please login.', 'warning')
return redirect(url_for('login'))
# Create new user
new_user = User(
email=email,
full_name=full_name,
role=role,
hospital=hospital,
license_number=license_number,
phone=phone,
is_verified=True # Auto-verify for now
)
new_user.set_password(password)
db.session.add(new_user)
db.session.commit()
flash('Registration successful! Please login.', 'success')
return redirect(url_for('login'))
return render_template('register.html')
# ============================================================================
# OTP-BASED REGISTRATION API ENDPOINTS
# ============================================================================
@app.route('/api/register', methods=['POST'])
def api_register():
"""API endpoint for email/password registration with OTP verification"""
try:
data = request.get_json()
email = data.get('email', '').strip()
password = data.get('password', '')
full_name = data.get('full_name', '').strip()
# Validation
if not all([email, password, full_name]):
return {'success': False, 'error': 'Please fill in all required fields'}, 400
if not validate_email(email):
return {'success': False, 'error': 'Invalid email address'}, 400
is_valid, message = validate_password(password)
if not is_valid:
return {'success': False, 'error': message}, 400
# Check if user already exists
existing_user = User.query.filter_by(email=email).first()
if existing_user:
if existing_user.email_verified:
return {'success': False, 'error': 'Email already registered. Please login.'}, 400
else:
# User exists but not verified - resend OTP
otp_code = existing_user.generate_otp()
db.session.commit()
send_otp_email(existing_user, otp_code)
return {
'success': True,
'message': 'Verification email resent',
'user_id': existing_user.id,
'email': email
}, 200
# Create new user (not verified yet)
new_user = User(
email=email,
full_name=full_name,
role='doctor', # Default role, will be set during onboarding
email_verified=False,
is_active=True
)
new_user.set_password(password)
# Generate OTP
otp_code = new_user.generate_otp()
db.session.add(new_user)
db.session.commit()
# Send OTP email
send_otp_email(new_user, otp_code)
print(f"ā New user registered: {email}, OTP: {otp_code}")
return {
'success': True,
'message': 'Registration successful! Please check your email for verification code.',
'user_id': new_user.id,
'email': email
}, 200
except Exception as e:
db.session.rollback()
print(f"Registration error: {str(e)}")
import traceback
traceback.print_exc()
return {'success': False, 'error': f'Registration failed: {str(e)}'}, 500
@app.route('/api/verify-otp', methods=['POST'])
def api_verify_otp():
"""API endpoint to verify OTP code"""
try:
data = request.get_json()
user_id = data.get('user_id')
otp_code = data.get('otp_code', '').strip()
if not user_id or not otp_code:
return {'success': False, 'error': 'User ID and OTP code are required'}, 400
user = User.query.get(user_id)
if not user:
return {'success': False, 'error': 'User not found'}, 404
# Verify OTP
success, message = user.verify_otp(otp_code)
if success:
db.session.commit()
# Don't auto-login, redirect to login page
print(f"ā Email verified for: {user.email}")
return {
'success': True,
'message': 'Email verified successfully! Please login with your credentials.',
'redirect': url_for('login')
}, 200
else:
db.session.commit() # Save attempt count
return {'success': False, 'error': message}, 400
except Exception as e:
db.session.rollback()
print(f"OTP verification error: {str(e)}")
import traceback
traceback.print_exc()
return {'success': False, 'error': f'Verification failed: {str(e)}'}, 500
@app.route('/api/resend-otp', methods=['POST'])
def api_resend_otp():
"""API endpoint to resend OTP code"""
try:
data = request.get_json()
user_id = data.get('user_id')
if not user_id:
return {'success': False, 'error': 'User ID is required'}, 400
user = User.query.get(user_id)
if not user:
return {'success': False, 'error': 'User not found'}, 404
if user.email_verified:
return {'success': False, 'error': 'Email already verified'}, 400
# Generate new OTP
otp_code = user.generate_otp()
db.session.commit()
# Send email
send_otp_email(user, otp_code)
print(f"ā OTP resent to: {user.email}, New OTP: {otp_code}")
return {
'success': True,
'message': 'Verification code resent successfully!'
}, 200
except Exception as e:
db.session.rollback()
print(f"Resend OTP error: {str(e)}")
import traceback
traceback.print_exc()
return {'success': False, 'error': f'Failed to resend code: {str(e)}'}, 500
@app.route('/verify-email')
def verify_email():
"""Email verification page"""
if current_user.is_authenticated:
return redirect(url_for('dashboard'))
return render_template('verify_email.html')
@app.route('/login', methods=['GET', 'POST'])
def login():
"""User login"""
if current_user.is_authenticated:
return redirect(url_for('dashboard'))
if request.method == 'POST':
email = request.form.get('email', '').strip()
password = request.form.get('password', '')
remember = request.form.get('remember', False)
if not email or not password:
flash('Please enter both email and password.', 'danger')
return render_template('login.html')
user = User.query.filter_by(email=email).first()
if user and user.check_password(password):
if not user.is_active:
flash('Your account has been deactivated. Please contact support.', 'danger')
return render_template('login.html')
# Check email verification (grandfather existing users)
# If email_verified is None, it's an old user - allow login
# If email_verified is False, require verification
if user.email_verified is False:
flash('Please verify your email address before logging in. Check your inbox for the verification code.', 'warning')
return render_template('login.html', show_resend_verification=True, user_email=user.email, user_id=user.id)
login_user(user, remember=remember)
user.last_login = datetime.utcnow()
db.session.commit()
flash(f'Welcome back, {user.full_name}!', 'success')
next_page = request.args.get('next')
return redirect(next_page) if next_page else redirect(url_for('dashboard'))
else:
flash('Invalid email or password.', 'danger')
return render_template('login.html')
@app.route('/logout')
@login_required
def logout():
"""User logout"""
logout_user()
flash('You have been logged out successfully.', 'success')
return redirect(url_for('landing'))
# ============ GOOGLE OAUTH ROUTES ============
@app.route('/auth/google')
def google_login():
"""Initiate Google OAuth login"""
# Check if OAuth is configured
if not app.config['GOOGLE_CLIENT_ID'] or not app.config['GOOGLE_CLIENT_SECRET']:
flash('Google OAuth is not configured. Please contact administrator.', 'warning')
return redirect(url_for('login'))
redirect_uri = url_for('google_callback', _external=True)
return oauth.google.authorize_redirect(redirect_uri)
@app.route('/auth/google/callback')
def google_callback():
"""Handle Google OAuth callback"""
try:
# Get the token from Google
token = oauth.google.authorize_access_token()
# Get user info from Google
user_info = token.get('userinfo')
if not user_info:
user_info = oauth.google.parse_id_token(token)
# Extract user details
email = user_info.get('email')
name = user_info.get('name', email.split('@')[0])
google_id = user_info.get('sub')
picture = user_info.get('picture') # Get profile photo
if not email:
flash("Unable to get email from Google.", "danger")
return redirect(url_for('login'))
# Check if user exists
user = User.query.filter_by(email=email).first()
if not user:
# Create new user with Google OAuth
user = User(
email=email,
full_name=name,
google_id=google_id,
profile_photo_url=picture,
role='doctor', # Default role
is_verified=True,
is_active=True,
email_verified=True, # Google users are verified by definition
onboarding_completed=False # Require onboarding
)
db.session.add(user)
db.session.commit()
flash(f'Welcome to NeuroSight, {name}! Please complete your profile.', 'info')
else:
# Update google_id and profile photo if not set
if not user.google_id:
user.google_id = google_id
if not user.profile_photo_url and picture:
user.profile_photo_url = picture
# Ensure email is verified for Google login
if not user.email_verified:
user.email_verified = True
db.session.commit()
# Log the user in
login_user(user)
user.last_login = datetime.utcnow()
db.session.commit()
# Check if user needs onboarding
if user.needs_onboarding():
return redirect(url_for('onboarding'))
return redirect(url_for('dashboard'))
except Exception as e:
print(f"OAuth error: {e}")
flash('Authentication failed. Please try again.', 'danger')
return redirect(url_for('login'))
# ============ PASSWORD RESET ROUTES ============
@app.route('/forgot-password', methods=['GET', 'POST'])
def forgot_password():
"""Forgot password page"""
if current_user.is_authenticated:
return redirect(url_for('dashboard'))
if request.method == 'POST':
email = request.form.get('email', '').strip()
if not email:
flash('Please enter your email address.', 'warning')
return render_template('forgot_password.html')
user = User.query.filter_by(email=email).first()
if user:
# Send reset email
if send_reset_email(user):
flash('Password reset instructions have been sent to your email.', 'success')
else:
flash('Unable to send email. Please contact support.', 'danger')
else:
# Don't reveal if email exists or not (security best practice)
flash('If that email exists in our system, you will receive password reset instructions.', 'info')
return redirect(url_for('login'))
return render_template('forgot_password.html')
@app.route('/reset-password/