Update app.py

app.py

@@ -77,13 +77,13 @@ def generate_curriculum(grade, achievement_standard):
 
     prompt = [
         SYSTEM_PROMPT,
-        f"대상 학년: 초등학교 {grade}학년",
+        f"대상 학년: {grade}",
         f"성취기준 및 주요 내용: {achievement_standard}",
         "개념기반 교육과정:", 
     ]
 
     full_text = ""
-    output_area = st.empty()  # 출력 영역 초기화
+    output_area = st.empty()  # 출력 영역 정의
 
     try:
         response = model.generate_content(prompt, stream=True)
@@ -106,25 +106,22 @@ colored_header(
 
 add_vertical_space(1)
 
-# 입력 위젯 (사이드바 사용, 확장 가능)
 with st.sidebar.expander("입력 설정", expanded=True):
     grade_options = [f"초등학교 {i}학년" for i in range(1, 7)]
     grade_selected = st.selectbox("학년 선택", grade_options, index=0)
-    grade = int(grade_selected.split()[1][:-2])  # 학년 추출
+    grade = grade_selected  # 선택된 학년 문자열 그대로 사용
 
     achievement_standard = st.text_area("성취기준 및 주요 내용 입력", height=200)
 
 generate_button = st.button("개념기반 교육과정 생성")
 
-# 생성 버튼 클릭시 동작
+# 단일 출력 영역을 유지하고, 필요한 경우에만 갱신
 if generate_button and achievement_standard:
     result = generate_curriculum(grade, achievement_standard)
-    st.session_state.generated_curriculum = result # 세션 상태에 결과 저장
+    st.session_state.generated_curriculum = result
 elif generate_button and not achievement_standard:
     st.warning("성취기준 및 주요 내용을 입력하세요.")
 
-
-# 출력 영역
-output_area = st.empty()
+# 이미 생성된 교육과정이 있을 경우 이를 출력
 if "generated_curriculum" in st.session_state and st.session_state.generated_curriculum:
     output_area.text_area("개념기반 교육과정", value=st.session_state.generated_curriculum, height=600)