Update app.py

app.py

@@ -1,137 +1,337 @@
+import time
 import gradio as gr
 
-# ------------------------------
-# 설계 상수
-# ------------------------------
-Hn = 8.0       # 설계 낙차 (m)
-Qn = 3.5       # 설계 유량 (m³/s)
-rain_thr = 40  # 고유량 판정 강수량 (mm/24h)
-dp_limit = 3.0 # 트래시랙 막힘 한계 (kPa)
-soc_limit = 90 # ESS 과충전 한계 (%)
-
-PF_TARGET = 0.95   # 목표 역률
-RHO = 1000         # 물 밀도 (kg/m³)
-G = 9.81           # 중력가속도 (m/s²)
-
-# ------------------------------
-# 출력 계산 함수
-# ------------------------------
-def electrical_power_kw(Q, H, eta):
-    return RHO * G * Q * H * eta / 1000.0
-
-# ------------------------------
-# 모드 판정 함수
-# ------------------------------
-def decide_mode(Q, rain, dp, soc):
-    # 정비 모드 최우선 (트래시랙 막힘, ESS 과충전)
-    if dp > dp_limit or soc >= soc_limit:
+# =========================
+# 설계/임계 기본값
+# =========================
+DEFAULTS = dict(
+    Hn=8.0,          # 설계 낙차 (m)
+    Qn=3.5,          # 설계 유량 (m³/s)
+    rain_thr=40.0,   # 고유량 판정 강수량 (mm/24h)
+    dp_limit=3.0,    # 트래시랙 막힘 경고 임계 (kPa) => 정비 모드
+    soc_limit=90.0,  # ESS 과충전 임계 (%) => 정비 모드
+    pf_target=0.95,  # 목표 역률
+    rho=1000.0,      # 물 밀도
+    g=9.80665        # 중력가속도
+)
+
+# =========================
+# 공용 함수
+# =========================
+def electrical_power_kw(Q, H, eta, rho=DEFAULTS["rho"], g=DEFAULTS["g"]):
+    if Q <= 0 or H <= 0 or eta <= 0: return 0.0
+    return (rho * g * Q * H * eta)/1000.0
+
+def efficiency_from_relative_flow(r):
+    # 간단 근사: r=Q/Qn (부분부하~정격에서 완만, 정격 넘어가면 소폭 하락)
+    r = max(0.0, min(1.3, float(r)))
+    pts = [
+        (0.00, 0.00), (0.10, 0.40), (0.30, 0.60), (0.50, 0.78),
+        (0.70, 0.86), (0.90, 0.90), (1.00, 0.91), (1.20, 0.88), (1.30, 0.85)
+    ]
+    for (x1,y1),(x2,y2) in zip(pts[:-1], pts[1:]):
+        if x1 <= r <= x2:
+            t = 0 if x2==x1 else (r-x1)/(x2-x1)
+            return y1 + t*(y2-y1)
+    return 0.0
+
+# =========================
+# 모드 결정(요청 기준)
+#  - 정비: ΔP > dp_limit OR SoC >= soc_limit  (최우선)
+#  - 저유량: Q < 0.4*Qn
+#  - 고유량: Q > 1.2*Qn OR rain >= rain_thr
+#  - 정상: 나머지
+# =========================
+def decide_mode(Q, Qn, rain, dp, soc, rain_thr, dp_limit, soc_limit):
+    if (dp is not None and dp > dp_limit) or (soc is not None and soc >= soc_limit):
         return "정비"
-    # 저유량
-    if Q < 0.4 * Qn:
+    if Q < 0.4*Qn:
         return "저유량"
-    # 고유량
-    if Q > 1.2 * Qn or rain >= rain_thr:
+    if (Q > 1.2*Qn) or (rain >= rain_thr):
         return "고유량"
-    # 정상
     return "정상"
 
