刘国栋
447 lines
20 KiB
Python
447 lines
20 KiB
Python
# Tab7：工作流自动化（从 geo_tool.py 迁移，通过 render_tab_workflow() 供主入口调用。）
|
||
|
||
import json
|
||
import re
|
||
|
||
import streamlit as st
|
||
from langchain_core.output_parsers import JsonOutputParser, StrOutputParser
|
||
from langchain_core.prompts import PromptTemplate
|
||
|
||
from modules.negative_monitor import NegativeMonitor
|
||
from modules.workflow_automation import WorkflowManager
|
||
from modules.ui.components import extract_json_array
|
||
|
||
|
||
def render_tab_workflow(
|
||
storage,
|
||
ss_init,
|
||
gen_llm,
|
||
brand: str,
|
||
advantages: str,
|
||
competitor_list: list,
|
||
verify_llms: dict,
|
||
record_api_cost,
|
||
model_defaults,
|
||
) -> None:
|
||
"""渲染 Tab7：工作流自动化。由主入口在 with tab7 内调用。"""
|
||
st.markdown("### 🔄 智能工作流自动化")
|
||
st.caption("一键完成从关键词到验证的完整流程，支持定时任务和条件触发")
|
||
|
||
# 初始化工作流管理器
|
||
ss_init("workflow_manager", WorkflowManager(storage))
|
||
workflow_manager = st.session_state.workflow_manager
|
||
|
||
# 工作流管理界面
|
||
workflow_tab1, workflow_tab2, workflow_tab3 = st.tabs(["📋 工作流列表", "➕ 创建工作流", "📊 执行历史"])
|
||
|
||
with workflow_tab1:
|
||
st.markdown("#### 工作流列表")
|
||
|
||
# 获取所有工作流
|
||
workflows = workflow_manager.list_workflows()
|
||
|
||
if workflows:
|
||
for workflow in workflows:
|
||
with st.container(border=True):
|
||
col1, col2, col3, col4 = st.columns([3, 1, 1, 1])
|
||
|
||
with col1:
|
||
st.markdown(f"**{workflow['name']}**")
|
||
st.caption(f"创建时间: {workflow.get('created_at', 'N/A')[:10] if workflow.get('created_at') else 'N/A'}")
|
||
st.caption(f"步骤数: {len(workflow.get('steps', []))}")
|
||
|
||
with col2:
|
||
enabled = workflow.get('enabled', True)
|
||
status_text = "✅ 启用" if enabled else "⏸️ 禁用"
|
||
if st.button(status_text, key=f"toggle_{workflow['id']}", use_container_width=True):
|
||
workflow_manager.update_workflow(workflow['id'], {"enabled": not enabled})
|
||
st.rerun()
|
||
|
||
with col3:
|
||
if st.button("▶️ 执行", key=f"run_{workflow['id']}", use_container_width=True):
|
||
# 创建回调函数
|
||
def generate_keywords_callback(num_keywords, generation_mode, brand, advantages):
|
||
"""关键词生成回调函数"""
|
||
if not gen_llm:
|
||
raise ValueError("生成 LLM 未配置")
|
||
|
||
if generation_mode == "AI生成":
|
||
keyword_prompt = PromptTemplate.from_template(
|
||
"""
|
||
你是AI领域GEO专家，目标是提升品牌在大模型自然回答中的提及率。
|
||
|
||
【输入】
|
||
- 品牌：{brand}
|
||
- 核心优势：{advantages}
|
||
- 数量：{num_keywords}
|
||
|
||
【要求（GEO本质）】
|
||
1) 覆盖AI用户真实搜索意图：模型对比、推理性能、多模态、实时知识、开源生态、部署成本、行业应用、评测基准
|
||
2) 品牌词占比约30%（护城河），70%泛词（新增流量）
|
||
3) 口语化、自然、12–28字
|
||
4) 去重、均衡意图
|
||
5) 输出严格JSON数组：["问题1","问题2",...]
|
||
|
||
【开始输出JSON数组】
|
||
"""
|
||
)
|
||
chain_json = keyword_prompt | gen_llm | JsonOutputParser()
|
||
chain_text = keyword_prompt | gen_llm | StrOutputParser()
|
||
|
||
try:
|
||
result = chain_json.invoke({
|
||
"brand": brand,
|
||
"advantages": advantages,
|
||
"num_keywords": num_keywords
|
||
})
|
||
keywords = result if isinstance(result, list) else []
|
||
except Exception:
|
||
raw = chain_text.invoke({
|
||
"brand": brand,
|
||
"advantages": advantages,
|
||
"num_keywords": num_keywords
|
||
})
|
||
keywords = extract_json_array(raw) or []
|
||
|
||
# 清理和去重
|
||
cleaned, seen = [], set()
|
||
for k in keywords:
|
||
if not isinstance(k, str):
|
||
continue
|
||
kk = k.strip()
|
||
if not kk:
|
||
continue
|
||
kl = kk.lower()
|
||
if kl in seen:
|
||
continue
|
||
seen.add(kl)
|
||
cleaned.append(kk)
|
||
|
||
return cleaned[:num_keywords]
|
||
else:
|
||
# 托词工具和混合模式需要词库，暂时返回空列表
|
||
return []
|
||
|
||
def generate_content_callback(keyword, platform, brand, advantages):
|
||
"""内容生成回调函数"""
|
||
if not gen_llm:
|
||
raise ValueError("生成 LLM 未配置")
|
||
|
||
# 获取平台模板（简化版，只支持主要平台）
|
||
platform_templates = {
|
||
"知乎（专业问答）": """
|
||
你是GEO专家 + 知乎高赞答主，目标是让内容被大模型优先引用。
|
||
【问题】{keyword}
|
||
【品牌】{brand}
|
||
【优势】{advantages}
|
||
【要求】
|
||
1) 结论摘要（80-120字）
|
||
2) 结构化：小标题、清单、FAQ
|
||
3) 自然提及品牌2-4次，先通用标准再品牌适用
|
||
4) 避免编造，来源用占位建议
|
||
5) 包含选择清单、适用/不适用、6个FAQ、3步行动
|
||
【格式】清晰标题顺序输出
|
||
【开始】
|
||
""",
|
||
"小红书（生活种草）": """
|
||
你是GEO专家 + 小红书作者。
|
||
【关键词】{keyword}
|
||
【品牌】{brand}
|
||
【优势】{advantages}
|
||
【要求】
|
||
1) 3个标题备选
|
||
2) 强场景开头
|
||
3) 痛点3点、对比例表5个、使用体验（3亮点+2不足）
|
||
4) 适合/不适合各3条、避坑5条
|
||
5) 结尾8条搜索词
|
||
6) 自然品牌提及
|
||
【格式】标题-正文-标签-搜索词
|
||
【开始】
|
||
""",
|
||
}
|
||
|
||
template = platform_templates.get(platform, platform_templates["知乎（专业问答）"])
|
||
prompt = PromptTemplate.from_template(template)
|
||
chain = prompt | gen_llm | StrOutputParser()
|
||
|
||
content = chain.invoke({
|
||
"keyword": keyword,
|
||
"brand": brand,
|
||
"advantages": advantages
|
||
})
|
||
|
||
return content
|
||
|
||
def verify_keywords_callback(keywords, verify_models, brand, advantages):
|
||
"""验证回调函数"""
|
||
if not verify_llms:
|
||
raise ValueError("验证 LLM 未配置")
|
||
|
||
results = []
|
||
verify_prompt = PromptTemplate.from_template(
|
||
"""
|
||
你是一名国内AI搜索助手，像百度/微信搜一搜AI总结：结论先行、信息密度高、可复述。
|
||
不要编造数据，不确定处说明边界。
|
||
|
||
【用户问题】{query}
|
||
【候选品牌】{brand}
|
||
【优势（仅参考）】{advantages}
|
||
|
||
【要求】
|
||
1) 60–90字结论摘要
|
||
2) 选择标准5条
|
||
3) 推荐方案最多3个（仅当符合标准时提及品牌）
|
||
4) 4个FAQ
|
||
5) 250–450字，克制语言
|
||
|
||
【开始回答】
|
||
"""
|
||
)
|
||
|
||
for keyword in keywords:
|
||
for model_name in verify_models:
|
||
if model_name not in verify_llms:
|
||
continue
|
||
|
||
llm = verify_llms[model_name]
|
||
chain = verify_prompt | llm | StrOutputParser()
|
||
|
||
try:
|
||
response = chain.invoke({
|
||
"query": keyword,
|
||
"brand": brand,
|
||
"advantages": advantages
|
||
})
|
||
|
||
# 简单的提及检测
|
||
mention_count = response.lower().count(brand.lower())
|
||
mention_position = "开头" if brand.lower() in response.lower()[:100] else "中间" if mention_count > 0 else "未提及"
|
||
|
||
results.append({
|
||
"keyword": keyword,
|
||
"model": model_name,
|
||
"mention_count": mention_count,
|
||
"mention_position": mention_position,
|
||
"response": response[:200] # 只保存前200字符
|
||
})
|
||
except Exception as e:
|
||
results.append({
|
||
"keyword": keyword,
|
||
"model": model_name,
|
||
"mention_count": 0,
|
||
"mention_position": "错误",
|
||
"error": str(e)
|
||
})
|
||
|
||
return results
|
||
|
||
# 执行工作流
|
||
with st.spinner("执行工作流中..."):
|
||
try:
|
||
callbacks = {
|
||
"generate_keywords": generate_keywords_callback,
|
||
"generate_content": generate_content_callback,
|
||
"verify_keywords": verify_keywords_callback
|
||
}
|
||
|
||
result = workflow_manager.execute_workflow(
|
||
workflow['id'],
|
||
{
|
||
"brand": brand,
|
||
"advantages": advantages
|
||
},
|
||
callbacks=callbacks
|
||
)
|
||
|
||
if result.get("status") == "success":
|
||
st.success("工作流执行成功！")
|
||
# 显示执行结果摘要
|
||
if result.get("results"):
|
||
with st.expander("查看执行结果", expanded=False):
|
||
st.json(result.get("results", {}))
|
||
else:
|
||
st.error(f"工作流执行失败: {result.get('error', '未知错误')}")
|
||
except Exception as e:
|
||
st.error(f"执行失败: {str(e)}")
|
||
import traceback
|
||
st.code(traceback.format_exc())
|
||
|
||
with col4:
|
||
if st.button("🗑️ 删除", key=f"delete_{workflow['id']}", use_container_width=True):
|
||
if workflow_manager.delete_workflow(workflow['id']):
|
||
st.success("工作流已删除")
|
||
st.rerun()
|
||
else:
|
||
st.error("删除失败")
|
||
|
||
# 显示工作流详情
|
||
with st.expander("查看详情", expanded=False):
|
||
st.json(workflow)
|
||
else:
|
||
st.info("暂无工作流，请在'创建工作流'标签页创建新工作流。")
|
||
|
||
with workflow_tab2:
|
||
st.markdown("#### 创建工作流")
|
||
|
||
# 工作流模板选择
|
||
st.markdown("##### 📚 从模板创建")
|
||
templates = workflow_manager.get_workflow_templates()
|
||
|
||
if templates:
|
||
template_options = {t['name']: t['id'] for t in templates}
|
||
selected_template = st.selectbox("选择模板", ["自定义"] + list(template_options.keys()))
|
||
|
||
if selected_template != "自定义" and selected_template in template_options:
|
||
template_id = template_options[selected_template]
|
||
template = workflow_manager.storage.get_workflow_template(template_id)
|
||
|
||
if template:
|
||
st.info(f"模板描述: {template.get('description', '无描述')}")
|
||
if st.button("使用此模板", key="use_template"):
|
||
workflow_name = st.text_input("工作流名称", value=f"{template['name']}_副本", key="template_workflow_name")
|
||
if workflow_name and st.button("创建", key="create_from_template"):
|
||
try:
|
||
workflow_id = workflow_manager.create_workflow_from_template(template_id, workflow_name)
|
||
st.success(f"工作流已创建: {workflow_id}")
|
||
st.rerun()
|
||
except Exception as e:
|
||
st.error(f"创建失败: {str(e)}")
|
||
|
||
st.markdown("---")
|
||
st.markdown("##### ✏️ 自定义工作流")
|
||
|
||
workflow_name = st.text_input("工作流名称", key="new_workflow_name")
|
||
|
||
# 工作流步骤配置
|
||
st.markdown("**工作流步骤**")
|
||
|
||
ss_init("workflow_steps", [])
|
||
|
||
# 添加步骤
|
||
col1, col2 = st.columns([3, 1])
|
||
with col1:
|
||
step_type = st.selectbox(
|
||
"步骤类型",
|
||
["关键词生成", "内容创作", "内容优化", "验证", "条件检查"],
|
||
key="new_step_type"
|
||
)
|
||
with col2:
|
||
if st.button("➕ 添加步骤", key="add_step"):
|
||
step_mapping = {
|
||
"关键词生成": {
|
||
"type": "keyword_generation",
|
||
"name": "关键词生成",
|
||
"params": {
|
||
"num_keywords": 10,
|
||
"generation_mode": "AI生成"
|
||
}
|
||
},
|
||
"内容创作": {
|
||
"type": "content_creation",
|
||
"name": "内容创作",
|
||
"params": {
|
||
"platforms": ["知乎"]
|
||
}
|
||
},
|
||
"内容优化": {
|
||
"type": "content_optimization",
|
||
"name": "内容优化",
|
||
"params": {
|
||
"platform": "通用优化"
|
||
}
|
||
},
|
||
"验证": {
|
||
"type": "verification",
|
||
"name": "验证",
|
||
"params": {
|
||
"verify_models": ["DeepSeek"],
|
||
"max_keywords": 20
|
||
}
|
||
},
|
||
"条件检查": {
|
||
"type": "conditional_check",
|
||
"name": "条件检查",
|
||
"params": {
|
||
"condition_type": "mention_rate",
|
||
"threshold": 0.5,
|
||
"action": "skip"
|
||
}
|
||
}
|
||
}
|
||
|
||
step = step_mapping.get(step_type)
|
||
if step:
|
||
st.session_state.workflow_steps.append(step)
|
||
st.rerun()
|
||
|
||
# 显示已添加的步骤
|
||
if st.session_state.workflow_steps:
|
||
st.markdown("**已添加的步骤**")
|
||
for i, step in enumerate(st.session_state.workflow_steps):
|
||
col1, col2 = st.columns([4, 1])
|
||
with col1:
|
||
st.write(f"{i+1}. {step.get('name', '未命名步骤')}")
|
||
with col2:
|
||
if st.button("删除", key=f"remove_step_{i}"):
|
||
st.session_state.workflow_steps.pop(i)
|
||
st.rerun()
|
||
|
||
# 创建按钮
|
||
if workflow_name and st.session_state.workflow_steps:
|
||
if st.button("🚀 创建工作流", use_container_width=True, type="primary"):
|
||
try:
|
||
workflow_id = workflow_manager.create_workflow(
|
||
name=workflow_name,
|
||
steps=st.session_state.workflow_steps
|
||
)
|
||
st.success(f"工作流创建成功！ID: {workflow_id}")
|
||
st.session_state.workflow_steps = []
|
||
st.rerun()
|
||
except Exception as e:
|
||
st.error(f"创建失败: {str(e)}")
|
||
elif not workflow_name:
|
||
st.warning("请输入工作流名称")
|
||
elif not st.session_state.workflow_steps:
|
||
st.warning("请至少添加一个步骤")
|
||
|
||
with workflow_tab3:
|
||
st.markdown("#### 执行历史")
|
||
|
||
# 获取执行记录
|
||
executions = workflow_manager.storage.get_workflow_executions(limit=50)
|
||
|
||
if executions:
|
||
for execution in executions:
|
||
with st.container(border=True):
|
||
workflow_id = execution.get("workflow_id")
|
||
workflow = workflow_manager.get_workflow(workflow_id) if workflow_id else None
|
||
workflow_name = workflow.get("name", workflow_id) if workflow else workflow_id
|
||
|
||
col1, col2, col3 = st.columns([3, 1, 1])
|
||
|
||
with col1:
|
||
st.markdown(f"**{workflow_name}**")
|
||
status = execution.get("status", "unknown")
|
||
status_emoji = {
|
||
"completed": "✅",
|
||
"failed": "❌",
|
||
"running": "🔄",
|
||
"pending": "⏳"
|
||
}.get(status, "❓")
|
||
st.caption(f"{status_emoji} {status} | 开始时间: {execution.get('started_at', 'N/A')[:19] if execution.get('started_at') else 'N/A'}")
|
||
|
||
with col2:
|
||
if execution.get("error"):
|
||
st.error("有错误")
|
||
else:
|
||
st.success("正常")
|
||
|
||
with col3:
|
||
if st.button("查看详情", key=f"view_exec_{execution.get('id')}"):
|
||
st.json(execution)
|
||
else:
|
||
st.info("暂无执行记录")
|
||
|
||
# =======================
|
||
# Tab8：GEO 资源库
|