Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- 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"""
|