-# ------------------------------
-# 모드별 동작 정책
-# ------------------------------
-def mode_policy(mode, Q, H):
+# =========================
+# 모드별 운전 정책
+# =========================
+def mode_policy(mode, Q, Qn, Hn):
+    r = Q/Qn if Qn>0 else 0.0
+    eta = efficiency_from_relative_flow(r)
+
     if mode == "정상":
-        eta = 0.90
-        p_kw = electrical_power_kw(Q, H, eta)
-        return {"eta": eta, "p_kw": p_kw, "desc": "정상 운전: 효율 최적점에서 발전"}
-    elif mode == "저유량":
-        eta = 0.75
-        p_kw = electrical_power_kw(Q, H, eta)
-        return {"eta": eta, "p_kw": p_kw, "desc": "저유량 모드: 출력 감소, 효율 저하"}
-    elif mode == "고유량":
-        eta = 0.70
-        p_kw = electrical_power_kw(Q, H, eta) * 0.6  # 커테일링
-        return {"eta": eta, "p_kw": p_kw, "desc": "고유량 모드: 출력 제한, 바이패스 병행"}
-    elif mode == "정비":
-        eta = 0.0
-        p_kw = 0.0
-        return {"eta": eta, "p_kw": p_kw, "desc": "정비 모드: 발전 중지, 정비 필요"}
-    else:
-        return {"eta": 0, "p_kw": 0, "desc": "알 수 없는 모드"}
-
-# ------------------------------
-# 시각화 함수
-# ------------------------------
-def donut(label, value, max_val, unit=""):
-    pct = int((value / max_val) * 100) if max_val > 0 else 0
-    if pct > 100: pct = 100
-    color = "#16a34a" if pct > 70 else "#facc15" if pct > 40 else "#dc2626"
+        eta_eff = max(eta, 0.88)
+        p = electrical_power_kw(Q, Hn, eta_eff)
+        return dict(eta=eta_eff, p_kw=p, note="정상 운전: 효율 최적점", vane="유량 비례", gate="유량 비례",
+                    curtail=False, bypass=False, service=False)
+
+    if mode == "저유량":
+        eta_eff = max(0.70, eta*0.95)
+        p = electrical_power_kw(Q, Hn, eta_eff)*0.90
+        return dict(eta=eta_eff, p_kw=p, note="저유량: 출력 축소·효율 최적화", vane="15–55%", gate="15–55%",
+                    curtail=False, bypass=False, service=False)
+
+    if mode == "고유량":
+        eta_eff = max(0.68, eta*0.9)
+        p = electrical_power_kw(Q, Hn, eta_eff)*0.60  # 커테일
+        return dict(eta=eta_eff, p_kw=p, note="고유량: 출력 제한·바이패스 병행", vane="각도 축소", gate="각도 축소",
+                    curtail=True, bypass=True, service=False)
+
+    if mode == "정비":
+        return dict(eta=0.0, p_kw=0.0, note="정비: 발전 중지·정비 필요(스크린/ESS)", vane="0%", gate="0%",
+                    curtail=True, bypass=True, service=True)
+
+    return dict(eta=0.0, p_kw=0.0, note="알 수 없는 모드", vane="-", gate="-",
+                curtail=False, bypass=False, service=False)
+
+# =========================
+# 뷰(UI) 구성 요소
+# =========================
+def mode_badge(mode:str)->str:
+    colors = {
+        "정상":   ("#22c55e","●"),
+        "저유량": ("#eab308","●"),
+        "고유량": ("#f97316","●"),
+        "정비":   ("#ef4444", None)
+    }
+    col, dot = colors.get(mode, ("#6b7280","●"))
+    if mode == "정비":
+        icon = """
+        <svg width="22" height="22" viewBox="0 0 64 64" style="vertical-align:middle">
+          <polygon points="32,6 58,54 6,54" fill="#ef4444" stroke="#b91c1c" stroke-width="3"/>
+          <rect x="30" y="22" width="4" height="18" fill="#ffffff" rx="2"/>
+          <circle cx="32" cy="46" r="3" fill="#ffffff"/>
+        </svg>
+        """
+        return f"""
+        <div style="padding:10px 14px;border-radius:14px;background:{col};
+                    color:#fff;font-weight:800;display:inline-flex;gap:10px;align-items:center;">
+          {icon}<span>정비 모드</span>
+        </div>"""
+    return f"""
+    <div style="padding:10px 14px;border-radius:14px;background:{col};
+                color:#fff;font-weight:800;display:inline-flex;gap:8px;align-items:center;">
+      <span style="font-size:18px">{dot}</span><span>운전 모드 : {mode}</span>
+    </div>"""
+
+def donut(label:str, value:float, maxv:float, unit:str=""):
+    maxv = max(1e-6, maxv)
+    pct = max(0.0, min(value/maxv, 1.0))
+    r = 42; C = 2*3.14159*r; offset = C*(1-pct)
+    return f"""
+    <div style="display:flex;flex-direction:column;align-items:center;gap:6px;">
+      <svg width="120" height="120" viewBox="0 0 120 120">
+        <circle cx="60" cy="60" r="{r}" stroke="#e5e7eb" stroke-width="12" fill="none"/>
+        <circle cx="60" cy="60" r="{r}" stroke="#3b82f6" stroke-width="12" fill="none"
+                stroke-dasharray="{C:.1f}" stroke-dashoffset="{offset:.1f}"
+                stroke-linecap="round" transform="rotate(-90 60 60)"/>
+        <text x="60" y="60" text-anchor="middle" dominant-baseline="middle"
+              font-size="16" font-weight="700">{value:.1f}{unit}</text>
+      </svg>
+      <div style="font-size:14px;color:#475569">{label}</div>
+    </div>
+    """
+
+def chip(label:str, on:bool, on_color="#10b981"):
+    bg = on_color if on else "#e5e7eb"
+    fg = "#fff" if on else "#111827"
+    text = "ON" if on else "OFF"
     return f"""
-    <div style="width:160px;height:160px;border-radius:50%;border:14px solid {color};
-                display:flex;align-items:center;justify-content:center;font-size:18px;
-                font-weight:bold;color:#111;">
-      <div style="text-align:center">
-        {label}<br>{value:.1f}{unit}<br><span style="font-size:14px;color:#666">{pct}%</span>
+    <div style="padding:6px 10px;border-radius:999px;background:{bg};color:{fg};
+                font-size:12px;font-weight:700;display:inline-block;">
+      {label}: {text}
+    </div>
+    """
+
+def gauge_bar(label:str, value:float, maxv:float, unit:str=""):
+    maxv = max(1e-6, maxv)
+    pct = int(round(max(0.0, min(value/maxv,1.0))*100))
+    return f"""
+    <div style="margin:4px 0 10px 0;">
+      <div style="font-size:13px;margin-bottom:4px">{label}: <b>{value:.1f}{unit}</b></div>
+      <div style="width:100%;height:10px;border-radius:7px;background:#e5e7eb">
+        <div style="width:{pct}%;height:10px;border-radius:7px;background:#6366f1"></div>
       </div>
     </div>
     """
 
