"""Public stage descriptions; never forward graph task inputs or model reasoning."""

STAGES = {
    'session': ('会话准备', '正在接续您的咨询，请稍等……'),
    'prepare': ('问题准备', '正在仔细看您提出的问题……'),
    'analyze': ('需求识别', '正在了解您想解决的事情……'),
    'direct': ('生成答复', '正在为您准备回复……'),
    'clarify': ('信息补充', '正在看看还需要向您了解哪些情况……'),
    'search': ('公司查找', '正在根据您提供的线索查找企业资料……'),
    'confirm': ('公司确认', '正在整理企业信息，方便您确认是哪一家……'),
    'basic': ('工商信息查询', '正在查阅您已确认企业的工商资料……'),
    'honors': ('资质信息查询', '正在查阅企业的荣誉和资质资料……'),
    'handoff': ('需求整理', '正在结合您提供的情况，梳理需要关注的事项……'),
    'plan': ('检索规划', '正在梳理与您咨询事项有关的资料方向……'),
    'retrieve': ('知识库检索', '正在查阅与您咨询事项有关的政策和办事资料……'),
    'recommend': ('原文分析与回答', '正在对照资料原文，把与您问题有关的重点说明白……'),
}

COMPLETED = {
    'session': '可以继续了解您的问题了。',
    'prepare': '已读完您的问题，接下来了解具体需求。',
    'analyze': '已结束需求梳理，接下来按您的咨询内容继续。',
    'direct': '这次回复已准备好。',
    'clarify': '需要向您了解的问题已整理好。',
    'search': '企业资料查找已结束，接下来整理查找结果。',
    'confirm': '企业信息确认这一步已结束。',
    'basic': '已结束本次工商资料查阅。',
    'honors': '已结束本次荣誉和资质资料查阅。',
    'handoff': '已有情况已整理，接下来查阅相关资料。',
    'plan': '资料查阅方向已整理好。',
    'retrieve': '资料查阅已结束，接下来对照原文整理答复。',
    'recommend': '这次资料答复已整理好，请查看具体内容。',
}


def progress(stage, status='running'):
    title, running = STAGES[stage]
    message = {'running': running, 'completed': COMPLETED[stage],
               'waiting_input': '还需要您补充或确认一点情况，才能继续帮您查找。',
               'failed': f'抱歉，这次{title}没能完成，请查看具体提示。'}[status]
    return {'stage': stage, 'status': status, 'message': message}