from chat import gen_title, generate_text

from flask import Flask, request, jsonify
app = Flask(__name__)


@app.route('/title', methods=['POST'])
def request_title():
    try:
        spu = request.form.get("spu", None)  
        referencr_title = request.form.get("referencr_title", None) 
        desc = request.form.get("plm_info")
        if  not desc:
            return jsonify({"error": "desc is required"}), 400

        tags=request.form.get("tags", None) 
        result=gen_title(desc,tags,referencr_title)
        response = {
            "spu": spu,
            "result": result
        }
        return jsonify(response), 200
    except Exception as e:
        return jsonify({"error": str(e)}), 500

@app.route('/description', methods=['POST'])
def request_description():
    try:
        spu = request.form.get("spu", None)  
        reference_url = request.form.get("reference_url", None) 
        graphic_label = request.form.get("graphic_label", None)
        img = request.form.get("img")
        if  not img:
            return jsonify({"error": "img is required"}), 400
        plm_info = request.form.get("plm_info")
        if  not plm_info:
            return jsonify({"error": "plm_info is required"}), 400
        result=generate_text(plm_info,img,graphic_label)
        response = {
            "spu": spu,
            "result": result
        }
        return jsonify(response), 200
    except Exception as e:
        return jsonify({"error": str(e)}), 500

if __name__ == "__main__":
    app.run(host="0.0.0.0",port=6868,debug=True)