-# ------------------------------
-# 실행 함수
-# ------------------------------
-def run(Q_in, rain_mm, dp_kpa, soc_perc):
-    try:
-        Q_in = float(Q_in)
-        rain_mm = float(rain_mm)
-        dp_kpa = float(dp_kpa)
-        soc_perc = float(soc_perc)
-    except:
-        return "⚠️ 입력 오류", "", ""
+# =========================
+# 핵심 실행 로직
+# =========================
+def _evaluate(Q_in, rain_mm, dp_kpa, soc_pct, Hn, Qn, rain_thr, dp_limit, soc_limit):
+    mode = decide_mode(Q_in, Qn, rain_mm, dp_kpa, soc_pct, rain_thr, dp_limit, soc_limit)
+    st = mode_policy(mode, Q_in, Qn, Hn)
+    p_ref = electrical_power_kw(Qn, Hn, 0.9)
 
-    mode = decide_mode(Q_in, rain_mm, dp_kpa, soc_perc)
-    st = mode_policy(mode, Q_in, Hn)
+    header = mode_badge(mode)
+    donuts = f"""
+    <div style="display:flex;gap:28px;flex-wrap:wrap;align-items:center">
+      {donut("효율(η)", st['eta']*100, 100, "%")}
+      {donut("출력(kW)", st['p_kw'], max(1.0, p_ref), " kW")}
+    </div>
+    """
+
+    chips = f"""
+    <div style="display:flex;gap:8px;flex-wrap:wrap">
+      {chip("출력 제한", st['curtail'])}
+      {chip("바이패스", st['bypass'])}
+      {chip("정비 필요", st['service'], on_color="#ef4444")}
+    </div>
+    """
 
     summary = (
         f"**현재 모드**: {mode}\n\n"
-        f"- 현재 유량 Q: {Q_in:.2f} m³/s (설계 Qₙ={Qn} m³/s)\n"
+        f"- 현재 유량 Q: {Q_in:.2f} m³/s (설계 Qₙ={Qn:.2f} m³/s)\n"
         f"- 낙차 H: {Hn:.2f} m\n"
-        f"- 강수량: {rain_mm:.1f} mm/24h\n"
-        f"- 트래시랙 ΔP: {dp_kpa:.1f} kPa\n"
-        f"- ESS SoC: {soc_perc:.0f} %\n\n"
-        f"▶ {st['desc']}"
+        f"- 강수량: {rain_mm:.1f} mm/24h (임계 {rain_thr:.0f})\n"
+        f"- 트래시랙 ΔP: {dp_kpa:.1f} kPa (임계 {dp_limit:.1f})\n"
+        f"- ESS SoC: {soc_pct:.0f}% (임계 {soc_limit:.0f}%)\n\n"
+        f"▶ {st['note']}"
     )
 
