Update app.py

app.py

@@ -2,10 +2,11 @@ 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.get("GEMINI_API_KEY"))
+genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 
 # 모델 설정
 generation_config = {
@@ -25,7 +26,6 @@ safety_settings = [
 # 전역 변수로 chat_session 초기화
 chat_session = None
 
-# 새로운 채팅 세션 시작 함수
 def start_new_chat_session():
     global chat_session
     chat_session = genai.GenerativeModel(
@@ -39,40 +39,67 @@ def start_new_chat_session():
         """,
     ).start_chat(history=[])
 
-# Streamlit UI 설정
-st.title("실시간 학습 보조 채팅 앱")
+# 초기 세션 시작
+start_new_chat_session()
 
-# 세션 상태를 통해 채팅 히스토리 관리
-if "messages" not in st.session_state:
-    st.session_state.messages = []
+# 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 "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")
-if user_input:
-    # 사용자 메시지 추가
-    st.session_state.messages.append({"role": "user", "content": user_input})
+# 사용자 입력 받기
+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)  # 출력 속도 조절
 
-    # 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})
+    except Exception as e:
+        st.session_state.history[-1]["ai"] = f"에러 발생: {str(e)}"
 
-    # 입력 필드 초기화
-    st.experimental_rerun()
+# 응답 생성 및 출력
+if user_input:
+    respond(user_input)
 
-# 채팅 히스토리 표시
-for message in st.session_state.messages:
-    if message["role"] == "user":
-        message(message["content"], is_user=True)
-    else:
-        message(message["content"], is_user=False)
+# 채팅 메시지 출력
+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']}")
 
-# 새로운 채팅 세션 버튼
-if st.button("새로운 채팅 시작"):
+# 초기화 버튼 클릭 시
+if st.button("초기화"):
     start_new_chat_session()
-    st.session_state.chat_session = chat_session
-    st.session_state.messages = []
-    st.experimental_rerun()
+    st.session_state.history = []