Update app.py

app.py

@@ -1,284 +1,137 @@
-import time
 import gradio as gr
 
-# ===== 물리 상수 =====
-RHO = 1000.0
-G   = 9.80665
-PF_TARGET = 0.98
-
-# ===== 효율 근사곡선 (상대유량 r = Q/Qn) =====
-def efficiency_from_relative_flow(r: float) -> float:
-    r = max(0.0, min(1.2, float(r)))
-    pts = [
-        (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(pts[:-1], pts[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: float, H: float, eta: float) -> float:
-    if Q <= 0 or H <= 0 or eta <= 0:
-        return 0.0
-    return (RHO * G * Q * H * eta) / 1000.0
-
-# ===== 운전 모드 (정상/저유량/고유량/비상) =====
-def decide_mode(Q, q_low, q_high, rain, dp, rain_thr, dp_lim, emerg):
-    if emerg: return "비상"
-    if Q < q_low: return "저유량"
-    if (Q > q_high) or (dp > dp_lim) or (rain >= rain_thr): return "고유량"
+# ------------------------------
+# 설계 상수
+# ------------------------------
+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:
+        return "정비"
+    # 저유량
+    if Q < 0.4 * Qn:
+        return "저유량"
+    # 고유량
+    if Q > 1.2 * Qn or rain >= rain_thr:
+        return "고유량"
+    # 정상
     return "정상"
 
-def mode_policy(mode, Q, Qn, Hn):
-    r   = min(Q, Qn)/Qn if Qn>0 else 0.0
-    eta = efficiency_from_relative_flow(r)
-    p   = electrical_power_kw(min(Q,Qn), Hn, eta)
-
-    vane = gate = round(min(100.0, max(10.0, r*100.0)), 1)
-    curtail = bypass = dumpload = False
-    soc_low, soc_high = 55, 70
-
-    if mode == "고유량":
-        curtail = bypass = dumpload = True
-        vane = gate = max(20.0, vane - 20.0)
-        p *= 0.6
-        soc_low, soc_high = 35, 45
-
+# ------------------------------
+# 모드별 동작 정책
+# ------------------------------
+def mode_policy(mode, Q, H):
+    if mode == "정상":
+        eta = 0.90
+        p_kw = electrical_power_kw(Q, H, eta)
+        return {"eta": eta, "p_kw": p_kw, "desc": "정상 운전: 효율 최적점에서 발전"}
     elif mode == "저유량":
-        vane = gate = max(15.0, min(55.0, vane))
-        p *= 0.85
-        soc_low, soc_high = 45, 65
-
-    elif mode == "비상":
-        # 🔴 안전정지: 전력생산 0, 밸브 닫힘, 우회 개방
-        curtail = True
-        bypass  = True
-        dumpload = False
-        vane = gate = 0.0
+        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   = 0.0
-        soc_low, soc_high = 30, 50
-
-    return dict(
-        eta=eta, p_kw=p, vane=vane, gate=gate,
-        curtail=curtail, bypass=bypass, dumpload=dumpload,
-        soc_low=soc_low, soc_high=soc_high
-    )
-
-# ===== 뷰 요소 (배지/도넛/칩/게이지) =====
-def emergency_svg_badge():
-    # 빨간 경고 삼각형 + 흰색 느낌표 (진짜 색상 적용)
-    return """
-    <div style="display:flex;align-items:center;gap:10px;">
-      <svg width="28" height="28" viewBox="0 0 64 64" aria-label="경고">
-        <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>
-      <span style="color:#ef4444;font-weight:900;font-size:18px;">비상 모드</span>
-    </div>
-    """
-
-def mode_badge(mode:str)->str:
-    if mode == "비상":
-        return emergency_svg_badge()
-    color = {"정상":"#22c55e","저유량":"#eab308","고유량":"#f97316"}.get(mode,"#6b7280")
-    label = f"운전 모드 : {mode}"
-    return f"""
-    <div style="padding:12px 16px;border-radius:14px;background:{color};
-                color:#fff;font-weight:800;display:inline-flex;gap:8px;align-items:center;">
-      <span style="font-size:18px">●</span>
-      <span style="font-size:18px">{label}</span>
-    </div>
-    """
-
-def donut(label:str, value:float, maxv:float, unit:str=""):
-    maxv = max(1e-9, maxv)
-    pct = max(0.0, min(value/maxv, 1.0))
-    r = 42
-    C = 2*3.14159*r
-    offset = C * (1 - pct)
+        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"
     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):
-    bg = "#10b981" if on else "#e5e7eb"
-    fg = "#fff" if on else "#111827"
-    text = "ON" if on else "OFF"
-    return f"""
-    <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-9, 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 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>
     </div>
     """
 
-# ===== 메인 실행 =====
-def run(Q_in, rain_mm, dp_kpa, emergency, Hn, Qn, rain_thr, dp_limit):
+# ------------------------------
+# 실행 함수
+# ------------------------------
+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)
-        Hn=float(Hn); Qn=float(Qn); rain_thr=float(rain_thr); dp_limit=float(dp_limit)
-        emergency=bool(emergency)
+        Q_in = float(Q_in)
+        rain_mm = float(rain_mm)
+        dp_kpa = float(dp_kpa)
+        soc_perc = float(soc_perc)
     except:
-        return "입력 오류를 확인하세요.", "", ""
+        return "⚠️ 입력 오류", "", ""
 
-    if Qn<=0 or Hn<=0:
-        return "설계값(Hₙ, Qₙ)은 0보다 커야 합니다.", "", ""
+    mode = decide_mode(Q_in, rain_mm, dp_kpa, soc_perc)
+    st = mode_policy(mode, Q_in, Hn)
 
-    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)
-
-    # 요약(한국어)
     summary = (
-        f"**유량 Q**: {Q_in:.2f} m³/s  |  **설계 Qₙ**: {Qn:.2f} m³/s "
-        f"(임계 {q_low:.2f}~{q_high:.2f})  \n"
-        f"**낙차 H**: {Hn:.2f} m  |  **강우 예보**: {rain_mm:.1f} mm/24h  |  **트래시랙 ΔP**: {dp_kpa:.1f} kPa  \n"
-        f"**효율 η**: {st['eta']:.3f}  |  **추정 출력**: {st['p_kw']:.1f} kW  |  **역률 목표**: {PF_TARGET}  \n"
-        f"**가이드베인 개도**: {st['vane']:.1f}%  |  **취수문 개도**: {st['gate']:.1f}%"
+        f"**현재 모드**: {mode}\n\n"
+        f"- 현재 유량 Q: {Q_in:.2f} m³/s (설계 Qₙ={Qn} 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']}"
     )
