import math

class Config:
    def __init__(self):
        # 目录提取拼接相关参数
        self.pantten_second_biaoti = '[2二ⅡⅠ][、.]\s{0,}?[\u4e00-\u9fa5]+'
        self.pantten_other_biaoti = '[2-9二三四五六七八九ⅡⅢⅣⅤⅥⅦⅧⅨ][、.]\s{0,}?[\u4e00-\u9fa5]+'
        self.pantten_biaoti = '[1-9一二三四五六七八九ⅠⅡⅢⅣⅤⅥⅦⅧⅨ][、.]\s{0,}?[\u4e00-\u9fa5a-zA-Z]+'

        # chatgpt 接口相关参数

        self.mulu_prompt = "生成目录#\n问：为论文题目《{}》生成目录，要求只有一级标题和二级标题，一级标题使用中文数字 例如一、xxx；二级标题使用阿拉伯数字 例如1.1 xxx；一级标题不少于7个；每个一级标题至少包含3个二级标题\n答：\n"
        self.first_title_prompt = "生成论文小标题内容#\n问：论文题目是《{}》，目录是“{}”，请把其中的大标题“{}”的内容补充完整，补充内容字数在{}字左右\n答：\n"
        self.small_title_prompt = "生成论文小标题内容#\n问：论文题目是《{}》，目录是“{}”，请把其中的小标题“{}”的内容补充完整，补充内容字数在{}字左右\n答：\n"
        self.references_prompt = "论文题目是“{}”，目录是“{}”，请为这篇论文生成15篇左右的参考文献，要求其中有有中文参考文献不低于12篇，英文参考文献不低于2篇"
        self.thank_prompt = "请以“{}”为题写一篇论文的致谢"
        self.kaitibaogao_prompt = "请以《{}》为题目生成研究的主要的内容、背景、目的、意义，要求不少于1500字"
        self.chinese_abstract_prompt = "生成论文摘要#\n"
        self.english_abstract_prompt = "请把“{}”这段文字翻译成英文"
        self.chinese_keyword_prompt = "请为“{}”这段论文摘要生成3-5个关键字"
        self.english_keyword_prompt = "请把“{}”这几个关键字翻译成英文"
        self.dabiaoti = ["二", "三", "四", "五", "六", "七", "八", "九"]
        self.project_data_txt_path = "/home/majiahui/project2/LLaMA-Efficient-Tuning/new_data_txt_1"

        # 流程相关参数
        self.thanks = "致谢"
        self.references = "参考文献"
        self.excursus = "附录"
        self.u = 3.5  # 均值μ
        self.sig = math.sqrt(6.0)
        self.zong_gradient = 6
        self.paper_word_count = 12000

        # flask port
        self.flask_port = "14003"

        # redis config
        self.reids_ip = '192.168.31.145'
        self.reids_port = 6379
        self.reids_db = 7
        self.reids_password = 'Zhicheng123*'