"""
农技智能问答
基于大模型的农业技术知识问答应用
"""

import json

import httpx
import streamlit as st
from config import CHAT_API_URL, CHAT_MODEL, HEADERS

# ─── Page Config ────────────────────────────────────────────────────────────
st.set_page_config(
    page_title="农技问答",
    page_icon="🌾",
    layout="centered",
    initial_sidebar_state="collapsed",
)

# ─── Minimal CSS ─────────────────────────────────────────────────────────────
st.markdown("""
<style>
html, body, [class*="css"] {
    font-family: "PingFang SC", "Microsoft YaHei", "Noto Sans SC", sans-serif;
}
</style>
""", unsafe_allow_html=True)


# ─── Quick Questions ─────────────────────────────────────────────────────────
QUICK_QUESTIONS = [
    "水稻稻瘟病怎么防治？",
    "小麦锈病怎么处理？",
    "玉米什么时候浇水最合适？",
    "种榴莲需要注意什么？",
]


# ─── Header ──────────────────────────────────────────────────────────────────
st.title("🌾 农技问答")
st.caption("有农业问题，随时来问")
st.markdown("<br>", unsafe_allow_html=True)

# ─── Quick Questions ─────────────────────────────────────────────────────────
with st.container(border=False):
    cols = st.columns(len(QUICK_QUESTIONS))
    for i, q in enumerate(QUICK_QUESTIONS):
        with cols[i]:
            if st.button(q, key=f"chip_{q}", use_container_width=True):
                st.session_state.user_input = q
                st.rerun()

st.markdown("<br>", unsafe_allow_html=True)

# ─── Input ───────────────────────────────────────────────────────────────────
user_input = st.text_area(
    "请输入您的问题",
    value=st.session_state.get("user_input", ""),
    height=110,
    placeholder="例如：水稻稻瘟病怎么防治？",
    label_visibility="collapsed",
)
submitted = st.button("发送", type="primary")

# ─── Settings Panel ──────────────────────────────────────────────────────────
with st.expander("⚙️ 模型设置"):
    col_t, col_p, col_c = st.columns(3)
    with col_t:
        temperature = st.slider("Temperature", 0.0, 1.0, 0.7, 0.1)
    with col_p:
        top_p = st.slider("Top P", 0.0, 1.0, 0.8, 0.1)
    with col_c:
        enable_thinking = st.checkbox("显示推理过程", value=True)

# ─── Request & Stream ────────────────────────────────────────────────────────
if submitted and user_input.strip():
    try:
        payload = {
            "model": CHAT_MODEL,
            "messages": [{"role": "user", "content": user_input.strip()}],
            "temperature": temperature,
            "top_p": top_p,
            "presence_penalty": 1.5,
            "chat_template_kwargs": {"enable_thinking": enable_thinking},
            "stream": True,
        }

        thinking_placeholder = st.empty()
        answer_placeholder = st.empty()

        full_reasoning = ""
        full_content = ""

        with httpx.stream(
            "POST", CHAT_API_URL, headers=HEADERS, json=payload, timeout=120
        ) as resp:
            resp.raise_for_status()
            for line in resp.iter_lines():
                if not line.startswith("data: "):
                    continue
                data_str = line[6:]
                if data_str == "[DONE]":
                    break
                try:
                    chunk = json.loads(data_str)
                except json.JSONDecodeError:
                    continue

                delta = chunk.get("choices", [{}])[0].get("delta", {})

                reasoning_piece = delta.get("reasoning_content", "")
                if reasoning_piece:
                    full_reasoning += reasoning_piece
                    if enable_thinking:
                        thinking_placeholder.info(
                            "**正在思考...**\n\n" + full_reasoning
                        )

                content_piece = delta.get("content", "")
                if content_piece:
                    full_content += content_piece
                    with answer_placeholder.container(border=True):
                        st.markdown("### 🌱 回答")
                        st.markdown(full_content)

        if full_reasoning and enable_thinking:
            thinking_placeholder.empty()
            with st.expander("查看推理过程"):
                st.markdown(full_reasoning)

    except httpx.HTTPStatusError as e:
        st.error(f"请求失败 (HTTP {e.response.status_code}): {e.response.text}")
    except Exception as e:
        st.error(f"请求异常: {e}")

# ─── Footer ───────────────────────────────────────────────────────────────────
st.divider()
st.caption("农业技术知识问答 · 仅供参考")