Update app.py

app.py

@@ -2,56 +2,47 @@ import os
 import streamlit as st
 import google.generativeai as genai
 
-# Gemini API 키 설정
+# Gemini API 설정
 genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 
 # 모델 설정
 generation_config = {
-  "temperature": 0.5,
-  "top_p": 0.95,
-  "top_k": 64,
-  "max_output_tokens": 8192,
-  "response_mime_type": "text/plain",
+    "temperature": 0.5,
+    "top_p": 0.95,
+    "top_k": 64,
+    "max_output_tokens": 8192,
+    "response_mime_type": "text/plain",
 }
 
 model = genai.GenerativeModel(
-  model_name="learnlm-1.5-pro-experimental",
-  generation_config=generation_config,
-  system_instruction="친절하고 든든한 튜터가 되어 학생이 목표를 달성하도록 안내하고, 학생이 주제에서 벗어나면 부드럽게 다시 집중하도록 도와주세요. 학생이 큰 개념을 단계적으로 이해하도록 돕는 질문을 하고, 그 개념을 깊이 파고들 수 있도록 탐구 질문을 던지세요. 학생이 부담을 느끼지 않도록 한 번에 한 질문만 하세요. 학생이 이해한 것을 보여주면 대화를 마무리하세요.",
+    model_name="learnlm-1.5-pro-experimental",
+    generation_config=generation_config,
+    system_instruction="친절하고 든든한 튜터가 되어 학생이 목표를 달성하도록 안내하고, 학생이 주제에서 벗어나면 부드럽게 다시 집중하도록 도와주세요. 학생이 큰 개념을 단계적으로 이해하도록 돕는 질문을 하고, 그 개념을 깊이 파고들 수 있도록 탐구 질문을 던지세요. 학생이 부담을 느끼지 않도록 한 번에 한 질문만 하세요. 학생이 이해한 것을 보여주면 대화를 마무리하세요.",
 )
 
+# 채팅 세션 초기화
+if "chat_session" not in st.session_state:
+    st.session_state.chat_session = model.start_chat(history=[])
 
-# Streamlit 앱
+# 스트림릿 UI 설정
 st.title("AI 튜터 챗봇")
 
-# 채팅 기록 저장
+# 채팅 메시지 표시
 if "messages" not in st.session_state:
     st.session_state.messages = []
 
-# 채팅 기록 표시
 for message in st.session_state.messages:
-    if message["role"] == "user":
-        st.chat_message("user").write(message["content"])
-    else:
-        st.chat_message("assistant").write(message["content"])
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
 
 # 사용자 입력 받기
-user_input = st.chat_input("질문을 입력하세요.")
-
-# 사용자 입력 처리
-if user_input:
-    # 채팅 기록에 사용자 입력 추가
-    st.session_state.messages.append({"role": "user", "content": user_input})
-    st.chat_message("user").write(user_input)
-
+if prompt := st.chat_input("학습 내용을 입력하면 AI 튜터가 질문을 통해서 개념을 이해하도록 도와줍니다."):
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    with st.chat_message("user"):
+        st.markdown(prompt)
 
     # Gemini API 호출
-    chat_session = model.start_chat(
-        history=st.session_state.messages
-    )
-  
-    response = chat_session.send_message(user_input)
-
-    # 응답 처리 및 채팅 기록에 추가
-    st.session_state.messages.append({"role": "assistant", "content": response.text})
-    st.chat_message("assistant").write(response.text)
+    response = st.session_state.chat_session.send_message(prompt)
+    with st.chat_message("assistant"):
+        st.markdown(response.text)
+    st.session_state.messages.append({"role": "assistant", "content": response.text})