from flask import render_template, request, jsonify
from app import app, model
from app.utils import process_image, get_prediction
import base64, re

@app.route('/')
def index():
    return render_template('index.html')

@app.route('/predict', methods=['POST'])
def predict():
    canvas_data = request.form.get('canvas_data')

    if not canvas_data:
        return jsonify({'error': 'No image provided'}), 400

    canvas_data = re.sub('^data:image/.+;base64,', '', canvas_data)
    image_data = base64.b64decode(canvas_data)

    img_array = process_image(image_data)
    digit, confidence, probabilities = get_prediction(img_array, model)

    return jsonify({
        'digit': digit,
        'confidence': round(confidence, 2),
        'probabilities': probabilities
    })