yongyeol commited on
Commit
8d2a103
Β·
verified Β·
1 Parent(s): 5236e83

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +6 -1
src/streamlit_app.py CHANGED
@@ -41,7 +41,7 @@ def get_school_info(region_code, school_name, api_key):
41
  school = data.get("schoolInfo", [{}])[1].get("row", [{}])[0]
42
  return school.get("SD_SCHUL_CODE"), school.get("ATPT_OFCDC_SC_CODE")
43
 
44
- # βœ… 학사일정 κ°€μ Έμ˜€κΈ°
45
  def get_schedule(region_code, school_code, year, month, api_key):
46
  from_ymd = f"{year}{month:02}01"
47
  to_ymd = f"{year}{month:02}31"
@@ -49,6 +49,7 @@ def get_schedule(region_code, school_code, year, month, api_key):
49
  res = requests.get(url)
50
  data = res.json()
51
  rows = data.get("SchoolSchedule", [{}])[1].get("row", [])
 
52
  return rows
53
 
54
  # βœ… μš”μ•½ 생성
@@ -63,7 +64,9 @@ def summarize_schedule(rows, school_name, year):
63
  lines.append(f"{dt}: {event}")
64
  text = "\n".join(lines)
65
  prompt = f"{school_name}κ°€ {year}년도에 κ°€μ§€λŠ” 학사일정은 λ‹€μŒκ³Ό κ°™μŠ΅λ‹ˆλ‹€:\n{text}\nμ£Όμš” 일정을 μš”μ•½ν•΄μ£Όμ„Έμš”."
 
66
  result = llm([{"role": "user", "content": prompt}])
 
67
  return result[0]["generated_text"].replace(prompt, "").strip()
68
 
69
  # βœ… μ§€μ—­/학ꡐ/년도/μ›” 선택 UI
@@ -98,7 +101,9 @@ if submitted:
98
  "start": datetime.strptime(row["AA_YMD"], "%Y%m%d").strftime("%Y-%m-%d")
99
  }
100
  for row in schedule_rows
 
101
  ]
 
102
  event_json = json.dumps(events, ensure_ascii=False)
103
 
104
  st.components.v1.html(f"""
 
41
  school = data.get("schoolInfo", [{}])[1].get("row", [{}])[0]
42
  return school.get("SD_SCHUL_CODE"), school.get("ATPT_OFCDC_SC_CODE")
43
 
44
+ # βœ… 학사일정 κ°€μ Έμ˜€κΈ° (μ›” λ‹¨μœ„)
45
  def get_schedule(region_code, school_code, year, month, api_key):
46
  from_ymd = f"{year}{month:02}01"
47
  to_ymd = f"{year}{month:02}31"
 
49
  res = requests.get(url)
50
  data = res.json()
51
  rows = data.get("SchoolSchedule", [{}])[1].get("row", [])
52
+ st.write("πŸ“¦ 뢈러온 일정 raw data:", rows)
53
  return rows
54
 
55
  # βœ… μš”μ•½ 생성
 
64
  lines.append(f"{dt}: {event}")
65
  text = "\n".join(lines)
66
  prompt = f"{school_name}κ°€ {year}년도에 κ°€μ§€λŠ” 학사일정은 λ‹€μŒκ³Ό κ°™μŠ΅λ‹ˆλ‹€:\n{text}\nμ£Όμš” 일정을 μš”μ•½ν•΄μ£Όμ„Έμš”."
67
+ st.write("πŸ“€ μš”μ•½μ— μ „λ‹¬λœ ν”„λ‘¬ν”„νŠΈ:", prompt)
68
  result = llm([{"role": "user", "content": prompt}])
69
+ st.write("πŸ“₯ λͺ¨λΈ 생성 κ²°κ³Ό:", result)
70
  return result[0]["generated_text"].replace(prompt, "").strip()
71
 
72
  # βœ… μ§€μ—­/학ꡐ/년도/μ›” 선택 UI
 
101
  "start": datetime.strptime(row["AA_YMD"], "%Y%m%d").strftime("%Y-%m-%d")
102
  }
103
  for row in schedule_rows
104
+ if "AA_YMD" in row and "EVENT_NM" in row
105
  ]
106
+ st.write("πŸ“… FullCalendar에 전달할 events:", events)
107
  event_json = json.dumps(events, ensure_ascii=False)
108
 
109
  st.components.v1.html(f"""