Create app.py

app.py

@@ -0,0 +1,199 @@
+import io
+from dataclasses import dataclass
+import math
+
+import gradio as gr
+import matplotlib.pyplot as plt
+
+# ---------------------------
+# Small-Hydro 4-Mode Dashboard (Gradio)
+# ---------------------------
+
+RHO = 1000.0         # kg/m^3
+G   = 9.80665        # m/s^2
+PF_TARGET = 0.98
+
+def efficiency_from_relative_flow(r: float) -> float:
+    """간이 효율곡선 (r = Q/Qn). 구간 보간."""
+    r = max(0.0, min(1.2, r))
+    curve = [
+        (0.00, 0.00),
+        (0.10, 0.40),
+        (0.30, 0.60),
+        (0.50, 0.75),
+        (0.70, 0.85),
+        (0.90, 0.90),
+        (1.00, 0.91),
+        (1.20, 0.88),
+    ]
+    for (x1,y1),(x2,y2) in zip(curve[:-1], curve[1:]):
+        if x1 <= r <= x2:
+            t = 0.0 if x2 == x1 else (r-x1)/(x2-x1)
+            return y1 + t*(y2-y1)
+    return 0.0
+
+def electrical_power_kw(Q_in: float, head_m: float, eta: float) -> float:
+    """P[kW] = rho*g*Q*H*eta/1000"""
+    if Q_in <= 0 or head_m <= 0 or eta <= 0:
+        return 0.0
+    return (RHO * G * Q_in * head_m * eta) / 1000.0
+
+def decide_mode(Q_in: float, q_low: float, q_high: float,
+                rain_mm: float, dp_kpa: float,
+                rain_thr: float, dp_limit: float,
+                emergency: bool) -> str:
+    if emergency:
+        return "Emergency"
+    if Q_in < q_low:
+        return "Low-Flow"
+    if Q_in > q_high or dp_kpa > dp_limit or rain_mm >= rain_thr:
+        return "Flood"
+    return "Normal"
+
+def mode_policy(mode: str, Q_in: float, Qn: float, Hn: float):
+    r   = min(Q_in, Qn)/Qn if Qn>0 else 0.0
+    eta = efficiency_from_relative_flow(r)
+    p_kw = electrical_power_kw(min(Q_in, Qn), Hn, eta)
+
+    gate = round(min(100.0, max(10.0, r*100.0)), 1)
+    vane = gate
+
+    curtail = False
+    bypass  = False
+    dump    = False
+    soc = (55, 70)
+
+    if mode == "Flood":
+        curtail = True; bypass = True; dump = True
+        vane = max(20.0, vane-20.0); gate = vane
+        p_kw *= 0.6
+        soc = (35, 45)
+    elif mode == "Low-Flow":
+        vane = max(15.0, min(55.0, vane)); gate = vane
+        p_kw *= 0.85
+        soc = (45, 65)
+    elif mode == "Emergency":
+        curtail = True; bypass = True; dump = False
+        vane = 0.0; gate = 0.0; p_kw = 0.0
+        soc = (30, 50)
+
+    return {
+        "eta": eta,
+        "p_kw": p_kw,
+        "gate_pct": gate,
+        "vane_pct": vane,
+        "curtail": curtail,
+        "bypass": bypass,
+        "dump": dump,
+        "soc_low": soc[0],
+        "soc_high": soc[1],
+    }
+
+def plot_efficiency(Qn: float, Q_in: float):
+    """효율곡선 + 현재 운전점 이미지(PNG) 반환"""
+    rs = [i/100 for i in range(0, 121)]
+    etas = [efficiency_from_relative_flow(r) for r in rs]
+    r_now = min(Q_in, Qn)/Qn if Qn>0 else 0.0
+    eta_now = efficiency_from_relative_flow(r_now)
+
+    fig = plt.figure(figsize=(5,3.2))
+    plt.plot(rs, etas, linewidth=2)
+    plt.scatter([r_now], [eta_now], s=60)
+    plt.xlabel("Relative flow r = Q / Qn")
+    plt.ylabel("Efficiency η")
+    plt.title("Efficiency curve & current operating point")
+    plt.grid(True, alpha=0.3)
+    buf = io.BytesIO()
+    fig.tight_layout()
+    fig.savefig(buf, format="png", dpi=140)
+    plt.close(fig)
+    buf.seek(0)
+    return buf
+
+def plot_power_bar(p_kw: float):
+    """출력 막대 그래프 PNG"""
+    fig = plt.figure(figsize=(4.2,3.2))
+    plt.bar(["Estimated Power"], [p_kw])
+    plt.ylabel("kW")
+    plt.title("Estimated Electrical Power")
+    for idx, v in enumerate([p_kw]):
+        plt.text(idx, v, f"{v:.1f} kW", ha="center", va="bottom")
+    buf = io.BytesIO()
+    fig.tight_layout()
+    fig.savefig(buf, format="png", dpi=140)
+    plt.close(fig)
+    buf.seek(0)
+    return buf
+
+def badge_for_mode(mode: str) -> str:
+    emoji = {"Normal":"🟢", "Low-Flow":"🟡", "Flood":"🟠", "Emergency":"🔴"}.get(mode,"⚪")
+    return f"### {emoji} Mode: **{mode}**"
+
+def run(Q_in, rain_mm, dp_kpa, emergency, Hn, Qn, rain_thr, dp_limit):
+    try:
+        Q_in     = float(Q_in); rain_mm = float(rain_mm); dp_kpa = float(dp_kpa)
+        Hn       = float(Hn);   Qn      = float(Qn)
+        rain_thr = float(rain_thr); dp_limit = float(dp_limit)
+        emergency = bool(emergency)
+    except Exception:
+        return "⚠️ 입력 오류", None, None, None
+
+    if Qn<=0 or Hn<=0:
+        return "⚠️ 설계값(Hₙ, Qₙ)은 0보다 커야 합니다.", None, None, None
+
+    q_low, q_high = 0.4*Qn, 1.2*Qn
+    mode = decide_mode(Q_in, q_low, q_high, rain_mm, dp_kpa, rain_thr, dp_limit, emergency)
+    st = mode_policy(mode, Q_in, Qn, Hn)
+
+    md = []
+    md.append(badge_for_mode(mode))
+    md.append("")
+    md.append(f"- **유량 Q**: {Q_in:.2f} m³/s  |  **설계 Qₙ**: {Qn:.2f} m³/s (임계: {q_low:.2f} ~ {q_high:.2f})")
+    md.append(f"- **낙차 H**: {Hn:.2f} m  |  **강우예보**: {rain_mm:.1f} mm/24h  |  **ΔP**: {dp_kpa:.1f} kPa")
+    md.append("")
+    md.append("**추정/설정**")
+    md.append(f"- 효율 η: **{st['eta']:.3f}**")
+    md.append(f"- 출력: **{st['p_kw']:.1f} kW**")
+    md.append(f"- 가이드베인: **{st['vane_pct']:.1f}%**  |  취수문: **{st['gate_pct']:.1f}%**")
+    md.append(f"- 커테일: **{st['curtail']}**, 바이패스: **{st['bypass']}**, 덤프로드: **{st['dump']}**")
+    md.append(f"- ESS SoC 목표: **{st['soc_low']}–{st['soc_high']}%**  |  역률 목표: **{PF_TARGET}**")
+
+    eff_img = plot_efficiency(Qn, Q_in)
+    p_img   = plot_power_bar(st["p_kw"])
+    return "\n".join(md), eff_img, p_img, st["p_kw"]
+
+# ---------------------------
+# Gradio UI
+# ---------------------------
+with gr.Blocks(theme=gr.themes.Soft(), title="Small-Hydro Smart Modes Dashboard") as app:
+    gr.Markdown("## 🌊 소수력 스마트 운전모드 대시보드")
+
+    with gr.Row():
+        Q_in   = gr.Number(label="현재 유량 Q (m³/s)", value=3.0)
+        rain   = gr.Number(label="강우 예보 (mm/24h)", value=10)
+        dp     = gr.Number(label="트래시랙 ΔP (kPa)", value=1.0)
+        emer   = gr.Checkbox(label="비상 알람", value=False)
+
+    with gr.Accordion("고급 설정 (설계/임계값)", open=False):
+        with gr.Row():
+            Hn  = gr.Number(label="설계 낙차 Hₙ (m)", value=8.0)
+            Qn  = gr.Number(label="설계 유량 Qₙ (m³/s)", value=3.5)
+        with gr.Row():
+            rain_thr = gr.Number(label="홍수 판단 강우 임계 (mm/24h)", value=40.0)
+            dp_limit = gr.Number(label="트래시랙 ΔP 임계 (kPa)", value=3.0)
+
+    run_btn = gr.Button("시뮬레이션 실행", variant="primary")
+
+    out_md   = gr.Markdown()
+    eff_plot = gr.Image(label="효율곡선 & 현재 운전점", type="auto")
+    p_plot   = gr.Image(label="출력 추정 (kW)", type="auto")
+
+    run_btn.click(
+        fn=run,
+        inputs=[Q_in, rain, dp, emer, Hn, Qn, rain_thr, dp_limit],
+        outputs=[out_md, eff_plot, p_plot, gr.Number(visible=False)]
+    )
+
+# 로컬 테스트용
+if __name__ == "__main__":
+    app.launch()