-    donuts = f"""
-    <div style="display:flex;gap:28px;flex-wrap:wrap;align-items:center">
-      {donut("효율 η", st['eta']*100, 100, "%")}
-      {donut("출력 P", st['p_kw'], 100, " kW")}
+    maintenance_html = ""
+    if mode == "정비":
+        maintenance_html = """
+        <div style="margin-top:8px;padding:10px 12px;border:1px solid #fecaca;border-radius:10px;background:#fef2f2">
+          <div style="font-weight:800;color:#b91c1c;margin-bottom:6px">정비 모드 핵심 동작</div>
+          <ul style="margin:0 0 0 16px; padding:0; color:#7f1d1d">
+            <li><b>발전 OFF</b> (출력 0 kW, 효율 0)</li>
+            <li><b>가이드베인/취수문 0%</b> (단계적 닫힘)</li>
+            <li><b>바이패스 ON</b> (수위/토크 상승 방지)</li>
+            <li><b>정비 안내</b>: 스크린 청소 또는 SoC 낮추기</li>
+          </ul>
+        </div>
+        """
+
+    panel = f"""
+    <div style="display:flex;flex-direction:column;gap:14px">
+      {header}
+      {donuts}
+      {chips}
+      {maintenance_html}
     </div>
     """
+    return summary, panel, ""
 
-    return summary, donuts, ""
+def run(Q_in, rain_mm, dp_kpa, soc_pct, Hn, Qn, rain_thr, dp_limit, soc_limit):
+    try:
+        Q_in=float(Q_in); rain_mm=float(rain_mm); dp_kpa=float(dp_kpa); soc_pct=float(soc_pct)
+        Hn=float(Hn); Qn=float(Qn); rain_thr=float(rain_thr); dp_limit=float(dp_limit); soc_limit=float(soc_limit)
+    except:
+        return "⚠️ 입력 오류를 확인하세요.", "", ""
+    return _evaluate(Q_in, rain_mm, dp_kpa, soc_pct, Hn, Qn, rain_thr, dp_limit, soc_limit)
 
-# ------------------------------
+# =========================
+# 프리셋 (즉시 전환용)
+# =========================
+def preset_normal_and_run(Hn, Qn, rain_thr, dp_limit, soc_limit):
+    Q = 0.8*float(Qn); rain=10.0; dp=1.0; soc=50.0
+    md, panel, extra = _evaluate(Q, rain, dp, soc, float(Hn), float(Qn), float(rain_thr), float(dp_limit), float(soc_limit))
+    return Q, rain, dp, soc, md, panel, extra
+
+def preset_low_and_run(Hn, Qn, rain_thr, dp_limit, soc_limit):
+    Q = 0.3*float(Qn); rain=5.0; dp=1.0; soc=50.0
+    md, panel, extra = _evaluate(Q, rain, dp, soc, float(Hn), float(Qn), float(rain_thr), float(dp_limit), float(soc_limit))
+    return Q, rain, dp, soc, md, panel, extra
+
+def preset_high_and_run(Hn, Qn, rain_thr, dp_limit, soc_limit):
+    Q = 1.3*float(Qn); rain=max(50.0, float(rain_thr)); dp=1.0; soc=50.0
+    md, panel, extra = _evaluate(Q, rain, dp, soc, float(Hn), float(Qn), float(rain_thr), float(dp_limit), float(soc_limit))
+    return Q, rain, dp, soc, md, panel, extra
+
+def preset_maint_and_run(Hn, Qn, rain_thr, dp_limit, soc_limit):
+    # 정비 모드: 스크린 막힘으로 유도(ΔP > dp_limit)
+    Q = 0.8*float(Qn); rain=10.0; dp=float(dp_limit)+0.5; soc=50.0
+    md, panel, extra = _evaluate(Q, rain, dp, soc, float(Hn), float(Qn), float(rain_thr), float(dp_limit), float(soc_limit))
+    return Q, rain, dp, soc, md, panel, extra
+
+# =========================
+# 자동 시뮬레이션 (2초 간격)
+# =========================
+def autoplay(Hn, Qn, rain_thr, dp_limit, soc_limit):
+    steps = [
+        ("정상",   preset_normal_and_run(Hn, Qn, rain_thr, dp_limit, soc_limit)),
+        ("저유량", preset_low_and_run(Hn, Qn, rain_thr, dp_limit, soc_limit)),
+        ("고유량", preset_high_and_run(Hn, Qn, rain_thr, dp_limit, soc_limit)),
+        ("정비",   preset_maint_and_run(Hn, Qn, rain_thr, dp_limit, soc_limit)),
+    ]
+    for _, tup in steps:
+        Q, rain, dp, soc, md, panel, extra = tup
+        yield (Q, rain, dp, soc, md, panel, extra)
+        time.sleep(2.0)
+
+# =========================
 # Gradio UI
-# ------------------------------
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("## 🌊 스마트 소수력 발전 모드 시뮬레이터")
-    with gr.Row():
-        Q_in = gr.Number(label="현재 유량 Q (m³/s)", value=2.0)
-        rain_mm = gr.Number(label="강수량 (mm/24h)", value=10)
+# =========================
+with gr.Blocks(theme=gr.themes.Soft(), title="소수력 스마트 운전모드 대시보드 (정상/저유량/고유량/정비)") as app:
+    gr.Markdown("## 🌊 소수력 스마트 운전모드 대시보드")
+
     with gr.Row():
-        dp_kpa = gr.Number(label="트래시랙 ΔP (kPa)", value=1.0)
-        soc_perc = gr.Number(label="ESS SoC (%)", value=50)
+        with gr.Column(scale=1):
+            Q_in  = gr.Number(label="현재 유량 Q (m³/s)", value=2.0)
+            rain  = gr.Number(label="강수량 (mm/24h)", value=10)
+            dp    = gr.Number(label="트래시랙 ΔP (kPa)", value=1.0)
+            soc   = gr.Number(label="ESS SoC (%)", value=50)
+
+            with gr.Accordion("고급 설정 (설계/임계값)", open=False):
+                with gr.Row():
+                    Hn  = gr.Number(label="설계 낙차 Hₙ (m)", value=DEFAULTS["Hn"])
+                    Qn  = gr.Number(label="설계 유량 Qₙ (m³/s)", value=DEFAULTS["Qn"])
+                with gr.Row():
+                    rain_thr = gr.Number(label="고유량 판단 강우 임계 (mm/24h)", value=DEFAULTS["rain_thr"])
+                    dp_limit = gr.Number(label="ΔP 임계 (kPa, 정비)", value=DEFAULTS["dp_limit"])
+                    soc_limit = gr.Number(label="ESS SoC 임계 (% , 정비)", value=DEFAULTS["soc_limit"])
+
+            run_btn   = gr.Button("시뮬레이션 실행", variant="primary")
+
+            gr.Markdown("### 🔘 빠른 전환 (아래 버튼을 누르면 **즉시** 해당 모드로 세팅 + 실행)")
+            with gr.Row():
+                btn_norm = gr.Button("정상", variant="secondary")
+                btn_low  = gr.Button("저유량", variant="secondary")
+                btn_high = gr.Button("고유량", variant="secondary")
+                btn_maint= gr.Button("정비", variant="stop")
+
+            demo_btn  = gr.Button("🎬 자동 시연 (정상 → 저유량 → 고유량 → 정비, 2초 간격)")
+
+        with gr.Column(scale=2):
+            out_md  = gr.Markdown()
+            out_vis = gr.HTML()
+            out_extra = gr.HTML()
 
-    run_btn = gr.Button("실행")
+    # 실행 버튼
+    run_btn.click(
+        fn=run,
+        inputs=[Q_in, rain, dp, soc, Hn, Qn, rain_thr, dp_limit, soc_limit],
+        outputs=[out_md, out_vis, out_extra]
+    )
 
-    summary = gr.Markdown()
-    donuts = gr.HTML()
-    extra = gr.Markdown()
+    # 즉시 전환 버튼들 (입력칸도 함께 업데이트)
+    btn_norm.click(
+        fn=preset_normal_and_run,
+        inputs=[Hn, Qn, rain_thr, dp_limit, soc_limit],
+        outputs=[Q_in, rain, dp, soc, out_md, out_vis, out_extra]
+    )
+    btn_low.click(
+        fn=preset_low_and_run,
+        inputs=[Hn, Qn, rain_thr, dp_limit, soc_limit],
+        outputs=[Q_in, rain, dp, soc, out_md, out_vis, out_extra]
+    )
+    btn_high.click(
+        fn=preset_high_and_run,
+        inputs=[Hn, Qn, rain_thr, dp_limit, soc_limit],
+        outputs=[Q_in, rain, dp, soc, out_md, out_vis, out_extra]
+    )
+    btn_maint.click(
+        fn=preset_maint_and_run,
+        inputs=[Hn, Qn, rain_thr, dp_limit, soc_limit],
+        outputs=[Q_in, rain, dp, soc, out_md, out_vis, out_extra]
+    )
 
-    run_btn.click(fn=run, inputs=[Q_in, rain_mm, dp_kpa, soc_perc],
-                  outputs=[summary, donuts, extra])
+    # 자동 시연 (2초 간격으로 단계적 출력 + 입력칸 동시 업데이트)
+    demo_btn.click(
+        fn=autoplay,
+        inputs=[Hn, Qn, rain_thr, dp_limit, soc_limit],
+        outputs=[Q_in, rain, dp, soc, out_md, out_vis, out_extra]
+    )
 
-# ------------------------------
-# 실행 시작
-# ------------------------------
 if __name__ == "__main__":
-    demo.launch()
+    app.launch()