routes.py

from flask import Blueprint, render_template, request, redirect, url_for, flash, jsonify, session
from models import db, Deck, Card
from forms import DeckForm, CardForm
import random

bp = Blueprint('main', __name__)

# ============= DECK ROUTES =============

@bp.route('/')
def index():
    decks = Deck.query.order_by(Deck.created_at.desc()).all()
    total_cards = sum(deck.card_count for deck in decks)
    return render_template('index.html', decks=decks, total_cards=total_cards)

@bp.route('/decks/create', methods=['POST'])
def create_deck():
    form = DeckForm()
    if form.validate_on_submit():
        deck = Deck(
            title=form.title.data,
            description=form.description.data
        )
        db.session.add(deck)
        db.session.commit()
        flash('Deck created successfully!', 'success')
        return redirect(url_for('main.deck_detail', id=deck.id))
    else:
        for field, errors in form.errors.items():
            for error in errors:
                flash(error, 'error')
        return redirect(url_for('main.index'))

@bp.route('/decks/<int:id>')
def deck_detail(id):
    deck = Deck.query.get_or_404(id)
    form = CardForm()
    return render_template('deck.html', deck=deck, form=form)

@bp.route('/decks/<int:id>/edit', methods=['POST'])
def edit_deck(id):
    deck = Deck.query.get_or_404(id)
    form = DeckForm()
    if form.validate_on_submit():
        deck.title = form.title.data
        deck.description = form.description.data
        db.session.commit()
        flash('Deck updated successfully!', 'success')
    else:
        for field, errors in form.errors.items():
            for error in errors:
                flash(error, 'error')
    return redirect(url_for('main.deck_detail', id=id))

@bp.route('/decks/<int:id>/delete', methods=['POST'])
def delete_deck(id):
    deck = Deck.query.get_or_404(id)
    db.session.delete(deck)
    db.session.commit()
    flash('Deck deleted successfully!', 'success')
    return redirect(url_for('main.index'))

# ============= CARD ROUTES =============

@bp.route('/decks/<int:id>/cards/add', methods=['POST'])
def add_card(id):
    deck = Deck.query.get_or_404(id)
    form = CardForm()
    if form.validate_on_submit():
        card = Card(
            deck_id=deck.id,
            front=form.front.data,
            back=form.back.data
        )
        db.session.add(card)
        db.session.commit()
        flash('Card added successfully!', 'success')
    else:
        for field, errors in form.errors.items():
            for error in errors:
                flash(error, 'error')
    return redirect(url_for('main.deck_detail', id=id))

@bp.route('/cards/<int:id>/edit', methods=['POST'])
def edit_card(id):
    card = Card.query.get_or_404(id)
    form = CardForm()
    if form.validate_on_submit():
        card.front = form.front.data
        card.back = form.back.data
        db.session.commit()
        flash('Card updated successfully!', 'success')
    else:
        for field, errors in form.errors.items():
            for error in errors:
                flash(error, 'error')
    return redirect(url_for('main.deck_detail', id=card.deck_id))

@bp.route('/cards/<int:id>/delete', methods=['POST'])
def delete_card(id):
    card = Card.query.get_or_404(id)
    deck_id = card.deck_id
    db.session.delete(card)
    db.session.commit()
    flash('Card deleted successfully!', 'success')
    return redirect(url_for('main.deck_detail', id=deck_id))

# ============= STUDY/QUIZ ROUTES =============

@bp.route('/decks/<int:id>/study')
def study(id):
    deck = Deck.query.get_or_404(id)
    if not deck.cards:
        flash('This deck has no cards yet. Add some cards first!', 'error')
        return redirect(url_for('main.deck_detail', id=id))
    
    # Shuffle cards for study session
    cards = list(deck.cards)
    random.shuffle(cards)
    
    # Initialize session for tracking answers
    session['study_session'] = {
        'deck_id': deck.id,
        'correct': 0,
        'incorrect': 0,
        'missed_cards': []
    }
    
    # Convert cards to JSON-serializable format
    cards_json = [{
        'id': card.id,
        'front': card.front,
        'back': card.back
    } for card in cards]
    
    return render_template('study.html', deck=deck, cards_json=cards_json)

@bp.route('/decks/<int:id>/study/answer', methods=['POST'])
def study_answer(id):
    data = request.get_json()
    card_id = data.get('card_id')
    correct = data.get('correct', False)
    
    card = Card.query.get_or_404(card_id)
    
    # Update card statistics
    card.times_seen += 1
    if correct:
        card.times_correct += 1
    
    db.session.commit()
    
    # Update session tracking
    if 'study_session' in session:
        study_data = session['study_session']
        if correct:
            study_data['correct'] += 1
        else:
            study_data['incorrect'] += 1
            study_data['missed_cards'].append({
                'front': card.front,
                'back': card.back
            })
        session['study_session'] = study_data
    
    return jsonify({
        'success': True,
        'times_seen': card.times_seen,
        'times_correct': card.times_correct
    })

@bp.route('/decks/<int:id>/results')
def results(id):
    deck = Deck.query.get_or_404(id)
    
    # Get session data
    study_data = session.get('study_session', {
        'correct': 0,
        'incorrect': 0,
        'missed_cards': []
    })
    
    total = study_data['correct'] + study_data['incorrect']
    percent = int((study_data['correct'] / total * 100)) if total > 0 else 0
    
    return render_template('results.html', 
                         deck=deck, 
                         correct=study_data['correct'],
                         incorrect=study_data['incorrect'],
                         total=total,
                         percent=percent,
                         missed_cards=study_data['missed_cards'])