from flask import Flask, render_template, request, redirect, url_for, session
import constants

app = Flask(__name__)
app.secret_key = 'your_secret_key_here'

# Route for the login page
@app.route('/')
def login():
    return render_template('login.html', style=constants.LOGIN_STYLE)

# Handle login logic
@app.route('/login', methods=['POST'])
def handle_login():
    email = request.form.get('email')
    password = request.form.get('password')

    # Authenticate user (this part will be based on Salesforce or any authentication method)
    if email == "user@example.com" and password == "password123":  # Example check
        session['user'] = email
        return redirect(url_for('next_phase'))  # Move to the next phase after successful login
    else:
        return render_template('login.html', style=constants.LOGIN_STYLE, error="Invalid credentials!")

# Route for the next phase (e.g., entering details)
@app.route('/next_phase', methods=['GET', 'POST'])
def next_phase():
    if 'user' not in session:
        return redirect(url_for('login'))

    if request.method == 'POST':
        # Save details (can be integrated with Salesforce or any DB)
        # Redirect to reward points or next page
        return redirect(url_for('reward_points'))

    return render_template('next_phase.html', style=constants.PHASE_STYLE)

# Route for the reward points page
@app.route('/reward_points')
def reward_points():
    if 'user' not in session:
        return redirect(url_for('login'))

    return render_template('reward_points.html', style=constants.REWARD_STYLE)

if __name__ == '__main__':
    app.run(debug=True)