锦中人工智能助手

# app.py

from flask import Flask, request, jsonify

import nltk

from nltk.stem import WordNetLemmatizer

nltk.download('punkt')

nltk.download('wordnet')

app = Flask(__name__)

lemmatizer = WordNetLemmatizer()

def preprocess(text):

tokens = nltk.word_tokenize(text)

tokens = [lemmatizer.lemmatize(word.lower()) for word in tokens if word.isalpha()]

return ' '.join(tokens)

def get_response(user_input):

user_input = preprocess(user_input)

if '课程' in user_input:

return "您询问的是课程相关问题，请告诉我具体课程名称。"

elif '考试' in user_input:

return "您询问的是考试相关问题，请告诉我具体考试科目。"

else:

return "您好，我是校园智能体助手，您可以问我关于课程、考试、学习资源等问题。"

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

def query():

data = request.get_json()

user_input = data['input']

response = get_response(user_input)

return jsonify({'response': response})

if __name__ == '__main__':

app.run(debug=True)

# train_model.py

from sklearn.feature_extraction.text import TfidfVectorizer

from sklearn.svm import SVC

from sklearn.pipeline import Pipeline

import pickle

# 示例数据

X_train = [

"帮我查一下今天下午的课程安排",

"明天的数学考试时间是什么时候？",

"推荐一些编程学习资料",

"如何提交作业？"

]

y_train = [

"course",

"exam",

"study",

"assignment"

]

model = Pipeline([

('tfidf', TfidfVectorizer()),

('clf', SVC(kernel='linear'))

])

model.fit(X_train, y_train)

# 保存模型

with open('model.pkl', 'wb') as f:

pickle.dump(model, f)

# app.py (更新版)

from flask import Flask, request, jsonify

import nltk

from nltk.stem import WordNetLemmatizer

import pickle

nltk.download('punkt')

nltk.download('wordnet')

app = Flask(__name__)

lemmatizer = WordNetLemmatizer()

def preprocess(text):

tokens = nltk.word_tokenize(text)

tokens = [lemmatizer.lemmatize(word.lower()) for word in tokens if word.isalpha()]

return ' '.join(tokens)

# 加载模型

with open('model.pkl', 'rb') as f:

model = pickle.load(f)

def get_response(user_input):

user_input = preprocess(user_input)

intent = model.predict([user_input])[0]

if intent == 'course':

return "您询问的是课程相关问题，请告诉我具体课程名称。"

elif intent == 'exam':

return "您询问的是考试相关问题，请告诉我具体考试科目。"

elif intent == 'study':

return "我为您推荐一些编程学习资料，包括《Python编程从入门到实践》和《机器学习实战》。"

elif intent == 'assignment':

return "请登录学校教务系统提交作业，如有疑问请联系辅导员。"

else:

return "您好，我是校园智能体助手，您可以问我关于课程、考试、学习资源等问题。"

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

def query():

data = request.get_json()

user_input = data['input']

response = get_response(user_input)

return jsonify({'response': response})

if __name__ == '__main__':

app.run(debug=True)