from flask import Flask, request, jsonify
from api.inference import OCRPredictor

app = Flask(__name__)
predictor = OCRPredictor()

@app.route("/predict", methods=["POST"])
async def predict():
    try:
        if 'file' not in request.files:
            return jsonify({'error': 'No file part'}), 400
        
        file = request.files['file']
        if file.filename == '':
            return jsonify({'error': 'No selected file'}), 400

        result = await predictor.predict_character(file)
        return jsonify(result)

    except Exception as e:
        return jsonify({'error': str(e)}), 500

if __name__ == "__main__":
    app.run(host="0.0.0.0", port=7860)