Update app.py

app.py

@@ -2,11 +2,10 @@ import os
 import time
 import google.generativeai as genai
 import streamlit as st
-import re
 from streamlit_chat import message
 
 # Google Generative AI API 설정
-genai.configure(api_key=os.environ["GEMINI_API_KEY"])
+genai.configure(api_key=os.environ.get("GEMINI_API_KEY"))
 
 # 모델 설정
 generation_config = {
@@ -26,6 +25,7 @@ safety_settings = [
 # 전역 변수로 chat_session 초기화
 chat_session = None
 
+# 새로운 채팅 세션 시작 함수
 def start_new_chat_session():
     global chat_session
     chat_session = genai.GenerativeModel(
@@ -39,67 +39,40 @@ def start_new_chat_session():
         """,
     ).start_chat(history=[])
 
-# 초기 세션 시작
-start_new_chat_session()
+# Streamlit UI 설정
+st.title("실시간 학습 보조 채팅 앱")
 
-# Streamlit 앱 설정
-st.set_page_config(page_title="질문하는 AI 선생님", layout="centered")
-
-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 "history" not in st.session_state:
-    st.session_state.history = []
-
-# 채팅 세션 초기화
+# 세션 상태를 통해 채팅 히스토리 관리
+if "messages" not in st.session_state:
+    st.session_state.messages = []
 if "chat_session" not in st.session_state:
+    start_new_chat_session()
     st.session_state.chat_session = chat_session
 
-# 사용자 입력 받기
-user_input = st.text_input(
-    "입력", key="user_input", placeholder="여기에 학습 목표나 질문을 입력하세요...",
-    help="학습 목표를 입력하거나 학습하고자 하는 주제에 대한 질문을 입력하세요."
-)
-
-def respond(user_input):
-    st.session_state.history.append({"user": user_input, "ai": ""})
-    try:
-        response = st.session_state.chat_session.send_message(user_input, stream=True)
-        full_text = ""
-        
-        # 스트리밍 응답 처리
-        for chunk in response:
-            chunk_text = chunk.text
-            
-            # 단어 단위로 분리하여 실시간 출력
-            for word in chunk_text.split():
-                full_text += word + " "
-                
-                # 히스토리 최신 항목 업데이트
-                st.session_state.history[-1]["ai"] = full_text
-                time.sleep(0.05)  # 출력 속도 조절
+# 사용자가 입력한 메시지 처리
+user_input = st.text_input("당신: ", key="user_input")
+if user_input:
+    # 사용자 메시지 추가
+    st.session_state.messages.append({"role": "user", "content": user_input})
 
-    except Exception as e:
-        st.session_state.history[-1]["ai"] = f"에러 발생: {str(e)}"
+    # AI 응답 생성
+    response = st.session_state.chat_session.send_message(user_input)
+    ai_message = response.text
+    st.session_state.messages.append({"role": "assistant", "content": ai_message})
 
-# 응답 생성 및 출력
-if user_input:
-    respond(user_input)
+    # 입력 필드 초기화
+    st.experimental_rerun()
 
-# 채팅 메시지 출력
-for chat in st.session_state.history:
-    if "user" in chat:
-        message(chat["user"], is_user=True, key=f"user_{chat['user']}")
-    if "ai" in chat:
-        message(chat["ai"], is_user=False, key=f"ai_{chat['ai']}")
+# 채팅 히스토리 표시
+for message in st.session_state.messages:
+    if message["role"] == "user":
+        message(message["content"], is_user=True)
+    else:
+        message(message["content"], is_user=False)
 
-# 초기화 버튼 클릭 시
-if st.button("초기화"):
+# 새로운 채팅 세션 버튼
+if st.button("새로운 채팅 시작"):
     start_new_chat_session()
-    st.session_state.history = []
+    st.session_state.chat_session = chat_session
+    st.session_state.messages = []
+    st.experimental_rerun()