|
|
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/[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}") |
