Create src/streamlit_app.py

src/streamlit_app.py

@@ -1,40 +1,113 @@
-import altair as alt
-import numpy as np
-import pandas as pd
+import os
+import requests
 import streamlit as st
+from datetime import datetime
+from transformers import pipeline
 
-"""
-# Welcome to Streamlit!
-
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
-
-In the meantime, below is an example of what you can do with just a few lines of code:
-"""
-
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
-
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
-
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
-
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
-
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+# Gemma 모델 로딩
+@st.cache_resource
+def load_model():
+    return pipeline("text-generation", model="google/gemma-2-2b-it", max_new_tokens=512)
+
+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)
+    data = res.json()
+    school = data.get("schoolInfo", [{}])[1].get("row", [{}])[0]
+    return school.get("SD_SCHUL_CODE"), school.get("ATPT_OFCDC_SC_CODE")
+
+def get_schedule(region_code, school_code, year, api_key):
+    from_ymd = f"{year}0101"
+    to_ymd = f"{year}1231"
+    url = f"https://open.neis.go.kr/hub/SchoolSchedule?KEY={api_key}&Type=json&pIndex=1&pSize=500&ATPT_OFCDC_SC_CODE={region_code}&SD_SCHUL_CODE={school_code}&AA_FROM_YMD={from_ymd}&AA_TO_YMD={to_ymd}"
+    res = requests.get(url)
+    data = res.json()
+    rows = data.get("SchoolSchedule", [{}])[1].get("row", [])
+    return rows
+
+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日")
+        event = row["EVENT_NM"]
+        lines.append(f"{dt}: {event}")
+    text = "\n".join(lines)
+    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": "제주"
+}
+
+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}")