grouphyx/app.py

import os

import streamlit as st
from dotenv import load_dotenv

from agent import DebateManager
from datetime import datetime

# 确保在任何机器/任何启动方式下都能加载 .env（符合课程要求）
load_dotenv()

# 启动即检查 API Key，避免老师电脑运行时“无提示失败”
_api_key = os.getenv("OPENAI_API_KEY") or os.getenv("DEEPSEEK_API_KEY")
if not _api_key:
    st.error(
        "未检测到 API Key。\n\n"
        "请按课程要求在 multi_agent_workshop/ 目录下创建 .env 文件（可参考 .env.example），并设置：\n"
        "- OPENAI_API_KEY=sk-xxxxxx（推荐）\n"
        "或\n"
        "- DEEPSEEK_API_KEY=sk-xxxxxx\n\n"
        "然后重新运行：uv run streamlit run app.py"
    )
    st.stop()

from ui.rendering import render_decision_summary

# 从 ui.ui_components 模块导入所需的函数
from ui.ui_components import (
    local_css,
    show_welcome_guide,
    show_role_descriptions,
    show_example_topic,
    validate_input,
    show_progress_indicator,
    show_empty_state,
    show_debate_summary,
    show_debate_timeline,
    show_debate_comparison,
)

# 读取本地 CSS（原先大量内联 CSS 已经在 style.css，此处仅负责注入）
local_css("style.css")

st.set_page_config(
    page_title="多Agent决策工作坊",
    page_icon="🤖",
    layout="wide",
    initial_sidebar_state="expanded",
)


def save_debate_history(topic, selected_agents, debate_rounds, debate_history, decision_points):
    """保存辩论历史到session state"""
    if "saved_debates" not in st.session_state:
        st.session_state.saved_debates = []

    debate_record = {
        "id": len(st.session_state.saved_debates) + 1,
        "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
        "topic": topic[:100] + "..." if len(topic) > 100 else topic,
        "agents": selected_agents,
        "rounds": debate_rounds,
        "debate_history": debate_history,
        "decision_points": decision_points,
    }

    st.session_state.saved_debates.append(debate_record)
    return debate_record


def show_saved_debates(role_descriptions):
    """显示保存的辩论记录"""
    if "saved_debates" not in st.session_state or not st.session_state.saved_debates:
        return

    st.markdown("---")
    st.subheader("💾 历史记录")

    for record in reversed(st.session_state.saved_debates[-5:]):
        with st.expander(f"📅 {record['timestamp']} - {record['topic']}", expanded=False):
            col1, col2, col3 = st.columns(3)

            with col1:
                st.markdown(f"**参与角色**：{len(record['agents'])} 个")
            with col2:
                st.markdown(f"**评审轮次**：{record['rounds']} 轮")
            with col3:
                st.markdown(
                    f"**观点数量**：{sum(len(r.get('opinions', {})) for r in record['debate_history'][1:])} 条"
                )

            if st.button(f"查看详情", key=f"view_{record['id']}", use_container_width=True):
                st.session_state.debate_history = record["debate_history"]
                st.session_state.decision_points = record["decision_points"]
                st.rerun()


