# textsumm.py
# 中文化摘要模組，安裝 transformers 與 torch 即可使用

from transformers import pipeline

# 初始化摘要 pipeline
summarizer = pipeline("summarization", model="facebook/bart-large-cnn")

def 文字摘要(輸入文本, 最長長度=120, 最短長度=40):
    """
    輸入：一段文本
    輸出：摘要（中文說明）
    """
    if len(輸入文本.strip()) == 0:
        return "❗️ 請輸入需要摘要的內容。"

    try:
        結果 = summarizer(
            輸入文本,
            max_length=最長長度,
            min_length=最短長度,
            do_sample=False
        )
        return 結果[0]['summary_text']
    except Exception as e:
        return f"❌ 摘要生成失敗：{str(e)}"

# 若你要測試，可以取消下面註解
# if __name__ == "__main__":
#     測試文本 = "人工智慧（AI）是研究如何讓電腦模擬人類智能行為的學科，包括學習、推理、規劃、自然語言處理、知覺等。AI 技術已廣泛應用於語音辨識、影像分析、自駕車等領域。"
#     print(文字摘要(測試文本))