feat: 初始化麦麦智农作物生长模拟平台
- 基于 PCSE/WOFOST 构建作物生长模拟平台 - 新增 Streamlit 可视化应用（app.py）与模拟引擎（simulator.py） - 支持潜在生产（PP）与水分限制生产（WLP）两种模拟模式 - 支持冬小麦、玉米、春大麦、马铃薯、冬油菜、向日葵 6 种作物 - 提供 LAI 动态、生物量积累、土壤水分、产量对比等可视化图表 - 新增 pyproject.toml、justfile、Dockerfile 等工程配置 - 完善 README.md 项目文档与 .gitignore 忽略规则
This commit is contained in:
zhenghu
444
app.py
Normal file
444
app.py
Normal file
@@ -0,0 +1,444 @@
|
||||
"""
|
||||
麦麦智农 - 基于 PCSE/WOFOST 的作物生长模拟平台
|
||||
"""
|
||||
|
||||
import streamlit as st
|
||||
import plotly.graph_objects as go
|
||||
from plotly.subplots import make_subplots
|
||||
|
||||
from simulator import run_wofost, run_multi_crop, list_available_crops, CROP_META
|
||||
|
||||
# ─── Page Config ────────────────────────────────────────────────────────────
|
||||
st.set_page_config(
|
||||
page_title="麦麦智农",
|
||||
page_icon="🌾",
|
||||
layout="wide",
|
||||
initial_sidebar_state="expanded",
|
||||
)
|
||||
|
||||
# ─── Custom CSS ──────────────────────────────────────────────────────────────
|
||||
st.markdown("""
|
||||
<style>
|
||||
:root {
|
||||
--soil: #7a5c44;
|
||||
--leaf: #4a7c59;
|
||||
--leaf-light: #6b9e75;
|
||||
--wheat: #d4a574;
|
||||
--cream: #faf8f3;
|
||||
--paper: #ffffff;
|
||||
--ink: #2c2c2c;
|
||||
--ink-muted: #5a5a5a;
|
||||
--border: #e5e0d5;
|
||||
--shadow: rgba(0,0,0,0.04);
|
||||
--warn: #c98c5b;
|
||||
}
|
||||
|
||||
html, body, [class*="css"] {
|
||||
font-family: "PingFang SC", "Microsoft YaHei", "Noto Sans SC", sans-serif;
|
||||
color: var(--ink);
|
||||
}
|
||||
|
||||
.stApp {
|
||||
background: var(--cream);
|
||||
}
|
||||
|
||||
/* Sidebar */
|
||||
[data-testid="stSidebar"] {
|
||||
background: #f5f2eb;
|
||||
border-right: 1px solid var(--border);
|
||||
}
|
||||
[data-testid="stSidebar"] .stSlider label,
|
||||
[data-testid="stSidebar"] .stNumberInput label,
|
||||
[data-testid="stSidebar"] .stSelectbox label {
|
||||
color: var(--soil) !important;
|
||||
font-size: 0.85rem;
|
||||
font-weight: 500;
|
||||
}
|
||||
|
||||
/* Metric cards */
|
||||
.metric-card {
|
||||
background: var(--paper);
|
||||
border: 1px solid var(--border);
|
||||
border-radius: 14px;
|
||||
padding: 18px 16px;
|
||||
text-align: center;
|
||||
box-shadow: 0 2px 10px var(--shadow);
|
||||
}
|
||||
.metric-value {
|
||||
font-size: 1.8rem;
|
||||
font-weight: 700;
|
||||
color: var(--leaf);
|
||||
line-height: 1.1;
|
||||
}
|
||||
.metric-unit {
|
||||
font-size: 0.75rem;
|
||||
color: var(--ink-muted);
|
||||
margin-top: 4px;
|
||||
}
|
||||
.metric-label {
|
||||
font-size: 0.9rem;
|
||||
color: var(--ink);
|
||||
margin-top: 8px;
|
||||
font-weight: 500;
|
||||
}
|
||||
|
||||
/* Section headers */
|
||||
.section-header {
|
||||
font-size: 0.95rem;
|
||||
font-weight: 600;
|
||||
color: var(--soil);
|
||||
padding-bottom: 8px;
|
||||
margin-bottom: 14px;
|
||||
margin-top: 22px;
|
||||
border-bottom: 1px solid var(--border);
|
||||
}
|
||||
|
||||
/* Hero */
|
||||
.hero-title {
|
||||
font-size: 1.7rem;
|
||||
font-weight: 700;
|
||||
color: var(--soil);
|
||||
line-height: 1.2;
|
||||
}
|
||||
.hero-sub {
|
||||
font-size: 0.85rem;
|
||||
color: var(--ink-muted);
|
||||
margin-top: 4px;
|
||||
}
|
||||
|
||||
/* Alert boxes */
|
||||
.alert-good {
|
||||
background: #f4faf5;
|
||||
border-left: 3px solid var(--leaf-light);
|
||||
border-radius: 0 8px 8px 0;
|
||||
padding: 12px 14px;
|
||||
margin: 8px 0;
|
||||
font-size: 0.9rem;
|
||||
color: var(--ink);
|
||||
}
|
||||
.alert-warn {
|
||||
background: #fdf9f3;
|
||||
border-left: 3px solid var(--wheat);
|
||||
border-radius: 0 8px 8px 0;
|
||||
padding: 12px 14px;
|
||||
margin: 8px 0;
|
||||
font-size: 0.9rem;
|
||||
color: var(--ink);
|
||||
}
|
||||
|
||||
/* Sidebar title */
|
||||
.sidebar-title {
|
||||
font-size: 1.15rem;
|
||||
font-weight: 700;
|
||||
color: var(--soil);
|
||||
margin-bottom: 2px;
|
||||
}
|
||||
.sidebar-sub {
|
||||
font-size: 0.75rem;
|
||||
color: var(--ink-muted);
|
||||
margin-bottom: 12px;
|
||||
}
|
||||
|
||||
/* Streamlit overrides */
|
||||
.stButton > button {
|
||||
border-radius: 10px !important;
|
||||
background: var(--leaf) !important;
|
||||
border: none !important;
|
||||
color: #fff !important;
|
||||
}
|
||||
.stButton > button:hover {
|
||||
background: var(--leaf-light) !important;
|
||||
}
|
||||
|
||||
/* Info panel */
|
||||
.info-panel {
|
||||
background: var(--paper);
|
||||
border: 1px solid var(--border);
|
||||
border-radius: 12px;
|
||||
padding: 14px 16px;
|
||||
font-size: 0.85rem;
|
||||
color: var(--ink-muted);
|
||||
line-height: 1.6;
|
||||
}
|
||||
</style>
|
||||
""", unsafe_allow_html=True)
|
||||
|
||||
|
||||
# ─── Sidebar Inputs ──────────────────────────────────────────────────────────
|
||||
with st.sidebar:
|
||||
st.markdown('<div class="sidebar-title">🌾 麦麦智农</div>', unsafe_allow_html=True)
|
||||
st.markdown('<div class="sidebar-sub">基于 PCSE/WOFOST 的真实作物模拟</div>', unsafe_allow_html=True)
|
||||
st.markdown("<hr style='border:none;border-top:1px solid var(--border);margin:12px 0;'>", unsafe_allow_html=True)
|
||||
|
||||
st.markdown('<div class="section-header" style="margin-top:0">🗺 模拟地点</div>', unsafe_allow_html=True)
|
||||
st.markdown("""
|
||||
<div class="info-panel" style="margin-bottom:14px;">
|
||||
<b>西班牙 · 埃斯特雷马杜拉</b><br>
|
||||
Grid 31031 (37.64°N, 6.09°W)
|
||||
</div>
|
||||
""", unsafe_allow_html=True)
|
||||
|
||||
st.markdown('<div class="section-header">📅 模拟参数</div>', unsafe_allow_html=True)
|
||||
year = st.selectbox("年份", [2000], index=0, help="Demo 数据库当前仅提供 2000 年数据")
|
||||
|
||||
crops = list_available_crops(grid_no=31031, year=year)
|
||||
crop_options = {c["crop_no"]: f"{c['emoji']} {c['name']}" for c in crops}
|
||||
selected_crop_no = st.selectbox("作物", list(crop_options.keys()), format_func=lambda k: crop_options[k])
|
||||
|
||||
mode = st.radio("生产模式", ["pp", "wlp"], format_func=lambda m: {"pp": "潜在生产 (PP)", "wlp": "水分限制生产 (WLP)"}[m])
|
||||
|
||||
st.markdown('<div class="section-header">🌱 种植规模</div>', unsafe_allow_html=True)
|
||||
area = st.number_input("种植面积 (公顷)", min_value=0.1, max_value=10000.0, value=100.0, step=10.0)
|
||||
|
||||
|
||||
# ─── Run Simulation ──────────────────────────────────────────────────────────
|
||||
@st.cache_data(show_spinner=False)
|
||||
def cached_run(crop_no, year, mode):
|
||||
return run_wofost(crop_no=crop_no, year=year, mode=mode)
|
||||
|
||||
|
||||
@st.cache_data(show_spinner=False)
|
||||
def cached_multi_crop(year, mode):
|
||||
return run_multi_crop(year=year, mode=mode)
|
||||
|
||||
|
||||
with st.spinner("正在运行 WOFOST 作物模拟，请稍候..."):
|
||||
result = cached_run(selected_crop_no, year, mode)
|
||||
multi_crop_result = cached_multi_crop(year, mode)
|
||||
|
||||
meta = result["meta"]
|
||||
summary = result["summary"]
|
||||
df = result["df"]
|
||||
crop_info = CROP_META.get(selected_crop_no, {"name": "未知作物", "emoji": "🌱", "color": "#888"})
|
||||
|
||||
# 单位换算
|
||||
twso = summary["twso"] if summary["twso"] is not None else 0.0
|
||||
tagp = summary["tagp"] if summary["tagp"] is not None else 0.0
|
||||
max_lai = summary["max_lai"] if summary["max_lai"] is not None else 0.0
|
||||
total_twso_tons = twso * area / 1000.0
|
||||
|
||||
# ─── Main Layout ─────────────────────────────────────────────────────────────
|
||||
st.markdown(f"""
|
||||
<div style="display:flex; align-items:baseline; gap:12px; margin-bottom:4px;">
|
||||
<div class="hero-title">麦麦智农 · {crop_info['emoji']} {crop_info['name']} 生长模拟</div>
|
||||
</div>
|
||||
<div class="hero-sub">
|
||||
{meta['year']} 年 · {meta['mode_label']} · 基于 WOFOST 7.2 真实作物模型
|
||||
</div>
|
||||
""", unsafe_allow_html=True)
|
||||
|
||||
st.markdown("<br>", unsafe_allow_html=True)
|
||||
|
||||
# KPI row
|
||||
k1, k2, k3, k4 = st.columns(4)
|
||||
with k1:
|
||||
st.markdown(f"""
|
||||
<div class="metric-card">
|
||||
<div class="metric-value">{twso:,.0f}</div>
|
||||
<div class="metric-unit">kg / 公顷</div>
|
||||
<div class="metric-label">🌾 经济产量 (TWSO)</div>
|
||||
</div>""", unsafe_allow_html=True)
|
||||
with k2:
|
||||
st.markdown(f"""
|
||||
<div class="metric-card">
|
||||
<div class="metric-value">{tagp:,.0f}</div>
|
||||
<div class="metric-unit">kg / 公顷</div>
|
||||
<div class="metric-label">🌿 总生物量 (TAGP)</div>
|
||||
</div>""", unsafe_allow_html=True)
|
||||
with k3:
|
||||
st.markdown(f"""
|
||||
<div class="metric-card">
|
||||
<div class="metric-value">{max_lai:.2f}</div>
|
||||
<div class="metric-unit">最大叶面积指数</div>
|
||||
<div class="metric-label">☘️ LAI max</div>
|
||||
</div>""", unsafe_allow_html=True)
|
||||
with k4:
|
||||
st.markdown(f"""
|
||||
<div class="metric-card">
|
||||
<div class="metric-value">{total_twso_tons:,.1f}</div>
|
||||
<div class="metric-unit">吨 / 总产量</div>
|
||||
<div class="metric-label">📦 {area:.0f} 公顷总产</div>
|
||||
</div>""", unsafe_allow_html=True)
|
||||
|
||||
st.markdown("<br>", unsafe_allow_html=True)
|
||||
|
||||
# ─── Charts Row 1 ────────────────────────────────────────────────────────────
|
||||
chart_left, chart_right = st.columns(2)
|
||||
|
||||
with chart_left:
|
||||
st.markdown('<div class="section-header">📈 叶面积指数 (LAI) 动态</div>', unsafe_allow_html=True)
|
||||
if not df.empty and "LAI" in df.columns:
|
||||
fig_lai = go.Figure()
|
||||
fig_lai.add_trace(go.Scatter(
|
||||
x=df.index, y=df["LAI"],
|
||||
mode="lines",
|
||||
line=dict(color=crop_info["color"], width=2.5),
|
||||
fill="tozeroy",
|
||||
fillcolor=crop_info["color"] + "1A",
|
||||
name="LAI",
|
||||
))
|
||||
# 添加生育期标记
|
||||
ms = summary.get("milestones", {})
|
||||
if "flowering" in ms:
|
||||
fig_lai.add_vline(x=ms["flowering"], line=dict(color="#d4a574", width=1.5, dash="dot"),
|
||||
annotation_text="开花", annotation_font_color="#7c5e42",
|
||||
annotation_position="top left")
|
||||
if "maturity" in ms:
|
||||
fig_lai.add_vline(x=ms["maturity"], line=dict(color="#7c5e42", width=1.5, dash="dash"),
|
||||
annotation_text="成熟", annotation_font_color="#7c5e42",
|
||||
annotation_position="top right")
|
||||
fig_lai.update_layout(
|
||||
xaxis=dict(title="日期", color="#5a5a5a", gridcolor="rgba(0,0,0,0.06)"),
|
||||
yaxis=dict(title="LAI", color="#5a5a5a", gridcolor="rgba(0,0,0,0.06)"),
|
||||
paper_bgcolor="rgba(0,0,0,0)", plot_bgcolor="rgba(0,0,0,0)",
|
||||
font=dict(color="#2c2c2c", size=10),
|
||||
margin=dict(t=20, b=36, l=50, r=20), height=280,
|
||||
showlegend=False,
|
||||
)
|
||||
st.plotly_chart(fig_lai, use_container_width=True)
|
||||
else:
|
||||
st.info("暂无 LAI 数据")
|
||||
|
||||
with chart_right:
|
||||
st.markdown('<div class="section-header">📊 生物量与产量积累</div>', unsafe_allow_html=True)
|
||||
if not df.empty and "TAGP" in df.columns:
|
||||
fig_bio = go.Figure()
|
||||
fig_bio.add_trace(go.Scatter(
|
||||
x=df.index, y=df["TAGP"],
|
||||
mode="lines",
|
||||
line=dict(color="#7a9e7e", width=2.5),
|
||||
name="TAGP",
|
||||
))
|
||||
if "TWSO" in df.columns:
|
||||
fig_bio.add_trace(go.Scatter(
|
||||
x=df.index, y=df["TWSO"],
|
||||
mode="lines",
|
||||
line=dict(color=crop_info["color"], width=2.5),
|
||||
name="TWSO",
|
||||
))
|
||||
ms = summary.get("milestones", {})
|
||||
if "flowering" in ms:
|
||||
fig_bio.add_vline(x=ms["flowering"], line=dict(color="#d4a574", width=1.5, dash="dot"))
|
||||
if "maturity" in ms:
|
||||
fig_bio.add_vline(x=ms["maturity"], line=dict(color="#7c5e42", width=1.5, dash="dash"))
|
||||
fig_bio.update_layout(
|
||||
xaxis=dict(title="日期", color="#5a5a5a", gridcolor="rgba(0,0,0,0.06)"),
|
||||
yaxis=dict(title="干物质 (kg/ha)", color="#5a5a5a", gridcolor="rgba(0,0,0,0.06)"),
|
||||
paper_bgcolor="rgba(0,0,0,0)", plot_bgcolor="rgba(0,0,0,0)",
|
||||
font=dict(color="#2c2c2c", size=10),
|
||||
margin=dict(t=20, b=36, l=50, r=20), height=280,
|
||||
legend=dict(orientation="h", y=-0.18, font=dict(size=10)),
|
||||
)
|
||||
st.plotly_chart(fig_bio, use_container_width=True)
|
||||
else:
|
||||
st.info("暂无生物量数据")
|
||||
|
||||
# ─── Charts Row 2 ────────────────────────────────────────────────────────────
|
||||
chart2_left, chart2_right = st.columns(2)
|
||||
|
||||
with chart2_left:
|
||||
st.markdown('<div class="section-header">💧 土壤水分动态</div>', unsafe_allow_html=True)
|
||||
if not df.empty and "SM" in df.columns:
|
||||
fig_sm = go.Figure()
|
||||
fig_sm.add_trace(go.Scatter(
|
||||
x=df.index, y=df["SM"],
|
||||
mode="lines",
|
||||
line=dict(color="#5a8f9e", width=2.5),
|
||||
fill="tozeroy",
|
||||
fillcolor="rgba(90, 143, 158, 0.1)",
|
||||
name="SM",
|
||||
))
|
||||
fig_sm.update_layout(
|
||||
xaxis=dict(title="日期", color="#5a5a5a", gridcolor="rgba(0,0,0,0.06)"),
|
||||
yaxis=dict(title="土壤含水量 (cm³/cm³)", color="#5a5a5a", gridcolor="rgba(0,0,0,0.06)"),
|
||||
paper_bgcolor="rgba(0,0,0,0)", plot_bgcolor="rgba(0,0,0,0)",
|
||||
font=dict(color="#2c2c2c", size=10),
|
||||
margin=dict(t=20, b=36, l=50, r=20), height=260,
|
||||
showlegend=False,
|
||||
)
|
||||
st.plotly_chart(fig_sm, use_container_width=True)
|
||||
else:
|
||||
st.info("暂无土壤水分数据")
|
||||
|
||||
with chart2_right:
|
||||
st.markdown('<div class="section-header">🏅 同一年份作物产量对比</div>', unsafe_allow_html=True)
|
||||
if multi_crop_result:
|
||||
names = [f"{CROP_META.get(r['crop_no'], {}).get('emoji', '🌱')} {r['name']}" for r in multi_crop_result]
|
||||
twsos = [r["twso"] if r["twso"] is not None else 0.0 for r in multi_crop_result]
|
||||
colors = [CROP_META.get(r["crop_no"], {}).get("color", "#888") for r in multi_crop_result]
|
||||
fig_bar = go.Figure()
|
||||
fig_bar.add_trace(go.Bar(
|
||||
x=names, y=twsos,
|
||||
marker=dict(color=colors, opacity=0.85,
|
||||
line=dict(color="rgba(0,0,0,0.08)", width=1)),
|
||||
text=[f"{v:,.0f}" for v in twsos],
|
||||
textposition="outside",
|
||||
textfont=dict(color="#5a5a5a", size=10),
|
||||
))
|
||||
fig_bar.update_layout(
|
||||
xaxis=dict(color="#5a5a5a", gridcolor="rgba(0,0,0,0.04)"),
|
||||
yaxis=dict(title="经济产量 (kg/ha)", color="#5a5a5a", gridcolor="rgba(0,0,0,0.06)"),
|
||||
paper_bgcolor="rgba(0,0,0,0)", plot_bgcolor="rgba(0,0,0,0)",
|
||||
font=dict(color="#2c2c2c", size=11),
|
||||
margin=dict(t=20, b=30, l=60, r=20), height=260,
|
||||
showlegend=False,
|
||||
)
|
||||
st.plotly_chart(fig_bar, use_container_width=True)
|
||||
else:
|
||||
st.info("暂无对比数据")
|
||||
|
||||
# ─── Advisory Panel ───────────────────────────────────────────────────────────
|
||||
st.markdown('<div class="section-header">💡 农艺建议与生育期</div>', unsafe_allow_html=True)
|
||||
adv1, adv2 = st.columns(2)
|
||||
|
||||
with adv1:
|
||||
advisories = []
|
||||
if twso > 6000:
|
||||
advisories.append(("good", f"{crop_info['name']} 模拟产量达到 {twso:,.0f} kg/ha，属于高产水平，气候与土壤条件匹配良好。"))
|
||||
elif twso > 3000:
|
||||
advisories.append(("good", f"{crop_info['name']} 模拟产量为 {twso:,.0f} kg/ha，处于中等水平，可通过优化水肥管理进一步提升。"))
|
||||
else:
|
||||
advisories.append(("warn", f"{crop_info['name']} 模拟产量仅 {twso:,.0f} kg/ha，建议检查品种适宜性或水分胁迫情况。"))
|
||||
|
||||
if mode == "wlp":
|
||||
if not df.empty and "SM" in df.columns:
|
||||
sm_min = df["SM"].min()
|
||||
if sm_min < 0.15:
|
||||
advisories.append(("warn", f"模拟期间土壤水分最低降至 {sm_min:.3f}，出现明显水分胁迫，建议评估灌溉方案。"))
|
||||
else:
|
||||
advisories.append(("good", "模拟期间土壤水分状况总体良好，未出现极端干旱胁迫。"))
|
||||
else:
|
||||
advisories.append(("good", "当前为潜在生产模式，结果反映理想水肥条件下的产量上限。"))
|
||||
|
||||
if max_lai > 5:
|
||||
advisories.append(("good", f"最大 LAI 达到 {max_lai:.2f}，冠层覆盖充分，光能截获效率高。"))
|
||||
elif max_lai < 2:
|
||||
advisories.append(("warn", f"最大 LAI 仅 {max_lai:.2f}，冠层发育不足，可能存在播期或品种问题。"))
|
||||
|
||||
for typ, msg in advisories:
|
||||
css_class = "alert-good" if typ == "good" else "alert-warn"
|
||||
st.markdown(f'<div class="{css_class}">{msg}</div>', unsafe_allow_html=True)
|
||||
|
||||
with adv2:
|
||||
ms = summary.get("milestones", {})
|
||||
if ms:
|
||||
st.markdown(f"""
|
||||
<div class="info-panel">
|
||||
<b>生育期里程碑</b><br>
|
||||
播种/出苗：{ms.get('start', '—')}<br>
|
||||
开花期：{ms.get('flowering', '—')}<br>
|
||||
成熟期：{ms.get('maturity', '—')}<br>
|
||||
收获/结束：{ms.get('end', '—')}<br>
|
||||
<span style="color:#999;">生育期天数：{summary.get('duration', '—')} 天</span>
|
||||
</div>
|
||||
""", unsafe_allow_html=True)
|
||||
else:
|
||||
st.markdown('<div class="info-panel">暂无生育期数据</div>', unsafe_allow_html=True)
|
||||
|
||||
# ─── Footer ───────────────────────────────────────────────────────────────────
|
||||
st.markdown("<br>", unsafe_allow_html=True)
|
||||
st.markdown("""
|
||||
<div style="text-align:center; font-size:0.78rem; color:#aaa; padding:14px; border-top:1px solid #e5e0d5;">
|
||||
麦麦智农 · 基于 PCSE/WOFOST 真实作物生长模型 · 结果仅供参考
|
||||
</div>
|
||||
""", unsafe_allow_html=True)
|
||||
Reference in New Issue
Block a user