Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,43 +1,34 @@
 import os
-os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
-os.environ["HF_HOME"] = "/tmp/hf_cache"
-os.environ["HF_DATASETS_CACHE"] = "/tmp/hf_cache"
 import json
 import requests
 import streamlit as st
 from datetime import datetime
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 
-
+# ✅ 안전한 캐시 경로 설정 (최상단 필수)
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
+os.environ["HF_HOME"] = "/tmp/hf_cache"
+os.environ["HF_DATASETS_CACHE"] = "/tmp/hf_cache"
 
 # ✅ Streamlit 기본 설정
 st.set_page_config(page_title="학사일정 캘린더", layout="centered")
 st.title("📅 학사일정 캘린더 + AI 요약")
 st.markdown("NEIS API에서 학사일정을 불러오고 FullCalendar로 시각화합니다.")
 
-# ✅ 디버깅용 토큰 확인
+# ✅ 디버깅 출력
+token_present = os.environ.get("HUGGINGFACE_TOKEN") is not None
+st.write("🔐 토큰 있음:", token_present)
 st.write("✅ 캐시 경로:", os.environ.get("TRANSFORMERS_CACHE"))
-st.write("🔐 토큰 있음:", os.environ.get("HUGGINGFACE_TOKEN") is not None)
 
 # ✅ Gemma 모델 로딩 함수
 @st.cache_resource
 def load_model():
     token = os.environ.get("HUGGINGFACE_TOKEN")
     model_id = "google/gemma-2-2b-it"
-    cache_dir = "/tmp/hf_cache"  # ✅ Hugging Face Spaces에서 허용된 디렉토리
-
-    tokenizer = AutoTokenizer.from_pretrained(
-        model_id,
-        use_auth_token=token,
-        cache_dir=cache_dir
-    )
-
-    model = AutoModelForCausalLM.from_pretrained(
-        model_id,
-        use_auth_token=token,
-        cache_dir=cache_dir
-    )
+    cache_dir = "/tmp/hf_cache"
 
+    tokenizer = AutoTokenizer.from_pretrained(model_id, use_auth_token=token, cache_dir=cache_dir)
+    model = AutoModelForCausalLM.from_pretrained(model_id, use_auth_token=token, cache_dir=cache_dir)
     return pipeline("text-generation", model=model, tokenizer=tokenizer)
 
 llm = load_model()
@@ -51,17 +42,19 @@ def get_school_info(region_code, school_name, api_key):
     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}"
+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
 
-# ✅ AI 요약 함수
+# ✅ 요약 생성
 def summarize_schedule(rows, school_name, year):
+    if not rows:
+        return "일정이 없어 요약할 수 없습니다."
     lines = []
     for row in rows:
         date = row["AA_YMD"]
@@ -73,7 +66,7 @@ def summarize_schedule(rows, school_name, year):
     result = llm([{"role": "user", "content": prompt}])
     return result[0]["generated_text"].replace(prompt, "").strip()
 
-# ✅ UI: 지역 / 학교 / 연도 선택
+# ✅ 지역/학교/년도/월 선택 UI
 region_options = {
     "B10": "서울", "C10": "부산", "D10": "대구", "E10": "인천", "F10": "광주", "G10": "대전",
     "H10": "울산", "I10": "세종", "J10": "경기", "K10": "강원", "M10": "충북", "N10": "충남",
@@ -84,9 +77,9 @@ 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")
@@ -94,11 +87,11 @@ if submitted:
         if not school_code:
             st.error("학교 정보를 찾을 수 없습니다.")
         else:
-            schedule_rows = get_schedule(region_code, school_code, year, api_key)
+            schedule_rows = get_schedule(region_code, school_code, year, month, api_key)
             if not schedule_rows:
                 st.info("해당 조건의 학사일정이 없습니다.")
             else:
-                # 캘린더용 JSON 생성
+                # ✅ 일정 출력용 FullCalendar 생성
                 events = [
                     {
                         "title": row["EVENT_NM"],
@@ -108,7 +101,6 @@ if submitted:
                 ]
                 event_json = json.dumps(events, ensure_ascii=False)
 
-                # FullCalendar 삽입
                 st.components.v1.html(f"""
                 <html>
                 <head>
@@ -133,10 +125,10 @@ if submitted:
                 </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}")
+                            st.markdown(f"**{school_name} {year}년 {month}월 일정 요약:**\n\n{summary}")