| import os |
| import requests |
| import streamlit as st |
| from datetime import datetime |
| from transformers import pipeline |
|
|
| |
| @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() |
|
|
| |
| 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: |
| |
| 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) |
|
|
| |
| st.components.v1.html(f""" |
| <html> |
| <head> |
| <link href='https://cdn.jsdelivr.net/npm/fullcalendar@6.1.8/index.global.min.css' rel='stylesheet' /> |
| <script src='https://cdn.jsdelivr.net/npm/fullcalendar@6.1.8/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}") |
