Update app.py

app.py

@@ -74,7 +74,7 @@ def generate_curriculum(grade, achievement_standard):
         SYSTEM_PROMPT,
         f"**대상 학년:** 초등학교 {grade}학년",
         f"**성취기준 및 주요 내용:** {achievement_standard}",
-        "**개념기반 교육과정:**",
+        "**개념기반 교육과정:**", 
     ]
 
     full_text = ""
@@ -93,19 +93,16 @@ def generate_curriculum(grade, achievement_standard):
 colored_header(
     label="개념기반 교육과정 설계 도우미",
     description="초등학교 교육과정에 맞는 개념기반 교육과정을 생성합니다.",
-    color_name="blue-70",  # 색상 변경 가능
+    color_name="blue-70",
 )
 
-add_vertical_space(1)  # 공간 추가
+add_vertical_space(1)
 
 # 입력 위젯 (사이드바 사용, 확장 가능)
-with st.sidebar.expander("입력 설정", expanded=True):  # 확장 가능 패널
-    grade_options = [
-        "초등학교 1학년", "초등학교 2학년", "초등학교 3학년", "초등학교 4학년",
-        "초등학교 5학년", "초등학교 6학년"
-    ]
+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 = grade_options.index(grade_selected) + 1
+    grade = int(grade_selected.split()[1][:-1]) # 학년 추출
 
     achievement_standard = st.text_area("성취기준 및 주요 내용 입력", height=200)
 
@@ -113,7 +110,7 @@ generate_button = st.button("교육과정 생성")
 
 # 생성 버튼 클릭시 동작
 if generate_button and achievement_standard:
-    with st.spinner("생성 중..."): # 로딩 표시 추가
+    with st.spinner("생성 중..."): # 로딩 표시
         result = generate_curriculum(grade, achievement_standard)
         st.session_state.generated_curriculum = result
         st.text_area("개념기반 교육과정", value=result, height=600)