generated from dellevin/template
50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
fen-ci 分词蓝图
|
|
"""
|
|
from flask import Blueprint, render_template, request, jsonify
|
|
|
|
from utils.fen_ci_utils import tokenizer, token_auth
|
|
|
|
bp = Blueprint('fen_ci', __name__, url_prefix='/fen-ci')
|
|
|
|
|
|
def _authenticate():
|
|
"""检查请求头中的 Authorization token"""
|
|
auth_header = request.headers.get('Authorization')
|
|
if not auth_header or not auth_header.startswith('Bearer '):
|
|
return None, "缺少或格式错误的Authorization头。请使用 'Authorization: Bearer <your_token>'"
|
|
token = auth_header.split(" ", 1)[1]
|
|
user = token_auth.get_user(token)
|
|
if not user:
|
|
return None, "无效的Token"
|
|
return user, None
|
|
|
|
|
|
@bp.route('/')
|
|
def page():
|
|
return render_template('fen_ci.html')
|
|
|
|
|
|
@bp.route('/tokenize', methods=['POST'])
|
|
def tokenize():
|
|
user, error_msg = _authenticate()
|
|
if not user:
|
|
return jsonify({'success': False, 'error': error_msg}), 401
|
|
|
|
data = request.get_json()
|
|
input_text = data.get('input_text', '')
|
|
|
|
try:
|
|
tokens = tokenizer.tokenize(input_text)
|
|
stats = tokenizer.get_stats(tokens)
|
|
return jsonify({
|
|
'success': True,
|
|
'tokens': tokens,
|
|
'stats': stats,
|
|
'original_text': input_text,
|
|
'requested_by': user
|
|
})
|
|
except Exception as e:
|
|
return jsonify({'success': False, 'error': f'分词处理失败: {str(e)}'}), 500
|