Code - 信息分类后导出表格

Excel基础操作,搭建工具类雏形

import os
from openpyxl import Workbook
from dotenv import load_dotenv

1. 加载环境变量(基础:避免密钥硬编码)

load_dotenv() # 读取.env文件(需自行创建,写入LLM_API_KEY=你的密钥)
LLM_API_KEY = os.getenv("LLM_API_KEY")

2. 简单工具类(基础:Excel保存功能,无API调用、无异步)

class SimpleClassifier:

def __init__(self):
    # 初始化Excel,写入表头
    self.wb = Workbook()
    self.ws = self.wb.active
    self.ws.title = "分类结果"
    self.headers = ["文本ID", "原文", "分类标签"]
    self.ws.append(self.headers)
    self.text_id = 1  # 自增ID

# 模拟分类(无真实API,仅手动返回标签,方便测试)
def simple_classify(self, text):
    # 模拟逻辑:含“政策”→政策通知,含“咨询”→群众咨询,否则→其他
    if "政策" in text:
        label = "政策通知"
    elif "咨询" in text:
        label = "群众咨询"
    else:
        label = "其他"
    # 写入Excel
    self.ws.append([self.text_id, text, label])
    self.text_id += 1
    self.wb.save("简单分类结果.csv")  # 保存文件
    return {"text": text, "label": label}

测试代码(直接运行可看到效果)

if name == "__main__":

classifier = SimpleClassifier()
# 测试2条文本
res1 = classifier.simple_classify("济南市2024年民生政策通知")
res2 = classifier.simple_classify("群众咨询社保缴费流程")
print("分类完成:", res1, res2)
添加新评论