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)