Update app.py

app.py

@@ -4,10 +4,10 @@ import gradio as gr
 from datetime import datetime
 from transformers import pipeline
 
-# ────────────────────────────── 모델 불러오기 ──────────────────────────────
+# ──────── Gemma 모델 로딩 (1회 캐시) ────────
 llm = pipeline("text-generation", model="google/gemma-2-2b-it", max_new_tokens=512)
 
-# ────────────────────────────── API 처리 함수 ──────────────────────────────
+# ──────── API 함수 정의 ────────
 def get_school_info(region_code, school_name, api_key):
     url = f"https://open.neis.go.kr/hub/schoolInfo?KEY={api_key}&Type=json&pIndex=1&pSize=1&SCHUL_NM={school_name}&ATPT_OFCDC_SC_CODE={region_code}"
     res = requests.get(url)
@@ -36,20 +36,27 @@ def summarize_schedule(rows, school_name, year):
     result = llm([{"role": "user", "content": prompt}])
     return result[0]["generated_text"].replace(prompt, "").strip()
 
-# ────────────────────────────── 그라디오 인터페이스 함수 ──────────────────────────────
-def load_and_summarize(region_code, school_name, year, month):
+# ──────── 지역 코드 목록 ────────
+region_options = {
+    "B10": "서울", "C10": "부산", "D10": "대구", "E10": "인천", "F10": "광주", "G10": "대전",
+    "H10": "울산", "I10": "세종", "J10": "경기", "K10": "강원", "M10": "충북", "N10": "충남",
+    "P10": "전북", "Q10": "전남", "R10": "경북", "S10": "경남", "T10": "제주"
+}
+
+# ──────── 월 일정만 먼저 보여주는 함수 ────────
+def fetch_month_schedule(region_code, school_name, year, month):
     api_key = os.environ.get("NEIS_API_KEY", "a69e08342c8947b4a52cd72789a5ecaf")
 
     if not school_name.strip():
-        return "❗ 학교명을 입력해주세요.", "", ""
+        return "❗ 학교명을 입력해주세요.", "", "", None
 
     school_code, confirmed_region = get_school_info(region_code, school_name, api_key)
     if not school_code:
-        return "❗ 학교 정보를 찾을 수 없습니다. 입력을 확인해주세요.", "", ""
+        return "❗ 학교 정보를 찾을 수 없습니다.", "", "", None
 
     schedule_rows = get_schedule(confirmed_region, school_code, year, api_key)
     if not schedule_rows:
-        return "ℹ️ 학사일정 정보가 없습니다.", "", ""
+        return "ℹ️ 학사일정 정보가 없습니다.", "", "", None
 
     # 월별 일정 필터링
     month_rows = [r for r in schedule_rows if r["AA_YMD"].startswith(f"{year}{month:02}")]
@@ -58,20 +65,18 @@ def load_and_summarize(region_code, school_name, year, month):
         for r in month_rows
     ) or "📭 선택한 월에 해당하는 일정이 없습니다."
 
-    # 요약 생성
-    summary = summarize_schedule(schedule_rows, school_name, year)
+    return f"✅ {school_name}의 {year}년 {month}월 일정입니다.", month_display, "✨ [아래 버튼으로 요약 생성]", schedule_rows
 
-    return f"✅ {school_name}의 {year}년 {month}월 일정입니다.", month_display, summary
-
-# ────────────────────────────── Gradio UI 구성 ──────────────────────────────
-region_options = {
-    "B10": "서울", "C10": "부산", "D10": "대구", "E10": "인천", "F10": "광주", "G10": "대전",
-    "H10": "울산", "I10": "세종", "J10": "경기", "K10": "강원", "M10": "충북", "N10": "충남",
-    "P10": "전북", "Q10": "전남", "R10": "경북", "S10": "경남", "T10": "제주"
-}
+# ──────── 요약 버튼이 눌렸을 때 실행되는 함수 ────────
+def generate_summary(school_name, year, rows):
+    if not rows:
+        return "⚠️ 먼저 일정을 불러와 주세요."
+    return summarize_schedule(rows, school_name, year)
 
+# ──────── Gradio 인터페이스 구성 ────────
 with gr.Blocks(title="학사일정 요약 캘린더") as demo:
     gr.Markdown("## 🏫 학사일정 조회 및 AI 요약 (google/gemma-2-2b-it)")
+
     with gr.Row():
         region = gr.Dropdown(label="지역 교육청", choices=list(region_options.keys()), value="B10", interactive=True)
         school_name = gr.Textbox(label="학교명", placeholder="예: 상리초등학교")
@@ -79,12 +84,23 @@ with gr.Blocks(title="학사일정 요약 캘린더") as demo:
         year = gr.Dropdown(label="년도", choices=[2022, 2023, 2024, 2025], value=2024)
         month = gr.Slider(label="월", minimum=1, maximum=12, value=7, step=1)
 
-    btn = gr.Button("📅 일정 불러오기 및 요약")
-
+    load_btn = gr.Button("��� 월 일정 불러오기")
     status = gr.Markdown()
     month_output = gr.Textbox(label="📆 선택한 월의 일정", lines=8)
-    summary_output = gr.Textbox(label="📝 1년 학사일정 요약 (Gemma)", lines=6)
+    summary_notice = gr.Markdown()
+    stored_rows = gr.State()
+
+    with gr.Accordion("✨ 1년치 AI 요약 보기", open=False):
+        summarize_btn = gr.Button("🤖 요약 생성하기")
+        summary_output = gr.Textbox(label="📝 학사일정 요약", lines=6)
+
+    # 이벤트 연결
+    load_btn.click(fn=fetch_month_schedule,
+                   inputs=[region, school_name, year, month],
+                   outputs=[status, month_output, summary_notice, stored_rows])
 
-    btn.click(fn=load_and_summarize, inputs=[region, school_name, year, month], outputs=[status, month_output, summary_output])
+    summarize_btn.click(fn=generate_summary,
+                        inputs=[school_name, year, stored_rows],
+                        outputs=summary_output)
 
 demo.launch()