-    if mode == "비상":
-        summary += "\n> 🔴 비상: 발전 정지, 베인/취수문 0%, 바이패스 개방, 운영자 승인 대기"
 
-    # 시각 카드
-    p_ref = electrical_power_kw(Qn, Hn, 0.9)
-    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")}
+      {donut("효율 η", st['eta']*100, 100, "%")}
+      {donut("출력 P", st['p_kw'], 100, " kW")}
     </div>
     """
-    bars = f"""
-      {gauge_bar("가이드베인 개도", st['vane'], 100.0, " %")}
-      {gauge_bar("취수문 개도", st['gate'], 100.0, " %")}
-    """
-    chips = f"""
-    <div style="display:flex;gap:8px;flex-wrap:wrap">
-      {chip("출력 제한", st['curtail'])}
-      {chip("바이패스", st['bypass'])}
-      {chip("덤프로드", st['dumpload'])}
-      <div style="padding:6px 10px;border-radius:999px;background:#0ea5e9;color:#fff;font-size:12px;font-weight:700">
-        ESS SoC 목표: {st['soc_low']}–{st['soc_high']}%
-      </div>
-    </div>
-    """
-
-    # 비상 모드 핵심 동작(엄선 5개) – 비상일 때만 노출
-    emergency_actions_html = ""
-    if mode == "비상":
-        emergency_actions_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>발전 정지</b> (출력 0 kW, 효율 0)</li>
-            <li><b>가이드베인 0%</b>, <b>취수문 0%</b> (단계적 닫힘)</li>
-            <li><b>바이패스 개방</b> (수위/토크 상승 방지)</li>
-            <li><b>인버터 차단 및 DC 링크 방전</b></li>
-            <li><b>로그 기록 및 운영자 승인 대기</b> (복구 조건 확인)</li>
-          </ul>
-        </div>
-        """
 