with st.sidebar:
    st.markdown(
        """
    <div style="text-align: center; padding: 20px 0; border-bottom: 1px solid #E2E8F0; margin-bottom: 24px;">
        <div style="font-size: 48px; margin-bottom: 8px;">🤖</div>
        <h2 style="margin: 0; color: #1E293B; font-size: 1.5rem;">多Agent决策工作坊</h2>
        <p style="margin: 8px 0 0 0; color: #64748B; font-size: 0.875rem;">智能辅助决策系统</p>
    </div>
        """,
        unsafe_allow_html=True,
    )

    st.markdown("---")

    st.subheader("📋 方案描述")

    with st.expander("🧩 一键插入结构化模板（推荐）", expanded=False):
        st.markdown(
            """
如果你不知道怎么写方案，建议用下面模板，AI 输出会更稳定：

- 目标（要解决什么问题？）
- 目标用户（谁用？使用场景？）
- 核心功能（3-5 条）
- 约束条件（时间/预算/技术/合规等）
- 成功指标（如何衡量？）
- 你最担心的风险（可选）
            """
        )
        if st.button("📌 插入模板到输入框", use_container_width=True):
            st.session_state.topic = (
                "【目标】\n" 
                "\n" 
                "【目标用户/场景】\n" 
                "\n" 
                "【核心功能（3-5条）】\n" 
                "1. \n2. \n3. \n" 
                "\n" 
                "【约束条件】\n" 
                "- 时间：\n- 预算：\n- 技术：\n" 
                "\n" 
                "【成功指标】\n" 
                "\n" 
                "【已知风险/担忧（可选）】\n"
            )
            st.rerun()

    topic = st.text_area(
        "请输入需要评审的方案内容",
        placeholder="请尽量结构化描述：目标 / 用户 / 核心功能 / 约束 / 指标...",
        height=180,
        help="建议提供至少50字的详细描述，以获得更准确的分析结果",
    )

    if st.button("📝 使用示例方案", key="use_example"):
        st.session_state.topic = show_example_topic()
        st.rerun()

    if "topic" in st.session_state:
        topic = st.session_state.topic
        st.text_area("方案内容", value=topic, height=180, key="topic_display")

    st.markdown("---")

    st.subheader("👥 参与角色")
    role_descriptions = show_role_descriptions()

    # 快捷评审模式（更像产品，演示更丝滑）
    mode = st.radio(
        "评审模式",
        options=["快速（2角色）", "标准（3角色）", "全面（5角色）"],
        index=1,
        help="用于快速配置参与角色数量", 
    )

    if mode.startswith("快速"):
        default_roles = ["product_manager", "tech_expert"]
    elif mode.startswith("全面"):
        default_roles = list(role_descriptions.keys())
    else:
        default_roles = ["product_manager", "tech_expert", "user_representative"]

    with st.expander("查看角色说明（可选）", expanded=False):
        for role_key, role_info in role_descriptions.items():
            st.markdown(f"**{role_info['icon']} {role_info['name']}**：{role_info['focus']}  ")
            st.markdown(f"<span class='small-muted'>{role_info['description']}</span>", unsafe_allow_html=True)
            st.markdown("---")

    selected_agents = st.multiselect(
        "选择参与评审的角色",
        options=list(role_descriptions.keys()),
        format_func=lambda x: f"{role_descriptions[x]['icon']} {role_descriptions[x]['name']}",
        default=default_roles,
        help="建议至少选择3-4个角色以获得全面的评估",
    )

    if selected_agents:
        st.info(f"已选择 {len(selected_agents)} 个角色")

    st.markdown("---")

    st.subheader("🔄 评审轮次")

    # 单轮版本：固定 1 轮，减少运行时间与 API 调用次数
    debate_rounds = 1

    est_low = max(1, len(selected_agents)) * 2
    est_high = max(1, len(selected_agents)) * 4

    st.markdown(
        f"""
    <div style=\"padding: 12px; background: #F1F5F9; border-radius: 8px; margin-top: 8px;\">
        <div style=\"font-size: 0.875rem; color: #64748B; line-height: 1.6;\">
            <strong>💡 提示：</strong>
            <br>• 每个角色各输出一次观点
            <br>• 优点：更快、更省调用、更适合课堂演示
            <br><br>
            <strong>⏱ 预计耗时：</strong>约 {est_low} ~ {est_high} 秒（与网络和模型负载有关）
        </div>
    </div>
    """,
        unsafe_allow_html=True,
    )

    st.markdown("---")

    errors, warnings = validate_input(topic, selected_agents)

    if errors:
        for error in errors:
            st.error(error)

    if warnings:
        for warning in warnings:
            st.warning(warning)

    start_button = st.button(
        "🚀 开始评审",
        type="primary",
        disabled=not topic or not selected_agents or len(errors) > 0,
        use_container_width=True,
    )


