|
|
import os |
|
|
import json |
|
|
import requests |
|
|
import streamlit as st |
|
|
from datetime import datetime |
|
|
|
|
|
|
|
|
st.set_page_config(page_title="ํ์ฌ์ผ์ ์บ๋ฆฐ๋", layout="centered") |
|
|
st.title("๐
ํ์ฌ์ผ์ ์บ๋ฆฐ๋ + AI ์์ฝ") |
|
|
st.markdown("NEIS API์์ ํ์ฌ์ผ์ ์ ๋ถ๋ฌ์ค๊ณ FullCalendar๋ก ์๊ฐํํฉ๋๋ค.") |
|
|
|
|
|
|
|
|
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, month, api_key): |
|
|
from_ymd = f"{year}{month:02}01" |
|
|
to_ymd = f"{year}{month:02}31" |
|
|
url = f"https://open.neis.go.kr/hub/SchoolSchedule?KEY={api_key}&Type=json&pIndex=1&pSize=100&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): |
|
|
if not rows: |
|
|
return "์ผ์ ์ด ์์ด ์์ฝํ ์ ์์ต๋๋ค." |
|
|
|
|
|
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์ฃผ์ ์ผ์ ์ ์์ฝํด์ฃผ์ธ์." |
|
|
|
|
|
payload = { |
|
|
"model": "skt/A.X-4.0-Light", |
|
|
"messages": [ |
|
|
{"role": "system", "content": "๋น์ ์ ํ์ฌ์ผ์ ์ ์์ฝํด์ฃผ๋ AI์
๋๋ค."}, |
|
|
{"role": "user", "content": prompt} |
|
|
], |
|
|
"max_tokens": 256, |
|
|
"temperature": 0.0 |
|
|
} |
|
|
|
|
|
try: |
|
|
res = requests.post("http://localhost:8000/v1/chat/completions", json=payload) |
|
|
res.raise_for_status() |
|
|
return res.json()["choices"][0]["message"]["content"].strip() |
|
|
except Exception as e: |
|
|
return f"โ ์์ฝ ์คํจ: {e}" |
|
|
|
|
|
|
|
|
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) |
|
|
month = st.selectbox("์", options=list(range(1, 13)), index=6) |
|
|
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, month, 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 |
|
|
if "AA_YMD" in row and "EVENT_NM" in row |
|
|
] |
|
|
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("๋ชจ๋ธ์ด ์์ฝ ์ค..."): |
|
|
summary = summarize_schedule(schedule_rows, school_name, year) |
|
|
st.success("์์ฝ ์๋ฃ!") |
|
|
st.markdown(f"**{school_name} {year}๋
{month}์ ์ผ์ ์์ฝ:**\n\n{summary}") |
|
|
|
