Update app.py

app.py

@@ -9,15 +9,24 @@ openai.api_key = openai_api_key
 def openai_chat(text, grade_level):
     try:
         # 학년 수준에 따른 프롬프트 추가
-        grade_prompt = f"이 학생은 초등학교 {grade_level}학년입니다. 질문과 설명을 이 학년 수준에 맞게 해주세요.(학생의 학년에 맞게 질문과 답하는 내용의 범위를 해당 학년의 교육과정에 맞게 범위를 조정하세요."
-        
+        grade_prompt = ""
+        if grade_level.startswith("초등학교"):
+            grade_num = grade_level[5:].rstrip("학년")
+            grade_prompt = f"이 학생은 초등학교 {grade_num}학년입니다. 질문과 설명을 이 학년 수준에 맞게 해주세요.(학생의 학년에 맞게 질문과 답하는 내용의 범위를 해당 학년의 교육과정에 맞게 범위를 조정하세요."
+        elif grade_level.startswith("중학교"):
+            grade_num = grade_level[3:].rstrip("학년")
+            grade_prompt = f"이 학생은 중학교 {grade_num}학년입니다. 질문과 설명을 이 학년 수준에 맞게 해주세요.(학생의 학년에 맞게 질문과 답하는 내용의 범위를 해당 학년의 교육과정에 맞게 범위를 조정하세요."
+        elif grade_level.startswith("고등학교"):
+            grade_num = grade_level[4:].rstrip("학년")
+            grade_prompt = f"이 학생은 고등학교 {grade_num}학년입니다. 질문과 설명을 이 학년 수준에 맞게 해주세요.(학생의 학년에 맞게 질문과 답하는 내용의 범위를 해당 학년의 교육과정에 맞게 범위를 조정하세요."
+
         # 시스템 메시지를 세션 시작 시 한 번만 추가
         if not any(message["role"] == "system" for message in st.session_state.messages):
             st.session_state.messages.append({"role": "system", "content": grade_prompt})
         else:
             # 기존 시스템 메시지 업데이트 (선택 사항)
             st.session_state.messages[0]["content"] = grade_prompt
-        
+
         response = openai.ChatCompletion.create(
             model="gpt-4o",
             messages=st.session_state.messages,  # 전체 대화 세션 전달
@@ -112,7 +121,12 @@ assistant_icon_url = "https://cdn-icons-png.flaticon.com/512/4323/4323008.png"
 
 # 학년 수준 선택
 with st.sidebar:
-    grade_level = st.selectbox("📚 학년 수준을 선택하세요:", ["1", "2", "3", "4", "5", "6"], index=0, format_func=lambda x: f"초등학교 {x}학년")
+    grade_level_options = [
+        "초등학교 1학년", "초등학교 2학년", "초등학교 3학년", "초등학교 4학년", "초등학교 5학년", "초등학교 6학년",
+        "중학교 1학년", "중학교 2학년", "중학교 3학년",
+        "고등학교 1학년", "고등학교 2학년", "고등학교 3학년"
+    ]
+    grade_level = st.selectbox("📚 학년 수준을 선택하세요:", grade_level_options, index=0)
     # 초기화 버튼
     if st.button("💡 초기화"):
         del st.session_state.messages  # 메시지 세션 상태 삭제