from flask import Flask, request, jsonify
from transformers import pipeline

app = Flask(__name__)

# ✅ 경제 해석 모델: 분석기 (analyzer)
econ_analyzer = pipeline(
    "summarization",  # 🤫 사실은 요약 파이프라인을 활용한 “해석 생성기” 역할
    model="digit82/kobart-summarization",
    tokenizer="digit82/kobart-summarization"
)

@app.route("/")
def home():
    return "📊 CW 경제 분석기 (KoBART 기반) 동작 중"

@app.route("/econalyze", methods=["POST"])
def analyze_economy():
    data = request.get_json()
    econ_input = data.get("text", "")
    
    if not econ_input.strip():
        return jsonify({"error": "경제 지표 입력이 없습니다"}), 400

    try:
        result = econ_analyzer(
            econ_input,
            max_length=300,
            min_length=80,
            do_sample=False
        )
        return jsonify({"analysis": result[0]["summary_text"]})
    except Exception as e:
        return jsonify({"error": str(e)}), 500

if __name__ == "__main__":
    app.run(host="0.0.0.0", port=7861)