add swmirae4

app.py

@@ -2,11 +2,7 @@
 # -----------------------------
 # HF Router Inference API (OpenAI-compatible)
 # Toss-like UI + "SW 미래 영재학원" theme
-# Tabs:
-#   1) 예약·추천 (슬롯 예약 + 맞춤 추천)
-#   2) 질문하기 (채팅)
-#   3) 학원 소개
-#   4) 상태 점검
+# Tabs: 예약·추천 / 질문(채팅) / 학원 소개 / 상태 점검
 # -----------------------------
 
 import os, json, time, requests, gradio as gr
@@ -17,7 +13,7 @@ from datetime import datetime, timedelta
 # Config
 # =========================
 API_BASE   = os.getenv("HF_ENDPOINT_URL") or "https://router.huggingface.co/v1"  # OpenAI-compatible Router
-MODEL_QA   = os.getenv("HF_MODEL_ID") or "Qwen/Qwen2.5-7B-Instruct"             # Provider-attached model
+MODEL_QA   = os.getenv("HF_MODEL_ID") or "Qwen/Qwen2.5-7B-Instruct"              # Provider-attached model
 HF_TOKEN   = os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACEHUB_API_TOKEN")
 TIMEOUT_S  = 60
 
@@ -33,7 +29,6 @@ KWDAYS          = ["월", "화", "수", "목", "금", "토", "일"]
 # =========================
 def hf_chat(model: str, messages: list, temperature: float = 0.2, max_tokens: int = 256,
             timeout: int = TIMEOUT_S, retries: int = 3) -> str:
-    """Call HF Router /v1/chat/completions with basic retry/backoff."""
     if not HF_TOKEN:
         return "⚠️ HF_TOKEN이 없습니다. 환경변수에 HF_TOKEN을 설정하세요."
     headers = {"Authorization": f"Bearer {HF_TOKEN}", "Content-Type": "application/json"}
@@ -79,7 +74,6 @@ def save_reservations(resv: List[Dict[str, Any]]) -> None:
     os.replace(tmp, RESV_FILE)
 
 def make_slots(start: datetime | None = None, days: int = SLOT_DAYS_AHEAD) -> List[str]:
-    """Return list like 'YYYY-MM-DD HH:MM (요일)' for open slots (Mon-Sat only)."""
     now = datetime.now()
     base = start or now
     out = []
@@ -95,12 +89,10 @@ def booked_slots() -> set:
     return {item["slot"] for item in load_reservations()}
 
 def available_slots() -> List[str]:
-    # remove already booked
     bset = booked_slots()
     return [s for s in make_slots() if s not in bset]
 
