import streamlit as st
import openai
import time

# --- 页面基础设置 ---
st.set_page_config(
    page_title="AI互掐",
    page_icon="🤖",
    layout="wide"
)

st.title("🤖 AI思辨大会")
st.caption(f"当前时间: {time.strftime('%Y-%m-%d %H:%M:%S')}")

# --- 侧边栏配置 ---
with st.sidebar:
    st.header("🔗 API 配置")
    # 使用st.secrets来更安全地管理密钥，如果没有则让用户输入
    try:
        default_api_key = st.secrets["api_key"]
        default_base_url = st.secrets["base_url"]
    except (FileNotFoundError, KeyError):
        default_api_key = "**" # **替换为One-api建立的令牌，以“sk-”开头
        default_base_url = "http://localhost:3000/v1"  # 默认为本地One-API地址

    api_key = st.text_input("API Key", value=default_api_key, type="password")
    base_url = st.text_input("API Base URL", value=default_base_url)

    st.header("🎭 辩手设置")
    topic = st.text_area("请输入辩论主题...", value="人工智能的发展是否应该受到更严格的全球性法规限制？", height=150)

    col1, col2 = st.columns(2)
    with col1:
        model_a = st.text_input("正方模型", value="**") # **替换为申请可用模型名称
    with col2:
        model_b = st.text_input("反方模型", value="**") # **替换为申请可用模型名称

    moderator_model = st.text_input("主持人模型", value="**") # **替换为申请可用模型名称
    rounds = st.slider("辩论回合数", 1, 5, 2)

# --- 初始化会话状态 ---
# st.session_state 是Streamlit提供的用于在用户多次交互之间保存变量的“内存”
if "messages" not in st.session_state:
    st.session_state.messages = []

# --- 显示历史消息 ---
# 每当用户操作导致页面刷新时，这段代码都会执行，从而把历史消息显示出来
for message in st.session_state.messages:
    with st.chat_message(message["role"], avatar=message.get("avatar")):
        st.markdown(message["content"])

# --- 主逻辑 ---
if st.sidebar.button("🚀 开始辩论"):
    if not api_key or not base_url or not topic:
        st.warning("请在侧边栏填写完整的API配置和辩论主题。")
    else:
        # 清空上一次的辩论记录
        st.session_state.messages = []

        # 创建API客户端
        client = openai.OpenAI(base_url=base_url, api_key=api_key)

        # 定义辩手信息
        debater_a = {"name": "正方", "model": model_a, "avatar": "👍"}
        debater_b = {"name": "反方", "model": model_b, "avatar": "👎"}
        moderator = {"name": "主持人", "model": moderator_model, "avatar": "👨‍⚖️"}

        # 显示辩题
        st.info(f"辩题：{topic}")

        # 构建辩论历史（用于传递给模型的上下文）
        conversation_history = []

        try:
            # --- 辩论开始，按轮次循环 ---
            for i in range(rounds):
                st.subheader(f"第 {i + 1} 回合", divider='rainbow')

                # --- 正方发言 ---
                # 在第一轮，正方进行开场陈词；后续轮次则进行回应。
                if i == 0:
                    prompt_a = f"你是一位名叫{debater_a['name']}的辩手。你的立场是 **支持** “{topic}”。这是辩论的开场陈词，请开始你的第一轮发言，要求清晰、有逻辑、有说服力。"
                else:
                    prompt_a = f"你是一位名叫{debater_a['name']}的辩手。你的立场是 **支持** “{topic}”。以下是到目前为止的辩论历史：\n{conversation_history}\n\n现在轮到你在第{i + 1}回合发言，请直接反驳对方的观点并提出你的新论据。"

                with st.chat_message(debater_a['name'], avatar=debater_a['avatar']):
                    with st.spinner("正方正在思考..."):
                        response = client.chat.completions.create(
                            model=debater_a['model'],
                            messages=[{"role": "user", "content": prompt_a}]
                        )
                        a_text = response.choices[0].message.content
                        st.markdown(a_text)
                st.session_state.messages.append(
                    {"role": debater_a['name'], "content": a_text, "avatar": debater_a['avatar']})
                conversation_history.append({"role": debater_a['name'], "content": a_text})

                # --- 反方发言 ---
                prompt_b = f"你是一位名叫{debater_b['name']}的辩手。你的立场是 **反对** “{topic}”。以下是到目前为止的辩论历史：\n{conversation_history}\n\n现在轮到你在第{i + 1}回合发言，请直接反驳对方刚才的观点。"
                with st.chat_message(debater_b['name'], avatar=debater_b['avatar']):
                    with st.spinner("反方正在思考..."):
                        response = client.chat.completions.create(
                            model=debater_b['model'],
                            messages=[{"role": "user", "content": prompt_b}]
                        )
                        b_text = response.choices[0].message.content
                        st.markdown(b_text)
                st.session_state.messages.append(
                    {"role": debater_b['name'], "content": b_text, "avatar": debater_b['avatar']})
                conversation_history.append({"role": debater_b['name'], "content": b_text})

                # --- 主持人总结本回合 ---
                prompt_moderator = f"你是一位中立、客观的辩论主持人。请对刚才第{i + 1}回合中，正反双方的交锋进行简要、无偏见的总结。这是刚才的对话内容：\n{conversation_history[-2:]}"
                with st.chat_message(moderator['name'], avatar=moderator['avatar']):
                    with st.spinner("主持人正在总结..."):
                        response = client.chat.completions.create(
                            model=moderator['model'],
                            messages=[{"role": "user", "content": prompt_moderator}]
                        )
                        mod_text = response.choices[0].message.content
                        st.markdown(mod_text)
                st.session_state.messages.append(
                    {"role": moderator['name'], "content": mod_text, "avatar": moderator['avatar']})

            st.success("辩论结束！")

        except Exception as e:
            st.error(f"发生错误: {e}")