from reptile import reptile
from word_handle import processor

if __name__ == '__main__':
    key_words = str(input('请输入搜索关键词：'))
    page = int(input('请输入需要再部长综合排序中爬取的视频个数：'))
    reptile = reptile(key_words,page)

    path = reptile.get_dataset()
    test_path = './巴黎奥运会AI文本库.txt'
    userdict_path = './AI专有名词.txt'
    exl_path = './b站视频2024年巴黎运动会AI应用的弹幕的统计表格1.xlsx'
    wordcloud_path = './AImap1.png'
    process = processor(path,test_path,userdict_path,exl_path,wordcloud_path)
    process.analyse_vector_tfidf()
    process.draw_wordcloud()