demo_server.py.bak

#!/usr/env python3
# -*- coding: UTF-8 -*-

from flask import Flask, request, send_file, make_response
from flask_cors import CORS
import json
import random

import get_html

app = Flask(__name__)
CORS(app, resources={r"/*": {"origins": "https://yiyan.baidu.com"}})

wordbook = []

def make_json_response(data, status_code=200):
    response = make_response(json.dumps(data), status_code)
    response.headers["Content-Type"] = "application/json"
    return response


@app.route("/add_word", methods=['POST'])
async def add_word():
    """
        添加一个单词
    """
    word = request.json.get('word', "")
    wordbook.append(word)
    return make_json_response({"message": "单词添加成功"})

@app.route("/search_word", methods=['POST'])
async def search_word():
    """
        搜索单词
    """
    word = request.json.get('word', "")
    htmlStr = get_html.getHtmlContent(word)
    soup = get_html.getSoup(htmlStr)

    descList = []
    # 取前6个结果
    for s in soup.find_all('h2')[0:5]:
        spans = s.find_parent().find_all('span')
        desc = spans.pop().get_text().strip()
        title = s.a.get_text().strip()
        item = { "title": title, "desc": desc}
        descList.append(item)
    print('descList===>', descList)

    links = list(set(s.a for s in soup.find_all('h2')))
    # print('links====>',links)
    titles = list(set([link.get_text().strip() for link in links]))
    # print('titles====>',titles)
    # 清除空值
    if '' in titles:
        titles.remove('')
    # print('selector titles===>', titles)
    # print('htmlStr==>', htmlStr)
    prompt = "搜索结果列表List##(descList)##的JSON数据格式包含标题##title##和剧情描述##desc##。"\
             "请将搜索列表的标题##title##作为选项内容,"\
             "保留该选项内容对应的标题##title##与##(wordKey)##匹配度较高的前三个选项,"\
             "且只保留三个选项并按列表选项的形式排列展示,"\
             "优化列表选项的中文内容并删除非中文内容,"\
             "将每一个选项内容翻译成英文并且展示在后面," \
             "并在第二行summary该选项对应的剧情描述##desc##,"\
             "格式如下：##中文内容##(##英文内容##),"\
             "最后提示用户选择一个列表选项,"\
             "如果没有匹配到列表选项，必须展示三个与##(wordKey)##匹配度很高的电影名称推荐给用户。"
    return make_json_response({"message": "搜索结果",  "wordKey": word,"descList": descList, prompt: prompt})

@app.route("/delete_word", methods=['DELETE'])
async def delete_word():
    """
        删除一个单词
    """
    word = request.json.get('word', "")
    if word in wordbook:
        wordbook.remove(word)
    return make_json_response({"message": "单词删除成功"})


@app.route("/get_wordbook")
async def get_wordbook():
    """
        获得单词本
    """
    return make_json_response({"wordbook": wordbook})


@app.route("/generate_sentences", methods=['POST'])
async def generate_sentences():
    """
        生成句子
    """
    number = request.get_json()['word_number']
    number = min(number, len(wordbook))
    random_words = random.sample(wordbook, number)
    prompt = "利用英文单词（words）生成一个英文段落，要求这个段落不超过100个英文单词且必须全英文，" \
             "并包含上述英文单词，同时是一个有逻辑的句子，" \
             "并将英文段落翻译成中文，同时用体育老师的口吻陈述这个段落"
    # API返回字段"prompt"有特殊含义：开发者可以通过调试它来调试输出效果
    return make_json_response({"words": random_words, "prompt": prompt})


@app.route("/logo.png")
async def plugin_logo():
    """
        注册用的：返回插件的logo，要求48 x 48大小的png文件.
        注意：API路由是固定的，事先约定的。
    """
    return send_file('logo-n.png', mimetype='image/png')


@app.route("/.well-known/ai-plugin.json")
async def plugin_manifest():
    """
        注册用的：返回插件的描述文件，描述了插件是什么等信息。
        注意：API路由是固定的，事先约定的。
    """
    host = request.host_url
    with open(".well-known/ai-plugin.json", encoding="utf-8") as f:
        text = f.read().replace("PLUGIN_HOST", host)
        return text, 200, {"Content-Type": "application/json"}


@app.route("/.well-known/openapi.yaml")
async def openapi_spec():
    """
        注册用的：返回插件所依赖的插件服务的API接口描述，参照openapi规范编写。
        注意：API路由是固定的，事先约定的。
    """
    with open(".well-known/openapi.yaml", encoding="utf-8") as f:
        text = f.read()
        return text, 200, {"Content-Type": "text/yaml"}


@app.route('/')
def index():
    return 'welcome to my webpage!'

if __name__ == '__main__':
    app.run(debug=True, host='127.0.0.1', port=8081)