import json
import requests
from core.content_db import Content_Db


def question(cont):
    content_db = Content_Db()
    list = content_db.get_list('all','desc',10)
    answer_info = dict()
    chat_list = []
    for val in list:
        answer_info = dict()
        if val[0] == "member":
            answer_info["role"] = "user"
            answer_info["content"] = val[2]
        elif val[0] == "fay":
            answer_info["role"] = "bot"
            answer_info["content"] = val[2]
        chat_list.append(answer_info)
    content = {
        "prompt":"请简单回复我。" +  cont,
        "history":chat_list}
    url = "http://127.0.0.1:8000"
    req = json.dumps(content)
    headers = {'content-type': 'application/json'}
    r = requests.post(url, headers=headers, data=req)
    res = json.loads(r.text).get('response')
    return res