Update app.py

app.py

@@ -1,13 +1,17 @@
 import os
 import requests
-import gradio as gr
+import streamlit as st
 from datetime import datetime
 from transformers import pipeline
 
-# ──────── Gemma 모델 로딩 (1회 캐시) ────────
-llm = pipeline("text-generation", model="google/gemma-2-2b-it", max_new_tokens=512)
+# Gemma 모델 로딩
+@st.cache_resource
+def load_model():
+    return pipeline("text-generation", model="google/gemma-2-2b-it", max_new_tokens=512)
 
-# ──────── API 함수 정의 ────────
+llm = load_model()
+
+# 학교 정보 및 일정 가져오기 함수
 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)
@@ -28,79 +32,82 @@ def summarize_schedule(rows, school_name, year):
     lines = []
     for row in rows:
         date = row["AA_YMD"]
-        dt = datetime.strptime(date, "%Y%m%d").strftime("%-m월 %-d일")
+        dt = datetime.strptime(date, "%Y%m%d").strftime("%-m月 %-d日")
         event = row["EVENT_NM"]
         lines.append(f"{dt}: {event}")
     text = "\n".join(lines)
-    prompt = f"{school_name}의 {year}년 학사일정은 다음과 같다:\n{text}\n이 내용을 요약해줘. 주요 일정 중심으로 간결하게."
+    prompt = f"{school_name}가 {year}년도에 가지는 학사일정은 다음과 같습니다:\n{text}\n주요 일정을 가변적으로 요약해주세요."
     result = llm([{"role": "user", "content": prompt}])
     return result[0]["generated_text"].replace(prompt, "").strip()
 
-# ──────── 지역 코드 목록 ────────
+# Streamlit UI
+st.set_page_config(page_title="학사일정 캘린더", layout="centered")
+st.title("📅 학사일정 캘린더 + AI 요약")
+st.markdown("NEIS API에서 학사일정을 불러오고 FullCalendar로 시각화합니다.")
+
 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 "❗ 학교명을 입력해주세요.", "", "", None
-
-    school_code, confirmed_region = get_school_info(region_code, school_name, api_key)
-    if not school_code:
-        return "❗ 학교 정보를 찾을 수 없습니다.", "", "", None
-
-    schedule_rows = get_schedule(confirmed_region, school_code, year, api_key)
-    if not schedule_rows:
-        return "ℹ️ 학사일정 정보가 없습니다.", "", "", None
-
-    # 월별 일정 필터링
-    month_rows = [r for r in schedule_rows if r["AA_YMD"].startswith(f"{year}{month:02}")]
-    month_display = "\n".join(
-        f"{datetime.strptime(r['AA_YMD'], '%Y%m%d').strftime('%Y-%m-%d')} - {r['EVENT_NM']}"
-        for r in month_rows
-    ) or "📭 선택한 월에 해당하는 일정이 없습니다."
-
-    return f"✅ {school_name}의 {year}년 {month}월 일정입니다.", month_display, "✨ [아래 버튼으로 요약 생성]", schedule_rows
-
-# ──────── 요약 버튼이 눌렸을 때 실행되는 함수 ────────
-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="예: 상리초등학교")
-    with gr.Row():
-        year = gr.Dropdown(label="년도", choices=[2022, 2023, 2024, 2025], value=2024)
-        month = gr.Slider(label="월", minimum=1, maximum=12, value=7, step=1)
-
-    load_btn = gr.Button("📅 월 일정 불러오기")
-    status = gr.Markdown()
-    month_output = gr.Textbox(label="📆 선택한 월의 일정", lines=8)
-    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])
-
-    summarize_btn.click(fn=generate_summary,
-                        inputs=[school_name, year, stored_rows],
-                        outputs=summary_output)
-
-demo.launch()
+with st.form("query_form"):
+    region = st.selectbox("지역 교육청", options=list(region_options.keys()), format_func=lambda x: f"{region_options[x]} ({x})")
+    school_name = st.text_input("학교명", placeholder="예: 상리초등학교")
+    year = st.selectbox("년도", options=[2022, 2023, 2024, 2025], index=2)
+    submitted = st.form_submit_button("📅 학사일정 불러오기")
+
+if submitted:
+    with st.spinner("일정 불러오는 중..."):
+        api_key = os.environ.get("NEIS_API_KEY", "a69e08342c8947b4a52cd72789a5ecaf")
+        school_code, region_code = get_school_info(region, school_name, api_key)
+        if not school_code:
+            st.error("학교 정보를 찾을 수 없습니다.")
+        else:
+            schedule_rows = get_schedule(region_code, school_code, year, api_key)
+            if not schedule_rows:
+                st.info("해당 조건의 학사일정이 없습니다.")
+            else:
+                # 일정 데이터를 HTML 캘린더에 넣기 위한 JSON 생성
+                events = [
+                    {
+                        "title": row["EVENT_NM"],
+                        "start": datetime.strptime(row["AA_YMD"], "%Y%m%d").strftime("%Y-%m-%d")
+                    }
+                    for row in schedule_rows
+                ]
+
+                import json
+                event_json = json.dumps(events, ensure_ascii=False)
+
+                # FullCalendar 삽입
+                st.components.v1.html(f"""
+                <html>
+                <head>
+                  <link href='https://cdn.jsdelivr.net/npm/[email protected]/index.global.min.css' rel='stylesheet' />
+                  <script src='https://cdn.jsdelivr.net/npm/[email protected]/index.global.min.js'></script>
+                  <script>
+                    document.addEventListener('DOMContentLoaded', function() {{
+                      var calendarEl = document.getElementById('calendar');
+                      var calendar = new FullCalendar.Calendar(calendarEl, {{
+                        initialView: 'dayGridMonth',
+                        locale: 'ko',
+                        height: 600,
+                        events: {event_json}
+                      }});
+                      calendar.render();
+                    }});
+                  </script>
+                </head>
+                <body>
+                  <div id='calendar'></div>
+                </body>
+                </html>
+                """, height=650)
+
+                with st.expander("✨ 1년치 요약 보기", expanded=False):
+                    if st.button("🤖 요약 생성하기"):
+                        with st.spinner("Gemma 모델이 요약 중..."):
+                            summary = summarize_schedule(schedule_rows, school_name, year)
+                            st.success("요약 완료!")
+                            st.markdown(f"**{school_name} {year}년 학사일정 요약:**\n\n{summary}")