"""
@author: luojunhui
"""
from tasks.ai_tasks.category_generation_task import ArticlePoolCategoryGenerationTask
from tasks.ai_tasks.category_generation_task import VideoPoolCategoryGenerationTask
from tasks.ai_tasks.title_rewrite_task import TitleRewriteTask


if __name__ == '__main__':
    # 1. 标题重写
    title_rewrite_task = TitleRewriteTask()
    title_rewrite_task.deal()

    # 2. 视频内容池标题分类
    video_pool_category_generation_task = VideoPoolCategoryGenerationTask()
    video_pool_category_generation_task.deal()

    # 3. 文章内容池标题分类
    article_pool_category_generation_task = ArticlePoolCategoryGenerationTask()
    article_pool_category_generation_task.deal()