Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,26 +1,33 @@
 import os
+import json
 import requests
 import streamlit as st
 from datetime import datetime
-from transformers import pipeline
+from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 
+# ✅ 캐시 디렉토리 설정 (권한 오류 방지)
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
+
+# ✅ Streamlit 기본 설정
 st.set_page_config(page_title="학사일정 캘린더", layout="centered")
+st.title("📅 학사일정 캘린더 + AI 요약")
+st.markdown("NEIS API에서 학사일정을 불러오고 FullCalendar로 시각화합니다.")
 
-# ✅ 바로 토큰 확인 (디버깅용)
+# ✅ 디버깅용 토큰 확인
 st.write("🔐 토큰 있음:", os.environ.get("HUGGINGFACE_TOKEN") is not None)
 
-# Gemma 모델 로딩
+# ✅ Gemma 모델 로딩 함수
 @st.cache_resource
 def load_model():
-    return pipeline(
-        "text-generation",
-        model="google/gemma-2-2b-it",
-        max_new_tokens=512,
-        use_auth_token=os.environ.get("HUGGINGFACE_TOKEN")  # ✅ 직접 전달
-    )
+    token = os.environ.get("HUGGINGFACE_TOKEN")
+    model_id = "google/gemma-2-2b-it"
+    tokenizer = AutoTokenizer.from_pretrained(model_id, use_auth_token=token)
+    model = AutoModelForCausalLM.from_pretrained(model_id, use_auth_token=token)
+    return pipeline("text-generation", model=model, tokenizer=tokenizer)
+
 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)
@@ -28,6 +35,7 @@ def get_school_info(region_code, school_name, api_key):
     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"
@@ -37,23 +45,20 @@ def get_schedule(region_code, school_code, year, api_key):
     rows = data.get("SchoolSchedule", [{}])[1].get("row", [])
     return rows
 
+# ✅ AI 요약 함수
 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日")
+        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주요 일정을 가변적으로 요약해주세요."
+    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로 시각화합니다.")
-
+# ✅ UI: 지역 / 학교 / 연도 선택
 region_options = {
     "B10": "서울", "C10": "부산", "D10": "대구", "E10": "인천", "F10": "광주", "G10": "대전",
     "H10": "울산", "I10": "세종", "J10": "경기", "K10": "강원", "M10": "충북", "N10": "충남",
@@ -66,6 +71,7 @@ with st.form("query_form"):
     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")
@@ -77,7 +83,7 @@ if submitted:
             if not schedule_rows:
                 st.info("해당 조건의 학사일정이 없습니다.")
             else:
-                # 일정 데이터를 HTML 캘린더에 넣기 위한 JSON 생성
+                # 캘린더용 JSON 생성
                 events = [
                     {
                         "title": row["EVENT_NM"],
@@ -85,8 +91,6 @@ if submitted:
                     }
                     for row in schedule_rows
                 ]
-
-                import json
                 event_json = json.dumps(events, ensure_ascii=False)
 
                 # FullCalendar 삽입
@@ -114,6 +118,7 @@ if submitted:
                 </html>
                 """, height=650)
 
+                # 요약 버튼
                 with st.expander("✨ 1년치 요약 보기", expanded=False):
                     if st.button("🤖 요약 생성하기"):
                         with st.spinner("Gemma 모델이 요약 중..."):