-    panel = f"""
-    <div style="display:flex;flex-direction:column;gap:14px">
-      {header}
-      {donuts}
-      {bars}
-      {chips}
-      {emergency_actions_html}
-    </div>
-    """
-
-    return summary, panel, ""  # 세 번째 칸은 여유(추가 설명용)
-
-# ===== 프리셋 =====
-def preset_normal(Qn):   Qn=float(Qn); return 0.8*Qn, 10.0, 1.0, False
-def preset_low(Qn):      Qn=float(Qn); return 0.3*Qn,  5.0, 0.5, False
-def preset_high(Qn):     Qn=float(Qn); return 1.3*Qn, 60.0, 4.0, False
-def preset_emerg(Qn):    Qn=float(Qn); return 0.8*Qn, 10.0, 1.0, True
-
-# ===== 자동재생(데모) =====
-def autoplay(Qn, Hn, rain_thr, dp_limit):
-    steps = [
-        ("정상",   preset_normal(Qn)),
-        ("저유량", preset_low(Qn)),
-        ("고유량", preset_high(Qn)),
-        ("비상",   preset_emerg(Qn)),
-    ]
-    for name, (Q_in, rain, dp, emer) in steps:
-        md, panel, extra = run(Q_in, rain, dp, emer, Hn, Qn, rain_thr, dp_limit)
-        yield (Q_in, rain, dp, emer, md, panel, extra)
-        time.sleep(1.2)
-
-# ===== Gradio UI =====
-with gr.Blocks(theme=gr.themes.Soft(), title="소수력 스마트 운전모드 대시보드 (Emergency Enhanced)") as app:
-    gr.Markdown("## 🌊 소수력 스마트 운전모드 대시보드")
+    return summary, donuts, ""
 
+# ------------------------------
+# Gradio UI
+# ------------------------------
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("## 🌊 스마트 소수력 발전 모드 시뮬레이터")
     with gr.Row():
-        with gr.Column(scale=1):
-            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)
-
-            with gr.Row():
-                gr.Button("프리셋: 정상").click(preset_normal, [Qn], [Q_in, rain, dp, emer])
-                gr.Button("프리셋: 저유량").click(preset_low,   [Qn], [Q_in, rain, dp, emer])
-                gr.Button("프리셋: 고유량").click(preset_high,  [Qn], [Q_in, rain, dp, emer])
-                gr.Button("프리셋: 비상").click(preset_emerg,  [Qn], [Q_in, rain, dp, emer])
+        Q_in = gr.Number(label="현재 유량 Q (m³/s)", value=2.0)
+        rain_mm = gr.Number(label="강수량 (mm/24h)", value=10)
+    with gr.Row():
+        dp_kpa = gr.Number(label="트래시랙 ΔP (kPa)", value=1.0)
+        soc_perc = gr.Number(label="ESS SoC (%)", value=50)
 
-            run_btn   = gr.Button("시뮬레이션 실행", variant="primary")
-            demo_btn  = gr.Button("🎬 시연 자동재생", variant="secondary")
+    run_btn = gr.Button("실행")
 
-        with gr.Column(scale=2):
-            out_md  = gr.Markdown()
-            out_vis = gr.HTML()
-            out_extra = gr.HTML()
+    summary = gr.Markdown()
+    donuts = gr.HTML()
+    extra = gr.Markdown()
 
-    run_btn.click(
-        fn=run,
-        inputs=[Q_in, rain, dp, emer, Hn, Qn, rain_thr, dp_limit],
-        outputs=[out_md, out_vis, out_extra]
-    )
-
-    demo_btn.click(
-        fn=autoplay,
-        inputs=[Qn, Hn, rain_thr, dp_limit],
-        outputs=[Q_in, rain, dp, emer, out_md, out_vis, out_extra]
-    )
+    run_btn.click(fn=run, inputs=[Q_in, rain_mm, dp_kpa, soc_perc],
+                  outputs=[summary, donuts, extra])
 
+# ------------------------------
+# 실행 시작
+# ------------------------------
 if __name__ == "__main__":
-    app.launch()
+    demo.launch()