from flask import Flask, request
from openai import OpenAI
from waitress import serve

app = Flask(__name__)
client = OpenAI(base_url="https://open.bigmodel.cn/api/paas/v4", api_key="d1e97306540d12bb2f834be961fcacb1.SNBShlCxWYJCx0qZ")


@app.route('/ai_text', methods=['POST'])
def ai_text():
    _input = request.get_data(as_text=True)
    # 如果input值为空，则返回空字符串
    if not _input:
        return ''
    print(f'input: {_input}')
    # app.logger.info(f'input: {input}')
    completion = client.chat.completions.create(
        model="glm-4-flash",
        messages=[
            {"role": "user", "content": _input},
        ],
        stream=False
    )
    _output = completion.choices[0].message.content
    # app.logger.info(f'output: {output}')
    return _output


serve(app, host='0.0.0.0', port=6721)