Update app.py

app.py

@@ -1,38 +1,24 @@
 import gradio as gr
 
-# ---------------------------
-# Small-Hydro 4-Mode Simulator (Web, Gradio)
-# ---------------------------
+# ========== 계산 로직 ==========
+RHO = 1000.0
+G = 9.80665
+PF_TARGET = 0.98
 
-RHO = 1000.0          # kg/m^3
-G   = 9.80665         # m/s^2
-PF_TARGET = 0.98      # 역률 목표 표기용
-
-# 효율 근사: Kaplan/Crossflow 부분부하 합성 (상대유량 r=Q/Qn)
 def efficiency_from_relative_flow(r: float) -> float:
     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:]):
+    curve = [(0.0,0.0),(0.1,0.4),(0.3,0.6),(0.5,0.75),(0.7,0.85),
+             (0.9,0.90),(1.0,0.91),(1.2,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)
+            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
+def electrical_power_kw(Q_in, head_m, eta):
     return (RHO * G * Q_in * head_m * max(0.0, min(1.0, 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:
+def decide_mode(Q_in, q_low, q_high, rain_mm, dp_kpa, rain_thr, dp_limit, emergency):
     if emergency:
         return "Emergency"
     if Q_in < q_low:
@@ -41,89 +27,7 @@ def decide_mode(Q_in: float, q_low: float, q_high: float, rain_mm: float, dp_kpa
         return "Flood"
     return "Normal"
 
-def mode_policy(mode: str, Q_in: float, Qn: float, Hn: float, rain_mm: float):
-    # 상대 유량/효율/출력(커테일 전)
-    r   = min(Q_in, Qn) / Qn if Qn > 0 else 0.0
-    eta = efficiency_from_relative_flow(r)
-    p_kw_raw = electrical_power_kw(min(Q_in, Qn), Hn, eta)
-
-    # 기본 제어 변수
-    gate_open_pct  = round(min(100.0, max(10.0, r * 100.0)), 1)  # 취수문 개도 추정
-    guide_vane_pct = gate_open_pct                               # 가이드베인/피치 추정
-    curtail = False
-    use_bypass = False
-    dump_load = False
-    soc_range = (55, 70)  # Normal 기본
-
-    actions = []
-
-    if mode == "Normal":
-        soc_range = (55, 70)
-        actions += [
-            "가이드베인/피치: PID 추종으로 출력 최적화",
-            "ESS: SoC 55–70% 유지(수명 최적)",
-            "트래시랙 ΔP 모니터링 및 주기 청소",
-        ]
-
-    elif mode == "Flood":
-        soc_range = (35, 45)
-        curtail = True
-        use_bypass = True
-        dump_load = True
-        gate_open_pct  = max(20.0, gate_open_pct - 20.0)  # 수격 방지용 감개
-        guide_vane_pct = gate_open_pct
-        p_kw_raw *= 0.6  # 우회/커테일 반영한 보수 출력
-        actions += [
-            "가이드베인 단계적 감개(수격 방지 곡선 준수)",
-            "여수로/바이패스 개방, 배사문 간헐 개방(토사·탁도 저감)",
-            "ESS: SoC 목표 35–45% (강우 예보 반영)",
-            "덤프로드/가변부하 우선 가동으로 초과전력 흡수",
-            "트래시랙 자가청소 연속 구동(ΔP 임계 초과 시)",
-        ]
-
-    elif mode == "Low-Flow":
-        soc_range = (45, 65)
-        guide_vane_pct = max(15.0, min(55.0, guide_vane_pct))
-        gate_open_pct  = guide_vane_pct
-        p_kw_raw *= 0.85  # 부분부하 손실
-        actions += [
-            "저부하 효율점(가변속/각도 고정) 운전",
-            "ESS: 피크 보조 방전, SoC 45–65%",
-            "유입수속도 확보(NPSH 여유 유지), 공동현상 감시",
-        ]
-
-    elif mode == "Emergency":
-        soc_range = (30, 50)
-        curtail = True
-        use_bypass = True
-        dump_load = False
-        guide_vane_pct = 0.0
-        gate_open_pct  = 0.0
-        p_kw_raw = 0.0
-        actions += [
-            "즉시 무부하 → 정지(수격 곡선 준수)",
-            "취수문 폐쇄 & 바이패스/스필웨이 개방(치수 안전)",
-            "계통 분리, UPS/ESS로 계장 전원 유지",
-            "사후 점검: 베어링/샤프트/관로/서지탱크",
-        ]
-    else:
-        actions.append("정의되지 않은 모드")
-
-    estimates = {
-        "추정효율(eta)": f"{eta:.3f}",
-        "추정출력(kW)": f"{p_kw_raw:.1f}",
-        "설정_가이드베인(%)": f"{guide_vane_pct:.1f}",
-        "설정_취수문개도(%)": f"{gate_open_pct:.1f}",
-        "커테일": str(curtail),
-        "바이패스개방": str(use_bypass),
-        "덤프로드가동": str(dump_load),
-        "ESS_SoC_목표(%)": f"{soc_range[0]}–{soc_range[1]}",
-        "역률목표": f"{PF_TARGET}",
-    }
-    return actions, estimates
-
-def run_sim(Q_in, rain_mm, dp_kpa, emergency,
-            Hn, Qn, rain_thr, dp_limit):
+def run_sim(Q_in, rain_mm, dp_kpa, emergency, Hn, Qn, rain_thr, dp_limit):
     try:
         Q_in     = float(Q_in)
         rain_mm  = float(rain_mm)
@@ -140,58 +44,46 @@ def run_sim(Q_in, rain_mm, dp_kpa, emergency,
     q_high = 1.2 * Qn
 
     mode = decide_mode(Q_in, q_low, q_high, rain_mm, dp_kpa, rain_thr, dp_limit, emergency)
-    actions, estimates = mode_policy(mode, Q_in, Qn, Hn, rain_mm)
-
-    # 결과를 Markdown으로 정리
-    md = []
-    md.append(f"## 운전 모드: **{mode}**")
-    md.append("")
-    md.append("### 입력 요약")
-    md.append(f"- 유량 Q: **{Q_in:.2f} m³/s**")
-    md.append(f"- 강우 예보: **{rain_mm:.1f} mm/24h**")
-    md.append(f"- 트래시랙 ΔP: **{dp_kpa:.1f} kPa**")
-    md.append(f"- 설계 낙차 Hₙ: **{Hn:.2f} m**, 설계 유량 Qₙ: **{Qn:.2f} m³/s**  \n"
-              f"  (저유량 임계 **{q_low:.2f}**, 고유량 임계 **{q_high:.2f}**)")
-    md.append("")
-    md.append("### 권장 동작")
-    for i, a in enumerate(actions, 1):
-        md.append(f"{i}. {a}")
-    md.append("")
-    md.append("### 추정/설정 값")
-    for k, v in estimates.items():
-        md.append(f"- **{k}**: {v}")
-    md.append("")
-    md.append("> 참고: 출력 추정은 간이 효율곡선/커테일 반영으로 산정된 보수치입니다.")
-    return "\n".join(md)
-
-# ---------------------------
-# Gradio UI
-# ---------------------------
-with gr.Blocks(theme=gr.themes.Soft()) as app:
-    gr.Markdown("## 🌊 소수력 4-모드 시뮬레이터 (Hugging Face Spaces)")
+    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)
+
+    lines = []
+    lines.append(f"## 운전 모드: **{mode}**")
+    lines.append("")
+    lines.append("### 입력 요약")
+    lines.append(f"- 유량 Q: **{Q_in:.2f} m³/s**")
+    lines.append(f"- 강우 예보: **{rain_mm:.1f} mm/24h**")
+    lines.append(f"- 트래시랙 ΔP: **{dp_kpa:.1f} kPa**")
+    lines.append(f"- 설계 낙차 Hₙ: **{Hn:.2f} m**, 설계 유량 Qₙ: **{Qn:.2f} m³/s**  ")
+    lines.append(f"  (저유량 임계 **{q_low:.2f}**, 고유량 임계 **{q_high:.2f}**)")
+    lines.append("")
+    lines.append("### 추정/설정 값")
+    lines.append(f"- **효율(eta)**: {eta:.3f}")
+    lines.append(f"- **추정 출력(kW)**: {p_kw:.1f}")
+    lines.append(f"- **ESS 역률 목표**: {PF_TARGET}")
+    return "\n".join(lines)
+
+# ========== Gradio UI ==========
+with gr.Blocks(title="Small-Hydro 4-Mode Simulator") as demo:
+    gr.Markdown("## 🌊 소수력 4-모드 시뮬레이터 (Spaces)")
     with gr.Row():
         Q_in   = gr.Number(label="현재 유량 Q (m³/s)", value=3.0)
         rain   = gr.Number(label="강우 예보 (mm/24h)", value=0)
-        dp     = gr.Number(label="트래시랙 차압 ΔP (kPa)", value=0)
+        dp     = gr.Number(label="트래시랙 ΔP (kPa)", value=0)
         emer   = gr.Checkbox(label="비상 알람 (Emergency)", 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)
+            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()
-
-    run_btn.click(
-        fn=run_sim,
-        inputs=[Q_in, rain, dp, emer, Hn, Qn, rain_thr, dp_limit],
-        outputs=out_md
-    )
+    btn   = gr.Button("시뮬레이션 실행", variant="primary")
+    out   = gr.Markdown()
+    btn.click(run_sim, [Q_in, rain, dp, emer, Hn, Qn, rain_thr, dp_limit], out)
 
-# Spaces에서는 launch() 호출 불필요하지만, 로컬 테스트에선 아래 주석 해제
-# if __name__ == "__main__":
-#     app.launch()
+# Spaces에서는 launch를 호출해도 안전합니다.
+if __name__ == "__main__":
+    demo.launch()