锦中人工智能助手

# 安装依赖

# pip install flask transformers torch

from flask import Flask, request, jsonify

from transformers import pipeline

app = Flask(__name__)

# 加载预训练的问答模型

qa_pipeline = pipeline("question-answering", model="distilbert-base-cased-distilled-mrpc")

@app.route('/ask', methods=['POST'])

def ask():

data = request.get_json()

question = data.get('question')

context = data.get('context')

if not question or not context:

return jsonify({"error": "缺少问题或上下文"}), 400

result = qa_pipeline(question=question, context=context)

return jsonify({

"answer": result['answer'],

"score": result['score']

})

if __name__ == '__main__':

app.run(debug=True)

from flask_sqlalchemy import SQLAlchemy

app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///leave.db'

db = SQLAlchemy(app)

class LeaveRequest(db.Model):

id = db.Column(db.Integer, primary_key=True)

student_id = db.Column(db.String(100), nullable=False)

reason = db.Column(db.Text, nullable=False)

status = db.Column(db.String(20), default='pending')

@app.route('/submit_leave', methods=['POST'])

def submit_leave():

data = request.get_json()

new_request = LeaveRequest(

student_id=data['student_id'],

reason=data['reason']

)

db.session.add(new_request)

db.session.commit()

return jsonify({"message": "请假申请已提交"}), 201

from celery import Celery

from flask_mail import Mail, Message

app = Flask(__name__)

app.config['CELERY_BROKER_URL'] = 'redis://localhost:6379/0'

celery = Celery(app.name, broker=app.config['CELERY_BROKER_URL'])

mail = Mail(app)

@celery.task

def send_email(subject, body, recipient):

msg = Message(subject, body=body, recipients=[recipient])

mail.send(msg)

# 在提交请假后调用该任务

send_email.delay("请假通知", "您的请假申请已提交，请注意查收。", "counselor@example.com")