import os
import re
import os
import re
from openpyxl import Workbook

def writeintxt (words,outpath):
    with open(outpath, 'w', encoding='utf-8') as output_file:
        for line in words:
            output_file.write(line)

def findfrecuency(txt_path,keywords,outpath):
    if os.path.exists(txt_path):
        # 读取原始文本文件,按行读入lines
        resultword = []
        with open(txt_path, 'r', encoding='utf-8') as file:
            lines = file.readlines()
        for line in lines:
            for keyword in keywords:
                # 对于英文关键词，确保前后不是英文字符
                if re.match(r'^[a-zA-Z]+$', keyword):
                    pattern = r'(?<![a-zA-Z])' + re.escape(keyword) + r'(?![a-zA-Z])'
                    if re.search(pattern, line, re.IGNORECASE):
                        keyword_count[keyword] += 1
                        resultword.append(line)
                # 对于中文关键词，直接寻找
                elif keyword in line:
                    keyword_count[keyword] += 1
                    resultword.append(line)
        writeintxt(resultword,outpath) 
        return keyword_count#返回字典
    else:
        print("文件不存在\n")
        return {}
    
def writeinexcel (keyword_count,savepath):
    if keyword_count == {}:
        print("没有筛选到关键词\n")
    else:
        # 创建一个新的 Excel 工作簿
        wb = Workbook()
        ws = wb.active
        ws.title = "AI应用弹幕统计"
        ws.append(['AI应用', '出现数量'])
        # 将关键词和计数写入 Excel 工作表
        for keyword, count in sorted(keyword_count.items(), key=lambda item: item[1], reverse=True):
            if count > 0:
                ws.append([keyword, count])
        # 保存 Excel 文件
        wb.save(savepath)

if __name__ =='__main__':
    # 定义文件路径
    txt_path = 'd:\\学习\\软件工程\\swork\\res\\total300_3.txt'#总弹幕数据
    keywords = ['VR', 'ai', '元宇宙', '超高清', '3D', 'gpt', '建模', '大模型', 'nlp', 'cv', 'openai', '智慧体育', '计算机', '虚拟','ai音效','ai视频','AI修复']
    # 初始化一个字典来存储每个关键词的计数
    keyword_count = {keyword: 0 for keyword in keywords}
    #定义保存位置
    savepath = "d:\\学习\\软件工程\\swork\\res\\AI_dm_rank.xlsx"#存放排名，个数excel
    outpath = "d:\\学习\\软件工程\\swork\\res\\select_ai.txt"#存放有含ai技术评论的评价txt
    keyword_count = findfrecuency(txt_path,keywords,outpath) #得到排名字典
    writeinexcel (keyword_count,savepath) #将排名写入excel