Update app.py

app.py

@@ -2,63 +2,56 @@ import os
 import streamlit as st
 import google.generativeai as genai
 
-# Hugging Face Spaces에서 환경 변수 설정 필요
+# Gemini API 키 설정
 genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 
 # 모델 설정
 generation_config = {
-    "temperature": 1,
-    "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="교사가 학습 목표를 입력하면, 학생들이 학습 목표에 도달할 수 있게 질문을 계속 해줘, \n학생들은 네 질문에 답을 할 것이야. 그러면 바로 답을 제시하지 말고 질문형태로 피드백을 해줘. \n 질문을 통해서 학생들이 스스로 학습 목표를 도달할 수 있도록 도와줘.",
+  model_name="learnlm-1.5-pro-experimental",
+  generation_config=generation_config,
+  system_instruction="친절하고 든든한 튜터가 되어 학생이 목표를 달성하도록 안내하고, 학생이 주제에서 벗어나면 부드럽게 다시 집중하도록 도와주세요. 학생이 큰 개념을 단계적으로 이해하도록 돕는 질문을 하고, 그 개념을 깊이 파고들 수 있도록 탐구 질문을 던지세요. 학생이 부담을 느끼지 않도록 한 번에 한 질문만 하세요. 학생이 이해한 것을 보여주면 대화를 마무리하세요.",
 )
 
 
-# Streamlit 앱 설정
-st.set_page_config(page_title="질문하는 AI 선생님", layout="centered")
+# Streamlit 앱
+st.title("AI 튜터 챗봇")
 
-st.markdown(
-    """
-    <div style='font-size: 30px; font-weight: bold; text-align: center;'>질문하는 AI 선생님</div>
-    <div style='text-align: center; margin-bottom: 20px;'>학습주제를 입력하면, AI가 끊임 없는 질문을 통해 스스로 학습하게 도와줍니다.</div>
-    """,
-    unsafe_allow_html=True
-)
-
-# 세션 상태 초기화 (채팅 기록 저장)
+# 채팅 기록 저장
 if "messages" not in st.session_state:
     st.session_state.messages = []
 
-# 학습 목표 입력
-learning_objective = st.text_input("학습 목표를 입력하세요:")
-if learning_objective and st.session_state.messages == []:
-    st.session_state.messages.append({"role": "user", "content": learning_objective})
-
 # 채팅 기록 표시
 for message in st.session_state.messages:
-    with st.chat_message(message["role"]):
-        st.markdown(message["content"])
+    if message["role"] == "user":
+        st.chat_message("user").write(message["content"])
+    else:
+        st.chat_message("assistant").write(message["content"])
 
 # 사용자 입력 받기
-if prompt := st.chat_input("질문 또는 답변을 입력하세요:"):
-    st.session_state.messages.append({"role": "user", "content": prompt})
-    
-    with st.chat_message("assistant"):
-        full_response = ""
-        placeholder = st.empty()  # 텍스트 업데이트를 위한 placeholder
+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)
 
-        chat_session = model.start_chat(history=st.session_state.messages[:-1])
-        response = chat_session.send_message_streaming(prompt)
 
-        for chunk in response:
-            full_response += chunk.text
-            placeholder.markdown(full_response) # 실시간 업데이트
+    # 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": full_response})
+    # 응답 처리 및 채팅 기록에 추가
+    st.session_state.messages.append({"role": "assistant", "content": response.text})
+    st.chat_message("assistant").write(response.text)