def doc_generate_prompt(date, site, persons, records, doc_type):
    person_list = []

    for i in range(len(persons)):
        name = persons['姓名'][i]
        department = persons['单位'][i]
        duty = persons['职务'][i]
        person_list.append(f'{i + 1}. {name}，所属单位：{department}，所任职务：{duty},')

    persons_prompt = '\n'.join(person_list)

    prompt = f"""
        请结合以下会议信息和要求写一篇文章：
        
        会议时间：{date}；
        
        会议地点：{site}；
        
        参会主要人物：
        {persons_prompt}
        
        会议记录：
        {records}
        
        文章类型：{doc_type}
    """

    return prompt


def daoci_prompt(name, gender, age, political, dead_time, dead_reason, experience):
    experience_list = []
    for i in range(len(experience)):
        start_time = experience['起始时间'][i]
        end_time = experience['结束时间'][i]
        department = experience['所属单位'][i]
        duty = experience['所任职务'][i]
        deed = experience['工作职责或主要事迹'][i]

        experience_list.append(
            f'1. {start_time}至{end_time}，就职于{department}，担任{duty}职务，工作指责或主要事迹为：{deed}')

    experience_prompt = '\n            '.join(experience_list)

    prompt = f"""
    使用场景：悼词是单位（团体）代表人在追悼会上对去世者表示沉痛哀悼的讲话。　　

    格式：

        一、标题：悼词的标题有几种写法或用法：　　
            １.一是在悼词正文前写上"悼词"二字；　　
            ２.主持人在追悼会上要用"ｘｘｘ同志致悼词"；　　
            ３.贴出、刊印时要用"在追悼ｘｘｘ同志大会上ｘｘｘ同志致的悼词"。　　

        二、正文：　　
            １.写明用什么心情悼念什么人；　　
            ２.写明去世者生前的身份或担任的各种职务名称，何种原因在何年何月何日几时几分不幸去世的，终年岁数；　　
            ３.按时间先后顺序介绍去世者的简单生平；　　
            ４.对去世者的称颂，可概括成几个方面，文字力求简洁；　　
            ５.对评价去世者带来的损失，应实事求是；向去世者学习什么，可分成几点写明，用什么实际行动化悲痛为力量。　　

        三、结尾：自成一段。一般有两种写法：　　
            一是一句式："ｘｘｘ同志安息吧！"　　
            二是概括式："ｘｘｘ同志和我们永别了，我们要化悲痛为力量……，　　
            ｘｘｘ同志永远是我们学习的榜样。"一定要注意简短。
            
    致悼词对象基本信息：
        
        姓名：{name}
        
        性别：{gender}
        
        年龄：{age}
        
        政治面貌：{political}
        
        去世时间：{dead_time}
        
        去世原因：{dead_reason}
        
        生平经历：
            {experience_prompt}
    """

    print(prompt)

    return prompt


def material_retrieval_prompt(doc):
    return f"""
    从以下文字中提取具体党的政机关事件、地方政策等关键词，用于在互联网上检索相关信息进一步学习，请确保提取的关键词是具体的政策、重要党政事件等可以进一步学习的知识性事物：
    
    {doc}
    
    输出格式：以json格式输出一个包含关键字字符串的列表。
    """