from database.database_manager import DatabaseManager
from search.search_from_doubao import DouBao
from search.search_from_kimi import Kimi
from search.search_from_tongyi import TongYi


def send_dispatch_content_inner(content: str, category: str, title: str):
    """send dispatch content"""
    db_manager = DatabaseManager()
    db_manager.send_dispatch_content(category=category, title=title, content=content, ai_generate=1)


def search(input_text: str):
    """search messages"""
    kimi = Kimi()
    kimi_search_result = kimi.search(input_text)
    kimi.quit()
    tong_yi = TongYi()
    tong_yi_search_result = tong_yi.search(input_text)
    tong_yi.quit()
    dou_bao = DouBao()
    dou_bao_search_result = dou_bao.search(input_text)
    dou_bao.quit()
    return f"{kimi_search_result}\n{tong_yi_search_result}\n{dou_bao_search_result}"


def summarize(text: str) -> str:
    """Summarizes the given text."""
    prompt = ("请根据提供的草稿，生成一篇完整的文章。\n"
              "要求去重和格式化。\n"
              "整篇文章全部采用并且只使用简洁的有序列表格式，不需要小标题，例如：\n1、今天天气不错\n2、张三捉住一只猫\n"
              "\n以下是草稿内容：\n{}").format(text)
    kimi = Kimi()
    kimi_inference_result = kimi.search(prompt)
    kimi.quit()
    print(kimi_inference_result)
    return kimi_inference_result


def editor(subject: str, category: str, title: str):
    """这是一个编辑工具，根据提供的主题，进行素材检索和文章生成"""
    draft = search(subject)
    kimi_inference_result = summarize(draft)
    send_dispatch_content_inner(kimi_inference_result, category, title)