-def resv_to_rows(resv: List[Dict[str, Any]]) -> List[List[str]]:
-    # For Dataframe display
+def reservation_rows(resv: List[Dict[str, Any]]) -> List[List[str]]:
     rows = []
     for r in sorted(resv, key=lambda x: x.get("slot", "")):
         rows.append([
@@ -116,12 +108,11 @@ def resv_to_rows(resv: List[Dict[str, Any]]) -> List[List[str]]:
     return rows
 
 # =========================
-# Recommendation (LLM + fallback)
+# Recommendation
 # =========================
 def recommend_plan(grade: str, focuses: List[str], level: str, goals: str,
                    weeks: int, hours_per_week: int, constraints: str,
                    temperature: float, max_tokens: int) -> str:
-    """Create a tailored plan via LLM. Falls back to rule-based if LLM fails."""
     focus_str = ", ".join(focuses) if focuses else "일반"
     user_profile = (
         f"- 학년: {grade}\n- 관심 분야: {focus_str}\n- 수준: {level}\n"
@@ -139,7 +130,7 @@ def recommend_plan(grade: str, focuses: List[str], level: str, goals: str,
         "\n\n형식:\n"
         "1) 한 줄 요약\n"
         "2) 주차별 계획(주차/학습목표/활동/과제)\n"
-        "3) 추천 교재/툴(간단 링크명만, 실제 URL은 생략)\n"
+        "3) 추천 교재/툴(간단 링크명만, 실제 URL 생략)\n"
         "4) 예상 결과물(포트폴리오/대회/자격증)\n"
         "5) 다음 예약 제안(체험 수업 1회 + 정규 과정)\n"
         "문장 짧게. 불필요한 수식어 금지."
@@ -150,7 +141,6 @@ def recommend_plan(grade: str, focuses: List[str], level: str, goals: str,
     ]
     out = hf_chat(MODEL_QA, msgs, temperature=temperature, max_tokens=max_tokens)
     if out.startswith("❌") or out.startswith("⚠️"):
-        # Fallback (간단 규칙)
         rec = f"""[간단 추천안 — Fallback]
 한 줄 요약: {grade} 대상 {focus_str} {level} 과정, {weeks}주, 주 {hours_per_week}시간.
 
@@ -200,27 +190,26 @@ def reset_chat():
     return [], []
 
 # =========================
-# Booking handlers (Gradio)
+# Booking handlers
 # =========================
-def refresh_slots() -> List[str]:
-    return available_slots()
+def refresh_slots_update():
+    return gr.update(choices=available_slots(), value=None)
 
 def submit_booking(name: str, grade: str, focuses: List[str], level: str,
-                   contact: str, slot: str, notes: str) -> str:
+                   contact: str, slot: str, notes: str):
     name = (name or "").strip()
     contact = (contact or "").strip()
     slot = (slot or "").strip()
     focus_str = ", ".join(focuses) if focuses else "일반"
 
     if not name:
-        return "⚠️ 이름을 입력하세요."
+        return "⚠️ 이름을 입력하세요.", reservation_rows(load_reservations()), refresh_slots_update()
     if not contact:
-        return "⚠️ 연락처(전화/카카오/이메일) 입력하세요."
+        return "⚠️ 연락처(전화/카카오/이메일) 입력하세요.", reservation_rows(load_reservations()), refresh_slots_update()
     if not slot:
-        return "⚠️ 예약 슬롯을 선택하세요."
-    # re-check conflict
+        return "⚠️ 예약 슬롯을 선택하세요.", reservation_rows(load_reservations()), refresh_slots_update()
     if slot in booked_slots():
-        return "❌ 이미 예약된 슬롯입니다. 다른 시간을 선택하세요."
+        return "❌ 이미 예약된 슬롯입니다. 다른 시간을 선택하세요.", reservation_rows(load_reservations()), refresh_slots_update()
 
     resv = load_reservations()
     item = {
@@ -235,13 +224,9 @@ def submit_booking(name: str, grade: str, focuses: List[str], level: str,
     }
     resv.append(item)
     save_reservations(resv)
-    return f"✅ 예약 완료: {slot} · {name} ({grade}, {focus_str}/{level}) — 담당자가 곧 연락드립니다."
-
-def get_reservation_table():
-    resv = load_reservations()
-    header = ["슬롯", "이름", "학년", "관심 분야", "레벨", "연락처", "비고", "예약시각"]
-    rows = resv_to_rows(resv)
-    return [header] + rows if rows else [header]
+    return (f"✅ 예약 완료: {slot} · {name} ({grade}, {focus_str}/{level}) — 담당자가 곧 연락드립니다.",
+            reservation_rows(resv),
+            refresh_slots_update())
 
 # =========================
 # Health check
@@ -306,12 +291,10 @@ with gr.Blocks(title="율하 SW미래영재컴퓨터학원 — 예약·추천·Q
         """)
 
         with gr.Tabs():
-            # =========================
             # 1) 예약·추천
-            # =========================
             with gr.Tab("예약 · 추천"):
                 with gr.Row():
-                    # ----- Left: Recommendation -----
+                    # Left: Recommendation
                     with gr.Column(scale=5):
                         with gr.Group(elem_classes=["toss-card", "toss-input"]):
                             gr.Markdown("### 맞춤 추천")
@@ -322,7 +305,7 @@ with gr.Blocks(title="율하 SW미래영재컴퓨터학원 — 예약·추천·Q
                                 label="관심 분야", value=["파이썬"]
                             )
                             level = gr.Radio(["입문", "기초", "중급", "심화"], value="기초", label="수준")
-                            goals = gr.Textbox(label="목표(예: 대회 입상/자격증/포트폴리오/학교 수행평가 등)", lines=2)
+                            goals = gr.Textbox(label="목표(예: 대회 입상/자격증/포트폴리오/수행평가 등)", lines=2)
                             with gr.Row():
                                 weeks = gr.Slider(2, 16, value=8, step=1, label="기간(주)")
                                 hpw = gr.Slider(1, 6, value=2, step=1, label="주당 시간(시간)")
@@ -334,7 +317,7 @@ with gr.Blocks(title="율하 SW미래영재컴퓨터학원 — 예약·추천·Q
                         with gr.Group(elem_classes=["toss-card"]):
                             rec_out = gr.Textbox(label="추천 결과", lines=18)
 
-                    # ----- Right: Booking -----
+                    # Right: Booking
                     with gr.Column(scale=4):
                         with gr.Group(elem_classes=["toss-card", "toss-input"]):
                             gr.Markdown("### 예약")
@@ -354,34 +337,29 @@ with gr.Blocks(title="율하 SW미래영재컴퓨터학원 — 예약·추천·Q
                                 btn_book    = gr.Button("예약 확정", elem_classes=["toss-primary"])
                         with gr.Group(elem_classes=["toss-card"]):
                             gr.Markdown("### 현재 예약 현황")
-                            table = gr.Dataframe(headers=["슬롯", "이름", "학년", "관심 분야", "레벨", "연락처", "비고", "예약시각"],
-                                                 value=get_reservation_table(), interactive=False, wrap=True, height=260)
+                            table = gr.Dataframe(
+                                headers=["슬롯", "이름", "학년", "관심 분야", "레벨", "연락처", "비고", "예약시각"],
+                                value=reservation_rows(load_reservations()),
+                                interactive=False,
+                            )
                             res_msg = gr.Markdown("", elem_classes=["toss-note"])
 
-                # Bind: recommend
+                # Recommend
                 btn_rec.click(
                     fn=recommend_plan,
                     inputs=[grade, focus, level, goals, weeks, hpw, constraints, r_temp, r_max],
                     outputs=[rec_out],
                 )
-
-                # Bind: refresh slots
-                btn_refresh.click(fn=refresh_slots, outputs=[slot])
-
-                # Bind: booking
-                def _book_and_refresh(name_v, grade_v, focus_v, level_v, contact_v, slot_v, notes_v):
-                    msg = submit_booking(name_v, grade_v, focus_v, level_v, contact_v, slot_v, notes_v)
-                    # refresh dropdown + table
-                    return msg, get_reservation_table(), available_slots()
+                # Refresh slots
+                btn_refresh.click(fn=refresh_slots_update, outputs=[slot])
+                # Booking
                 btn_book.click(
-                    fn=_book_and_refresh,
+                    fn=submit_booking,
                     inputs=[name, b_grade, b_focus, b_level, contact, slot, notes],
                     outputs=[res_msg, table, slot],
                 )
 
-            # =========================
             # 2) 질문하기 (채팅)
-            # =========================
             with gr.Tab("질문하기 (채팅)"):
                 chat_history = gr.State([])  # list of (user, assistant)
                 with gr.Row():
@@ -420,9 +398,7 @@ with gr.Blocks(title="율하 SW미래영재컴퓨터학원 — 예약·추천·Q
                 gr.Markdown('<div class="toss-note">• 대화 기록은 위 채팅 영역에 순서대로 누적됩니다.'
                             '<br>• 과금 주의: Router 사용량/토큰에 따라 비용이 발생할 수 있습니다.</div>')
 
-            # =========================
             # 3) 학원 소개
-            # =========================
             with gr.Tab("학원 소개"):
                 with gr.Group(elem_classes=["toss-card"]):
                     gr.Markdown("""
@@ -440,9 +416,7 @@ with gr.Blocks(title="율하 SW미래영재컴퓨터학원 — 예약·추천·Q
 > 상담/체험 수업 문의는 '예약 · 추천' 탭에서 진행하세요.
 """)
 
-            # =========================
             # 4) 상태 점검
-            # =========================
             with gr.Tab("상태 점검"):
                 with gr.Group(elem_classes=["toss-card"]):
                     diag_btn = gr.Button("엔드포인트 점검", elem_classes=["toss-primary"])