Update app.py

app.py

@@ -1,10 +1,12 @@
 import os
-import streamlit as st
+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.getenv("GEMINI_API_KEY"))
+genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 
 # 모델 설정
 generation_config = {
@@ -31,44 +33,73 @@ def start_new_chat_session():
         safety_settings=safety_settings,
         generation_config=generation_config,
         system_instruction="""
-        교사가 학습 목표를 입력하면, 학생들이 학습 목표에 도달할 수 있게 질문을 계속 해줘,
-        학생들은 네 질문에 답을 할 것이야. 그러면 바로 답을 제시하지 말고 질문형태로 피드백을 해줘.
+        교사가 학습 목표를 입력하면, 학생들이 학습 목표에 도달할 수 있게 질문을 계속 해줘, 
+        학생들은 네 질문에 답을 할 것이야. 그러면 바로 답을 제시하지 말고 질문형태로 피드백을 해줘. 
         질문을 통해서 학생들이 스스로 학습 목표를 도달할 수 있도록 도와줘.
         """,
     ).start_chat(history=[])
 
-# Streamlit 웹 앱 시작
-st.title("교육용 Google Generative AI 챗봇")
+# 초기 세션 시작
+start_new_chat_session()
 
-# 초기화 버튼으로 새로운 채팅 세션 시작
-if st.button("새로운 대화 시작"):
-    start_new_chat_session()
-    st.session_state.chat_history = []
-    st.write("새로운 대화가 시작되었습니다!")
+# Streamlit 앱 설정
+st.set_page_config(page_title="질문하는 AI 선생님", layout="centered")
 
-# 사용자의 입력 받기
-user_input = st.text_input("학습 목표를 입력하거나 질문하세요:")
+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 'chat_history' not in st.session_state:
-    st.session_state.chat_history = []
-    start_new_chat_session()
+# 채팅 히스토리 관리
+if "history" not in st.session_state:
+    st.session_state.history = []
+
+# 채팅 세션 초기화
+if "chat_session" not in st.session_state:
+    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)  # 출력 속도 조절
+
+    except Exception as e:
+        st.session_state.history[-1]["ai"] = f"에러 발생: {str(e)}"
+
+# 응답 생성 및 출력
 if user_input:
-    # 사용자의 입력을 히스토리에 추가
-    st.session_state.chat_history.append({"role": "user", "content": user_input})
-    
-    # 챗봇의 응답 생성
-    response = chat_session.send_message(user_input)
-    chatbot_reply = response.get('content', '오류가 발생했습니다. 다시 시도해주세요.')
-    
-    # 챗봇의 응답을 히스토리에 추가
-    st.session_state.chat_history.append({"role": "bot", "content": chatbot_reply})
+    respond(user_input)
 
-# 채팅 히스토리 표시
-for i, chat in enumerate(st.session_state.chat_history):
-    if chat["role"] == "user":
-        message(chat['content'], is_user=True, key=f"user_{i}")
-    else:
-        message(chat['content'], key=f"bot_{i}")
+# 채팅 메시지 출력
+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("초기화"):
+    start_new_chat_session()
+    st.session_state.history = []