Spaces:
Sleeping
Sleeping
| import os | |
| import requests | |
| import streamlit as st | |
| from datetime import datetime | |
| from transformers import pipeline | |
| # Gemma ๋ชจ๋ธ ๋ก๋ฉ | |
| 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}") | |