Update app.py

app.py

@@ -7,10 +7,10 @@ genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 
 # 모델 설정
 generation_config = {
-    "temperature": 0.5,
+    "temperature": 0.7,
     "top_p": 0.95,
     "top_k": 40,
-    "max_output_tokens": 8192,
+    "max_output_tokens": 2000,
     "response_mime_type": "text/plain",
 }
 
@@ -25,7 +25,6 @@ st.markdown(
         .stApp {
             background-color: #fffafa; /* 은은한 핑크색 배경 */
         }
-
         /* 타이틀 스타일 */
         .main-title {
             font-size: 3rem;
@@ -34,7 +33,6 @@ st.markdown(
             text-align: center;
             margin-bottom: 20px;
         }
-
         /* 채팅 메시지 스타일 */
         .chat-message {
             border-radius: 15px;
@@ -42,38 +40,34 @@ st.markdown(
             margin: 10px 0;
             display: flex;
             align-items: center;
+            flex-wrap: wrap;
+            word-break: break-word;
         }
-
         .chat-message-user {
             background-color: #ffebef; /* 따뜻한 파스텔 핑크 */
             color: #8b4513;
             justify-content: flex-end;
         }
-
         .chat-message-assistant {
             background-color: #ffe4e6; /* 부드러운 파스텔 복숭아 핑크색 */
             color: #6b4226;
             justify-content: flex-start;
         }
-
         .chat-avatar {
             width: 40px;
             height: 40px;
             border-radius: 50%;
             margin-right: 10px;
         }
-
         .chat-avatar-user {
             margin-left: 10px;
             margin-right: 0;
         }
-
         /* 사용자 입력 창 스타일 */
         .stTextInput input {
             border-radius: 15px;
             border: 2px solid #cd857f; /* 따뜻한 핑크색 */
         }
-
         /* 버튼 스타일 */
         .stButton button {
             background-color: #cd857f; /* 따뜻한 핑크색 */
@@ -89,41 +83,38 @@ st.markdown(
 # 메인 타이틀
 st.markdown("<div class='main-title'>AI 튜터 🎓</div>", unsafe_allow_html=True)
 
-# 모델 선택 옵션 추가
-model_options = ["gemini-1.5-flash", "learnlm-1.5-pro-experimental"]
-selected_model_name = st.sidebar.selectbox("사용할 모델을 선택하세요:", model_options)
-
 # 모델 설정
 model = genai.GenerativeModel(
-    model_name=selected_model_name,
+    model_name="gemini-1.5-flash",
     generation_config=generation_config,
     system_instruction="친절하고 든든한 튜터가 되어 학생이 목표를 달성하도록 안내하고, 학생이 주제에서 벗어나면 부드럽게 다시 집중하도록 도와주세요. 학생이 큰 개념을 단계적으로 이해하도록 돕는 질문을 하고, 그 개념을 깊이 파고들 수 있도록 탐구 질문을 던지세요. 학생이 부담을 느끼지 않도록 한 번에 한 질문만 하세요. 학생이 이해한 것을 보여주면 대화를 마무리하세요.",
 )
 
 # 채팅 세션 초기화
-if "chat_session" not in st.session_state or st.session_state.get("current_model") != selected_model_name:
+if "messages" not in st.session_state:
+    st.session_state.messages = [
+        {"role": "system", "content": "학생들이 수업 주제나 내용을 입력하면, 질문을 통해서 학생들 스스로가 그 주제나 내용을 이해할 수 있도록, 마치 소크라테스처럼 끊임 없는 질문을 통해 스스로 깨달을 수 있도록 도와주세요."}
+    ]
     st.session_state.chat_session = model.start_chat(history=[])
-    st.session_state.current_model = selected_model_name
 
 # 사용자와 AI의 아이콘 URL 설정
 user_icon_url = "https://cdn-icons-png.flaticon.com/512/1995/1995531.png"  # 학생 아이콘 (책을 든 학생)
 assistant_icon_url = "https://cdn-icons-png.flaticon.com/512/4323/4323008.png"  # 튜터 아이콘 (안경 쓴 선생님)
 
 # 채팅 메시지 표시
-if "messages" not in st.session_state:
-    st.session_state.messages = []
-
 for message in st.session_state.messages:
-    role_class = "chat-message-user" if message["role"] == "user" else "chat-message-assistant"
-    avatar_url = user_icon_url if message["role"] == "user" else assistant_icon_url
-    avatar_class = "chat-avatar-user" if message["role"] == "user" else "chat-avatar"
-    st.markdown(
-        f"<div class='chat-message {role_class}'><img src='{avatar_url}' class='chat-avatar {avatar_class}'>{message['content']}</div>",
-        unsafe_allow_html=True
-    )
+    if message["role"] != "system":
+        role_class = "chat-message-user" if message["role"] == "user" else "chat-message-assistant"
+        avatar_url = user_icon_url if message["role"] == "user" else assistant_icon_url
+        avatar_class = "chat-avatar-user" if message["role"] == "user" else "chat-avatar"
+        st.markdown(
+            f"<div class='chat-message {role_class}'><img src='{avatar_url}' class='chat-avatar {avatar_class}'>{message['content']}</div>",
+            unsafe_allow_html=True
+        )
 
 # 사용자 입력 받기
 if prompt := st.chat_input("📝학습 내용을 입력하면 AI 튜터가 질문을 통해서 개념을 이해하도록 도와줍니다."):
+    # 사용자의 메시지를 세션에 추가
     st.session_state.messages.append({"role": "user", "content": prompt})
     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)
 
@@ -135,5 +126,5 @@ if prompt := st.chat_input("📝학습 내용을 입력하면 AI 튜터가 질
 # 초기화 버튼을 사이드바로 이동
 with st.sidebar:
     if st.button("💡 초기화"):
-        st.session_state.clear()  # 세션 상태 초기화
+        del st.session_state.messages  # 메시지 세션 상태 삭제
         st.rerun()