Update app.py

app.py

@@ -1,8 +1,6 @@
 import os
-import time
 import openai
 import streamlit as st
-from streamlit_chat import message
 
 # OpenAI API 설정
 openai_api_key = os.getenv("OPENAI_API_KEY")
@@ -11,11 +9,8 @@ openai.api_key = openai_api_key
 def openai_chat(text):
     try:
         response = openai.ChatCompletion.create(
-            model="gpt-4o",
-            messages=[
-                {"role": "system", "content": "학생들이 수업 주제나 내용을 입력하면, 질문을 통해서 학생들 스스로가 그 주제나 내용을 이해할 수 있도록, 마치 소크라테스처럼 끊임 없는 질문을 통해서 스스로 깨달을 수 있도록 도와주세요. 친절하고 든든한 튜터가 되어 학생이 목표를 달성하도록 안내하고, 학생이 주제에서 벗어나면 부드럽게 다시 집중하도록 도와주세요. 학생이 큰 개념을 단계적으로 이해하도록 돕는 질문을 하고, 그 개념을 깊이 파고들 수 있도록 탐구 질문을 던지세요. 학생이 부담을 느끼지 않도록 한 번에 한 질문만 하세요. 학생이 이해한 것을 보여주면 대화를 마무리하세요. 정답을 제시하기 보다는 질문을 통해서 학생이 스스로 이해에 도달할 수 있도록 도와주세요."},
-                {"role": "user", "content": text}
-            ],
+            model="gpt-4",
+            messages=st.session_state.messages,  # 전체 대화 세션 전달
             temperature=0.7,
             max_tokens=2000,
             top_p=0.9,
@@ -102,39 +97,39 @@ st.markdown(
 st.markdown("<div class='main-title'>AI 튜터 🎓</div>", unsafe_allow_html=True)
 
 # 채팅 세션 초기화
-if "chat_session" not in st.session_state:
-    st.session_state.chat_session = []
-
-# 채팅 메시지 표시
 if "messages" not in st.session_state:
-    st.session_state.messages = []
+    st.session_state.messages = [
+        {"role": "system", "content": "학생들이 수업 주제나 내용을 입력하면, 질문을 통해서 학생들 스스로가 그 주제나 내용을 이해할 수 있도록, 마치 소크라테스처럼 끊임 없는 질문을 통해 스스로 깨달을 수 있도록 도와주세요. 친절하고 든든한 튜터가 되어 학생이 목표를 달성하도록 안내하고, 학생이 주제에서 벗어나면 부드럽게 다시 집중하도록 도와주세요. 학생이 큰 개념을 단계적으로 이해하도록 돕는 질문을 하고, 그 개념을 깊이 파고들 수 있도록 탐구 질문을 던지세요. 학생이 부담을 느끼지 않도록 한 번에 한 질문만 하세요. 학생이 이해한 것을 보여주면 대화를 마무리하세요. 정답을 제시하기 보다는 질문을 통해서 학생이 스스로 이해에 도달할 수 있도록 도와주세요."}
+    ]
 
 # 사용자와 AI의 아이콘 URL 설정
 user_icon_url = "https://cdn-icons-png.flaticon.com/512/1995/1995531.png"  # 학생 아이콘 (책을 든 학생)
 assistant_icon_url = "https://cdn-icons-png.flaticon.com/512/4323/4323008.png"  # 튜터 아이콘 (안경 쓴 선생님)
 
+# 채팅 메시지 표시
 for message in st.session_state.messages:
-    role_class = "chat-message-user" if message["role"] == "user" else "chat-message-assistant"
-    avatar_url = user_icon_url if message["role"] == "user" else assistant_icon_url
-    avatar_class = "chat-avatar-user" if message["role"] == "user" else "chat-avatar"
-    st.markdown(
-        f"<div class='chat-message {role_class}'><img src='{avatar_url}' class='chat-avatar {avatar_class}'>{message['content']}</div>",
-        unsafe_allow_html=True
-    )
+    if message["role"] != "system":
+        role_class = "chat-message-user" if message["role"] == "user" else "chat-message-assistant"
+        avatar_url = user_icon_url if message["role"] == "user" else assistant_icon_url
+        avatar_class = "chat-avatar-user" if message["role"] == "user" else "chat-avatar"
+        st.markdown(
+            f"<div class='chat-message {role_class}'><img src='{avatar_url}' class='chat-avatar {avatar_class}'>{message['content']}</div>",
+            unsafe_allow_html=True
+        )
 
 # 사용자 입력 받기
 if prompt := st.chat_input("🖋학습 내용을 입력하면 AI 튜터가 질문을 통해서 개념을 이해하도록 도와줍니다."):
+    # 사용자의 메시지를 세션에 추가
     st.session_state.messages.append({"role": "user", "content": prompt})
     st.markdown(f"<div class='chat-message chat-message-user'><img src='{user_icon_url}' class='chat-avatar chat-avatar-user'>{prompt}</div>", unsafe_allow_html=True)
 
     # OpenAI API 호출
     response = openai_chat(prompt)
-    st.markdown(f"<div class='chat-message chat-message-assistant'><img src='{assistant_icon_url}' class='chat-avatar'>{response}</div>", unsafe_allow_html=True)
     st.session_state.messages.append({"role": "assistant", "content": response})
+    st.markdown(f"<div class='chat-message chat-message-assistant'><img src='{assistant_icon_url}' class='chat-avatar'>{response}</div>", unsafe_allow_html=True)
 
 # ���기화 버튼을 사이드바로 이동
 with st.sidebar:
     if st.button("💡 초기화"):
-        st.session_state.chat_session = []
-        st.session_state.messages = []
+        del st.session_state.messages  # 메시지 세션 상태 삭제
         st.rerun()