添加产品规格文档并优化项目结构
Made-with: Cursor
This commit is contained in:
刘国栋
@@ -0,0 +1,446 @@
|
||||
# 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 资源库
|
||||
Reference in New Issue
Block a user