st.title("🤖 多Agent决策工作坊")

show_welcome_guide()

if "debate_manager" not in st.session_state:
    st.session_state.debate_manager = DebateManager()

if "debate_history" not in st.session_state:
    st.session_state.debate_history = []

if "decision_points" not in st.session_state:
    st.session_state.decision_points = ""

if start_button:
    st.session_state.debate_manager.reset()

    show_progress_indicator(1, 4, "初始化评审环境")

    with st.spinner("正在初始化评审环境..."):
        for agent_name in selected_agents:
            st.session_state.debate_manager.add_agent(agent_name)

    show_progress_indicator(2, 4, "启动评审")

    with st.spinner("正在进行评审..."):
        debate_history = st.session_state.debate_manager.start_debate(topic, debate_rounds)
        st.session_state.debate_history = debate_history

    show_progress_indicator(3, 4, "生成决策要点")

    with st.spinner("正在生成决策要点..."):
        decision_points = st.session_state.debate_manager.generate_decision_points()
        st.session_state.decision_points = decision_points

    show_progress_indicator(4, 4, "完成")

    save_debate_history(topic, selected_agents, debate_rounds, debate_history, decision_points)

    st.success("✅ 评审完成！")
    st.balloons()


if st.session_state.debate_history:
    # 1) 先给“结论摘要”（首屏答案）
    if st.session_state.decision_points:
        st.subheader("✅ 一页结论摘要")

        # 产品化渲染：优先用结构化展示；若解析失败，仍保留原 Markdown 兜底（不删除原功能）
        try:
            render_decision_summary(st.session_state.decision_points)
        except Exception:
            st.markdown(
                f"""
            <div style="padding: 20px; background: #F0F9FF; border: 1px solid #BAE6FD; border-radius: 12px; border-left: 4px solid #3B82F6;">
                <div style="color: #1E293B; line-height: 1.75; white-space: pre-wrap;">{st.session_state.decision_points}</div>
            </div>
            """,
                unsafe_allow_html=True,
            )

        col1, col2 = st.columns(2)
        with col1:
            st.download_button(
                label="📄 下载结论（Markdown）",
                data=st.session_state.decision_points,
                file_name="decision_summary.md",
                mime="text/markdown",
                use_container_width=True,
            )
        with col2:
            if st.button("🔄 重新开始", use_container_width=True):
                st.session_state.debate_history = []
                st.session_state.decision_points = ""
                st.rerun()

    st.markdown("---")

    # 2) 再给简要统计
    show_debate_summary(st.session_state.debate_history, st.session_state.decision_points)

    # 3) 过程作为依据：默认折叠展示
    with st.expander("查看评审过程（依据）", expanded=False):
        show_debate_timeline(st.session_state.debate_history, role_descriptions)

        st.markdown("---")
        st.subheader("📊 详细辩论结果")
        show_debate_comparison(st.session_state.debate_history, role_descriptions)

    show_saved_debates(role_descriptions)
else:
    show_empty_state()

st.markdown("---")
st.markdown(
    """
<div style="padding: 16px; background: #F8FAFC; border: 1px solid #E2E8F0; border-radius: 8px;">
    <div style="display: flex; align-items: center; gap: 8px; margin-bottom: 8px;">
        <span style="font-size: 20px;">💡</span>
        <strong style="color: #1E293B;">使用提示</strong>
    </div>
    <div style="color: #64748B; line-height: 1.6; font-size: 0.95rem;">
        • 您可以随时调整方案内容与参与角色，然后重新开始评审<br>
        • 建议使用示例方案进行第一次测试，以了解系统功能<br>
        • 评审结果仅供参考，最终决策请结合实际情况和专业判断
    </div>
</div>
""",
    unsafe_allow_html=True,
)