Update app.py

app.py

@@ -87,6 +87,10 @@ st.markdown("<div class='main-title'>AI 튜터 🎓</div>", unsafe_allow_html=Tr
 model_options = ["learnlm-1.5-pro-experimental", "gemini-1.5-flash"]
 selected_model_name = st.sidebar.selectbox("사용할 모델을 선택하세요:", model_options)
 
+# 초등학교 학년 선택
+grades = ["1학년", "2학년", "3학년", "4학년", "5학년", "6학년"]
+selected_grade = st.sidebar.selectbox("학년을 선택하세요:", grades)
+
 model = genai.GenerativeModel(
     model_name=selected_model_name,
     generation_config=generation_config,
@@ -96,7 +100,7 @@ model = genai.GenerativeModel(
 # 채팅 세션 초기화
 if "messages" not in st.session_state or st.session_state.get("current_model") != selected_model_name:
     st.session_state.messages = [
-        {"role": "system", "content": "학생들이 수업 주제나 내용을 입력하면, 질문을 통해서 학생들 스스로가 그 주제나 내용을 이해할 수 있도록, 마치 소크라테스처럼 끊임 없는 질문을 통해 스스로 깨달을 수 있도록 도와주세요."}
+        {"role": "system", "content": f"{selected_grade} 수준에 맞추어 학생들이 수업 주제나 내용을 입력하면, 질문을 통해서 학생들 스스로가 그 주제나 내용을 이해할 수 있도록, 마치 소크라테스처럼 끊임 없는 질문을 통해 스스로 깨달을 수 있도록 도와주세요."}
     ]
     st.session_state.chat_session = model.start_chat(history=[])
     st.session_state.current_model = selected_model_name
@@ -123,7 +127,7 @@ if prompt := st.chat_input("📝학습 내용을 입력하면 AI 튜터가 질
     st.markdown(f"<div class='chat-message chat-message-user'>{prompt}<img src='{user_icon_url}' class='chat-avatar chat-avatar-user'></div>", unsafe_allow_html=True)
 
     # Gemini API 호출
-    response = st.session_state.chat_session.send_message(prompt)
+    response = st.session_state.chat_session.send_message(f"[{selected_grade}] {prompt}")
     st.session_state.messages.append({"role": "assistant", "content": response.text})
     st.markdown(f"<div class='chat-message chat-message-assistant'><img src='{assistant_icon_url}' class='chat-avatar'>{response.text}</div>", unsafe_allow_html=True)