from flask import Flask, request
from zhipuai import ZhipuAI

app = Flask(__name__)
client = ZhipuAI(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 ''
    completion = client.chat.completions.create(
        model="glm-4-flash",
        messages=[
            {"role": "user", "content": input},
        ],
    )
    return completion.choices[0].message.content


# app.run(host='0.0.0.0', port=6721)
from waitress import serve

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