from flask import Flask, render_template, request
import requests

app = Flask(__name__)

API_URL = "https://console.bce.baidu.com/api/ai_apaas/v1/instance/integrated"  # 建议从安全的地方获取这个值
API_KEY = "bce-v3/ALTAK-kqRV2nP5xQ7cZRGIL3owz/dabc97bee254b8b24d8afc42fefeb3afbd45e85e"  # 建议从安全的地方获取这个值


@app.route("/", methods=["GET", "POST"])
def index():
    if request.method == "POST":
        query = request.form["query"]  # 从 POST 请求中获取查询参数
        headers = {
            "Content-Type": "application/json",
            "X-Authorization": f"Bearer {API_KEY}"
        }
        data = {
            "query": query,
            "response_mode": "blocking"
        }
        try:
            response = requests.post(API_URL, headers=headers, json=data)
            response.raise_for_status()  # 这将抛出 HTTPError 异常如果 HTTP 请求返回了不成功的状态码
            result = response.json()  # 获取 API 响应的 JSON 数据
            return render_template("index.html", result=result)  # 渲染模板并将结果传递给模板
        except requests.RequestException as e:  # 处理所有请求异常
            return str(e), 500
    return render_template("index.html")  # 如果请求方法是 GET，则只渲染模板（无数据）


if __name__ == "__main__":
    from wsgiref.simple_server import make_server
    httpd = make_server('', 5001, app)
    print("Serving on port 5001...")
    httpd.serve_forever()