Update app.py

app.py

@@ -1,480 +1,141 @@
 import os
+import openai
 import streamlit as st
-import datetime
-import google.generativeai as genai
-import time
-
-# --- Streamlit 설정 ---
-st.set_page_config(
-    page_title="마음 건강 다이어리 💖",
-    page_icon="💖",
-    layout="wide",
-)
-
-# --- Custom CSS (스타일링) ---
-st.markdown(
-    """
-<style>
-/* 전체 폰트 변경 (Nanum Gothic, Google Fonts CDN 사용) */
-@import url('https://fonts.googleapis.com/css2?family=Nanum+Gothic:wght@400;700&display=swap');
-body {
-    font-family: 'Nanum Gothic', sans-serif !important;
-}
-
-/* 섹션 제목 스타일 */
-.section-title {
-    font-size: 28px; /* 더 크게 */
-    font-weight: bold;
-    color: #212529;
-    margin-bottom: 25px; /* 간격 증가 */
-    padding-bottom: 15px; /* 패딩 증가 */
-    border-bottom: 3px dotted #ffb3ba; /* 더 굵게 */
-}
-
-/* 섹션 부제목 스타일 */
-.section-subtitle {
-    font-size: 18px; /* 약간 더 크게 */
-    color: #555;
-    margin-bottom: 15px; /* 간격 증가 */
-    font-weight: 500; /* 살짝 굵게 */
-}
-
-/* 버튼 스타일 (둥근 모양, 파스텔톤, 그림자 효과, 강조) */
-div.stButton > button {
-    background-color: #ffb3ba; /* 메인 버튼 색상 (분홍) */
-    color: white;
-    padding: 14px 28px; /* 패딩 증가 */
-    font-size: 18px; /* 폰트 크기 증가 */
-    border-radius: 25px; /* 더 둥근 모양 */
-    border: none;
-    box-shadow: 5px 5px 10px rgba(0,0,0,0.2); /* 그림자 효과 더 강하게 */
-    transition: background-color 0.3s ease, transform 0.3s ease; /* 변환 효과 추가 */
-    font-weight: bold; /* 텍스트 굵게 */
-}
-div.stButton > button:hover {
-    background-color: #ff808a; /* 호버 시 버튼 색상 (더 진한 분홍) */
-    transform: scale(1.07); /* 살짝 더 커지는 효과 */
-    box-shadow: 5px 5px 15px rgba(0,0,0,0.3); /* 호버 시 그림자 효과 증가 */
-}
-
-/* 보조 버튼 스타일 (회색, 둥근 모양) */
-div.stButton > button.secondary-button {
-    background-color: #ced4da; /* 보조 버튼 색상 */
-    color: #495057;
-    padding: 12px 24px; /* 패딩 약간 증가 */
-    font-size: 16px; /* 폰트 크기 유지 */
-    border-radius: 25px; /* 더 둥근 모양 */
-    border: none;
-    transition: background-color 0.3s ease;
-}
-div.stButton > button.secondary-button:hover {
-    background-color: #bac0c5; /* 호버 시 보조 버튼 색상 */
-}
-
-/* Expander 스타일 (제목 굵게, 밑줄) */
-.streamlit-expanderHeader {
-    font-weight: bold;
-    color: #212529;
-    border-bottom: 2px dotted #ffb3ba; /* 점선 밑줄 (분홍) */
-    padding-bottom: 10px; /* 패딩 증가 */
-    margin-bottom: 20px; /* 간격 증가 */
-    font-size: 17px; /* 폰트 약간 키움 */
-}
-
-/* Info, Success, Error, Warning Box 스타일 (파스텔톤, 아이콘 추가) */
-div.stInfo, div.stSuccess, div.stError, div.stWarning {
-    border-radius: 15px; /* 더 둥글게 */
-    padding: 20px; /* 패딩 증가 */
-    margin-bottom: 20px; /* 간격 증가 */
-    box-shadow: 5px 5px 10px rgba(0,0,0,0.08); /* 그림자 효과 약간 증가 */
-    border-left: none !important; /* 기존 왼쪽 테두리 제거 */
-    background-color: #ffe0e5 !important; /* 기본 배경색 (연한 분홍) */
-}
-div.stInfo {
-    background-color: #e0f7fa !important; /* Info 배경색 */
-}
-div.stSuccess {
-    background-color: #e8f5e9 !important; /* Success 배경색 */
-}
-div.stError {
-    background-color: #ffebee !important; /* Error 배경색 */
-}
-div.stWarning {
-    background-color: #fffde7 !important; /* Warning 배경색 */
-}
-
-/* Info, Success, Error, Warning Box 아이콘 스타일 */
-div.stInfo::before, div.stSuccess::before, div.stError::before, div.stWarning::before {
-    content: "";
-    display: inline-block;
-    width: 30px; /* 아이콘 크기 증가 */
-    height: 30px; /* 아이콘 크기 증가 */
-    margin-right: 15px; /* 간격 증가 */
-    vertical-align: middle;
-    background-size: contain;
-    background-repeat: no-repeat;
-}
-div.stInfo::before {
-    background-image: url('data:image/svg+xml,%3Csvg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="%2303a9f4"%3E%3Cpath d="M12 2C6.48 2 2 6.48 2 12s4.48 10 10 10 10-4.48 10-10S17.52 2 12 2zm1 15h-2v-6h2v6zm0-8h-2V5h2v4z"/%3E%3C/svg%3E'); /* Info 아이콘 (물음표) */
-}
-div.stSuccess::before {
-    background-image: url('data:image/svg+xml,%3Csvg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="%234caf50"%3E%3Cpath d="M9 16.17L4.83 12l-1.42 1.41L9 19 21 7l-1.41-1.41z"/%3E%3C/svg%3E'); /* Success 아이콘 (체크) */
-}
-div.stError::before {
-    background-image: url('data:image/svg+xml,%3Csvg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="%23f44336"%3E%3Cpath d="M12 2C6.48 2 2 6.48 2 12s4.48 10 10 10 10-4.48 10-10S17.52 2 12 2zm1 15h-2v-2h2v2zm0-4h-2V7h2v6z"/%3E%3C/svg%3E'); /* Error 아이콘 (느낌표) */
-}
-div.stWarning::before {
-    background-image: url('data:image/svg+xml,%3Csvg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="%23ff9800"%3E%3Cpath d="M10.29 3.86L1.83 18a2 2 0 0 0 1.73 3h16.94a2 2 0 0 0 1.73-3L13.71 3.86a2 2 0 0 0-3.42 0zM12 15c-.55 0-1-.45-1-1v-2c0-.55.45-1 1-1s1 .45 1 1v2c0 .55-.45 1-1 1zm0-5c-.55 0-1-.45-1-1s.45-1 1-1 1 .45 1 1 .45 1 1 1z"/%3E%3C/svg%3E'); /* Warning 아이콘 (경고) */
-}
-
-/* Chat message 스타일 */
-.stChatMessage[data-streamlit="true"] {
-    border-radius: 15px; /* 더 둥글게 */
-    padding: 15px 20px; /* 패딩 증가 */
-    margin-bottom: 15px; /* 간격 증가 */
-    box-shadow: 3px 3px 7px rgba(0,0,0,0.08); /* 그림자 효과 약간 증가 */
-    width: fit-content;
-    max-width: 80%;
-    font-size: 16px; /* 폰트 크기 약간 증가 */
-}
-
-.stChatMessage[data-streamlit="true"].user {
-    background-color: #e0f7fa; /* User message background: light blue-green, slightly lighter */
-    align-self: flex-end; /* User messages to the right */
-}
-
-.stChatMessage[data-streamlit="true"].assistant {
-    background-color: #f0f8ff; /* Assistant message background: very light gray, slightly lighter */
-    align-self: flex-start; /* Assistant messages to the left */
-}
-
-
-/* Toast message 스타일 */
-div.streamlit-toast-container {
-    z-index: 10000; /* Toast를 항상 맨 위에 표시 */
-}
-div[data-testid="stToast"] {
-    border-radius: 15px; /* 더 둥글게 */
-    padding: 20px; /* 패딩 증가 */
-    box-shadow: 5px 5px 10px rgba(0,0,0,0.1); /* 그림자 효과 약간 증가 */
-}
-
-/*챗 컨테이너 스타일 삭제 (더이상 필요 없음)*/
-
-</style>
-""",
-    unsafe_allow_html=True,
-    )
 
+# OpenAI API 설정
+openai_api_key = os.getenv("OPENAI_API_KEY")
+openai.api_key = openai_api_key
 
-# --- API 키 설정 (Hugging Face Secrets에서 관리 권장) ---
-if "GEMINI_API_KEY" not in os.environ:
-    st.error("GEMINI_API_KEY 환경 변수가 설정되지 않았습니다. API 키를 설정해주세요.")
-    st.stop()
-genai.configure(api_key=os.environ["GEMINI_API_KEY"])
-
-# --- Gemini 모델 설정 ---
-generation_config = {
-    "temperature": 0.7,
-    "top_p": 0.95,
-    "top_k": 64,
-    "max_output_tokens": 25000,
-    "response_mime_type": "text/plain",
-}
-model = genai.GenerativeModel(
-    model_name="gemini-2.0-flash-exp",  # 또는 "gemini-pro"
-    generation_config=generation_config,
-)
-
-
-# --- 세션 상태 초기화 ---
-if "daily_emotions" not in st.session_state:
-    st.session_state["daily_emotions"] = {}
-if "current_tab" not in st.session_state:
-    st.session_state["current_tab"] = "❤️ 감정 체크인" # 더 이상 탭이 없으므로 무의미하지만 유지
-if "chat_session" not in st.session_state:
-    st.session_state["chat_session"] = model.start_chat(history=[])
-if "emotion_analysis_result" not in st.session_state:
-    st.session_state["emotion_analysis_result"] = None
-if "counseling_messages" not in st.session_state:
-    st.session_state["counseling_messages"] = []
-if "analysis_to_chatbot" not in st.session_state:
-    st.session_state["analysis_to_chatbot"] = ""
-if "counseling_history" not in st.session_state: # 상담 로그 기록을 위한 세션 상태 추가
-    st.session_state["counseling_history"] = []
-if "section_index" not in st.session_state: # 현재 섹션 index 저장
-    st.session_state["section_index"] = 0
-
-
-# --- 감정 분석 함수 ---
-def analyze_emotion(emotion, reason):
-    prompt = f"""
-    **오늘의 감정:** {emotion}
-    **감정 이유:** {reason}
-
-    **지시:**
-    1.  **감정 분석:**  제시된 감정과 이유를 바탕으로 **심리적인 관점**에서 감정을 분석하고, 2~3 문장으로 요약해주세요.
-    2.  **격려 및 공감:** 분석 내용을 바탕으로 **따뜻하고 친절하게** 격려와 공감 메시지를 1~2 문장으로 덧붙여주세요.
-
-    **출력 형식:**
-    **[감정 분석]**
-    - [감정 분석 내용]
-
-    **[격려 및 공감]**
-    - [격려 및 공감 메시지]
-
-    **주의:**
-    -   초등학생 3학년이 이해하기 쉬운 단어로, 친절하고 객관적인 어조로, **존댓말**을 사용해주세요.
-    -   부정적인 감정에 대한 분석일 경우, 긍정적인 방향으로 전환될 수 있도록 격려하는 내용을 포함해주세요.
-    """
-    chat_session = st.session_state["chat_session"]
+def openai_chat(text, grade_level):
     try:
-        response = chat_session.send_message(prompt)
-        analysis_text = response.text.strip()
-        return analysis_text
-    except google.api_core.exceptions.ResourceExhausted as e:
-        st.error(
-            f"API 할당량 초과 오류가 발생했습니다. 잠시 후 다시 시도해주세요. 오류 메시지: {e}"
+        # 학년 수준에 따른 프롬프트 추가
+        grade_prompt = f"이 학생은 초등학교 {grade_level}학년입니다. 질문과 설명을 이 학년 수준에 맞게 해주세요.(학생의 학년에 맞게 질문과 답하는 내용의 범위를 해당 학년의 교육과정에 맞게 범위를 조정하세요."
+        
+        # 시스템 메시지를 세션 시작 시 한 ��만 추가
+        if not any(message["role"] == "system" for message in st.session_state.messages):
+            st.session_state.messages.append({"role": "system", "content": grade_prompt})
+        else:
+            # 기존 시스템 메시지 업데이트 (선택 사항)
+            st.session_state.messages[0]["content"] = grade_prompt
+        
+        response = openai.ChatCompletion.create(
+            model="gpt-4o",
+            messages=st.session_state.messages,  # 전체 대화 세션 전달
+            temperature=0.7,
+            max_tokens=1000, # max_tokens 값 조정
+            top_p=0.9,
+            frequency_penalty=0,
+            presence_penalty=0
         )
-        return None
-
-
-# --- 감정 기록 함수 ---
-def record_emotion(emotion, reason):
-    today = datetime.date.today()
-    st.session_state["daily_emotions"][str(today)] = {"emotion": emotion, "reason": reason}
-    st.toast(f"{today} 감정: {emotion} 😊 기록 완료!", icon="✅")
-    st.session_state["emotion_analysis_result"] = None
-    st.session_state["analysis_to_chatbot"] = None
-    st.session_state["counseling_messages"] = []
-    st.session_state["counseling_history"] = [] # 감정 기록 시 상담 기록 초기화
-    st.session_state["section_index"] += 1 # 다음 섹션으로 이동
-
-
-# --- 감정 이력 표시 함수 ---
-def display_emotion_history():
-    st.subheader("📊 감정 이력")
-    st.markdown("날짜별 감정 변화와 그때의 이유를 확인해보세요.")
-
-    if not st.session_state["daily_emotions"]:
-        st.info("아직 기록된 감정이 없어요. '💖 감정 체크인' 섹션에서 오늘 하루의 감정을 기록해보세요.") # 탭 -> 섹션으로 변경, 아이콘 변경
-    else:
-        sorted_dates = sorted(st.session_state["daily_emotions"].keys(), reverse=True)
-        for date_str in sorted_dates:
-            emotion_data = st.session_state["daily_emotions"][date_str]
-            st.write(f"- {date_str}: **{emotion_data['emotion']}**")
-            if emotion_data['reason']:
-                st.write(f"  > {emotion_data['reason']}")
-    st.markdown("---")
-
-
-# --- AI 감정 분석 및 상담 시작 함수 ---
-def start_emotion_analysis_and_counseling():
-    today = str(datetime.date.today())
-    emotion_data = st.session_state["daily_emotions"].get(today)
-    if emotion_data and emotion_data['reason']:
-        with st.spinner("AI가 감정을 분석하고 있어요..."):
-            emotion_analysis_result = analyze_emotion(emotion_data['emotion'], emotion_data['reason'])
-            st.session_state["emotion_analysis_result"] = emotion_analysis_result
-            st.session_state["analysis_to_chatbot"] = emotion_analysis_result
-            st.success("AI 감정 분석 완료! '🔮 AI 감정 분석' 섹션에서 결과를 확인하고, '💬 AI 상담 챗봇' 섹션에서 상담을 시작해보세요.") # 탭 -> 섹션으로 변경, 아이콘 변경
-            st.session_state["section_index"] += 1 # 다음 섹션으로 이동
-    else:
-        st.warning("오늘의 감정과 감정 이유를 먼저 기록해주세요.")
+        return response.choices[0].message["content"]
+    except Exception as e:
+        return f"에러 발생: {str(e)}"
 
+# Streamlit 앱 설정
+st.set_page_config(page_title="AI 튜터", page_icon="🎓", initial_sidebar_state="expanded")
 
-# --- 상담 안내 함수 ---
-def display_counseling_info():
-    st.subheader("📢 상담 안내")
-    st.markdown("마음이 힘들거나 어려운 일이 있을 때, 혼자 고민하지 마세요. ")
-    st.markdown("학교 선생님이나 부모님께 이야기하거나, 전문 상담 선생님의 도움을 받는 것도 좋은 방법입니다.")
-
-    with st.expander("🏫 학교 상담"):
-        st.markdown(
-            """
-            **학교 상담 선생님**께서는 여러분의 마음 건강을 위해 항상 도움을 주실 준비가 되어 있습니다.
-            - **상담 가능 내용:** 학교생활의 어려움, 친구 관계, 가족 문제, 진로 고민 등
-            - **상담 신청 방법:** 담임 선생님께 말씀드리거나, 상담실에 직접 방문하여 신청할 수 있습니다.
-            - **상담 시간:** (예시) 점심시간, 방과 후 시간 등 (학교 상담실 운영 시간에 따라 다름)
-            """
-        )
+# 페이지 스타일 커스터마이징
+st.markdown(
+    """
+    <style>
+        /* 전체 배경색 설정 */
+        .stApp {
+            background-color: #fffafa; /* 은은한 핑크색 배경 */
+        }
+        /* 타이틀 스타일 */
+        .main-title {
+            font-size: 3rem;
+            color: #000000; /* 검은색 */
+            font-weight: 700;
+            text-align: center;
+            margin-bottom: 20px;
+        }
+        /* 채팅 메시지 스타일 */
+        .chat-message {
+            border-radius: 15px;
+            padding: 15px;
+            margin: 10px 0;
+            display: flex;
+            align-items: center;
+            flex-wrap: wrap;
+            word-break: break-word;
+        }
+        .chat-message-user {
+            background-color: #ffebef; /* 따뜻한 파스텔 핑크 */
+            color: #8b4513;
+            justify-content: flex-end;
+        }
+        .chat-message-assistant {
+            background-color: #ffe4e6; /* 부드러운 파스텔 복숭아 핑크색 */
+            color: #6b4226;
+            justify-content: flex-start;
+        }
+        .chat-avatar {
+            width: 40px;
+            height: 40px;
+            border-radius: 50%;
+            margin-right: 10px;
+        }
+        .chat-avatar-user {
+            margin-left: 10px;
+            margin-right: 0;
+        }
+        /* 사용자 입력 창 스타일 */
+        .stTextInput input {
+            border-radius: 15px;
+            border: 2px solid #cd857f; /* 따뜻한 핑크색 */
+        }
+        /* 버튼 스타일 */
+        .stButton button {
+            background-color: #cd857f; /* 따뜻한 핑크색 */
+            color: #fff;
+            border-radius: 15px;
+            padding: 10px 20px;
+        }
+    </style>
+    """,
+    unsafe_allow_html=True
+)
 
-    with st.expander("👨‍👩‍👧‍👦 학부모 상담"):
+# 메인 타이틀
+st.markdown("<div class='main-title'>AI 튜터 🎓</div>", unsafe_allow_html=True)
+
+# 채팅 세션 초기화
+if "messages" not in st.session_state:
+    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"  # 튜터 아이콘 (안경 쓴 선생님)
+
+# 학년 수준 선택
+with st.sidebar:
+    grade_level = st.selectbox("📚 학년 수준을 선택하세요:", ["1", "2", "3", "4", "5", "6"], index=0, format_func=lambda x: f"초등학교 {x}학년")
+    # 초기화 버튼
+    if st.button("💡 초기화"):
+        del st.session_state.messages  # 메시지 세션 상태 삭제
+        st.rerun()
+
+# 채팅 메시지 표시
+for message in st.session_state.messages:
+    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
         )
 
-    st.markdown("**🌟 중요한 건 혼자 힘들어하지 않고, 주변에 도움을 요청하는 용기입니다. 여러분은 혼자가 아니에요!**")
-    st.markdown("---")
-
-
-
-# --- 메인 화면 ---
-def main():
-    # 메인 컬럼 설정
-    col_empty_left, col_main, col_empty_right = st.columns([1, 3, 1])
-
-    with st.sidebar: # 사이드바 내용
-        st.markdown("# 💖 마음 건강 다이어리") # 앱 이름 변경, 아이콘 변경
-        st.markdown("### 당신의 마음을 기록하고, AI와 상담하세요") # 부제목 변경
-        st.markdown("매일 감정을 기록하고, AI와 마음 건강 상담을 시작해보세요! 🥰")
-        with st.expander("🚀 앱 사용 가이드", expanded=False):
-            st.markdown(
-                """
-            **1단계: 감정 체크인 💖**
-            - '💖 감정 체크인' 섹션에서 오늘 하루의 감정을 선택하고, 왜 그런 감정을 느꼈는지 이유를 적어주세요.
-            - '✅ 감정 기록 완료' 버튼을 눌러 감정을 기록하면 다음 섹션으로 이동합니다.
-
-            **2단계: AI 감��� 분석 🔮**
-            - '🔮 AI 감정 분석' 섹션에서 '✨ AI 감정 분석 시작' 버튼을 눌러 AI 감정 분석을 시작합니다.
-            - AI 분석 결과를 확인하고, '다음 단계' 버튼을 눌러 다음 섹션으로 이동하세요.
-
-            **3단계: AI 상담 챗봇 💬**
-            - '💬 AI 상담 챗봇' 섹션에서 AI 챗봇과 자유롭게 대화하며 마음을 정리해보세요.
-            - '📊 감정 이력' 버튼을 눌러 다음 섹션으로 이동하세요.
-
-            **4단계: 감정 이력 확인 📊**
-            - '📊 감정 이력' 섹션에서 날짜별 감정 변화와 감정 이유를 확인할 수 있습니다.
-            - '📢 상담 안내' 버튼을 눌러 다음 섹션으로 이동하세요.
-
-            **5단계: 상담 안내 📢**
-            - '📢 상담 안내' 섹션에서 학교 상담, 학부모 상담 정보를 확인하고 필요시 도움을 요청하세요.
-            - 모든 단계를 완료했습니다! '💖 감정 체크인' 섹션으로 돌아가 다시 시작해보세요.
-
-            **💖 마음 건강 다이어리와 함께 행복한 하루를 만들어가세요! 💪** # 마무리 문구 변경
-            - 이 앱을 통해 자신의 감정을 이해하고, AI 상담을 통해 마음을 건강하게 관리해보세요! 😊
-            """
-            )
-        st.markdown("---")
-        st.markdown("<p style='font-size: 14px; color: #777;'>Version 1.0</p>", unsafe_allow_html=True) # 버전 정보 추가
-
-
-    with col_main: # 메인 컬럼 내용
-
-        sections = [
-            "❤️ 감정 체크인",
-            "🔮 AI 감정 분석",
-            "💬 AI 상담 챗봇",
-            "📊 감정 이력",
-            "📢 상담 안내",
-        ]
-
-        current_section_index = st.session_state.get("section_index", 0)
-
-        # --- ❤️ 감정 체크인 섹션 ---
-        if st.session_state.section_index == 0:
-            with st.container():
-                st.markdown('<p class="section-title">💖 감정 체크인</p>', unsafe_allow_html=True) # 섹션 제목, 아이콘 변경
-                st.markdown('<p class="section-subtitle">오늘 하루를 되돌아보고, 당신의 감정을 기록해보세요.</p>', unsafe_allow_html=True) # 섹션 부제목
-
-                emotion_options = ["😄 행복해요", "😊 좋아요", "😶 평범해요", "🙁 슬퍼요", "😭 매우 슬퍼요", "😠 화나요", "😡 매우 화나요", "😥 불안해요", "😨 무서워요"]
-                emotion = st.radio("오늘 나의 감정", emotion_options, index=2) # 라디오 버튼 제목 변경
-                reason = st.text_area("✍️ 감정 이유 (선택 사항, 간단하게 적어보세요)", height=100) # 텍스트 area 제목 변경, 선택 사항 안내
-
-                if st.button("✅ 감정 기록 완료", use_container_width=True): # 버튼 텍스트 변경
-                    if reason:
-                        record_emotion(emotion, reason)
-                    else:
-                        record_emotion(emotion, reason)
-
-        # --- 🔮 AI 감정 분석 섹션 ---
-        if st.session_state.section_index == 1:
-            with st.container():
-                st.markdown('<p class="section-title">🔮 AI 감정 분석</p>', unsafe_allow_html=True) # 섹션 제목, 아이콘 변경
-                st.markdown('<p class="section-subtitle">AI가 당신의 감정을 분석하고, 따뜻한 위로와 공감 메시지를 전달합니다.</p>', unsafe_allow_html=True) # 섹션 부제목
-
-                if st.button("✨ AI 감정 분석 시작", use_container_width=True): # 버튼 텍스트 유지
-                    start_emotion_analysis_and_counseling()
-
-                if st.session_state["emotion_analysis_result"]:
-                    analysis_result_text = st.session_state["emotion_analysis_result"]
-                    st.info(analysis_result_text) # 분석 결과 표시
-                    if st.button("💬 AI 상담 챗봇 시작", use_container_width=True): # 다음 섹션 이동 버튼
-                        st.session_state.section_index += 1
-                        st.rerun()
-                elif st.session_state["daily_emotions"].get(str(datetime.date.today())) and st.session_state["daily_emotions"].get(str(datetime.date.today())).get('reason'):
-                    st.info("👆 **'✨ AI 감정 분석 시작'** 버튼을 클릭하여 감정 분석을 시작해보세요.") # 안내 문구 유지
-                else:
-                    st.info("🤔 먼저 '💖 감정 체크인' 섹션에서 감정을 기록해야 AI 감정 분석을 시작할 수 있습니다.") # 안내 문구 개선, 아이콘 변경
-
-        # --- 💬 AI 상담 챗봇 섹션 ---
-        if st.session_state.section_index == 2:
-            with st.container():
-                st.markdown('<p class="section-title">💬 AI 상담 챗봇</p>', unsafe_allow_html=True) # 섹션 제목
-                st.markdown('<p class="section-subtitle">AI 챗봇과 편안하게 대화하며 당신의 마음을 돌보세요.</p>', unsafe_allow_html=True) # 섹션 부제목
-                st.markdown("AI 챗봇은 익명의 상담 친구처럼 당신의 이야기에 귀 기울이고 공감하며, 긍정적인 방향으로 나아갈 수 있도록 부드럽게 대화를 이끌어 줄 것입니다.") # 챗봇 설명 추가
-
-                if prompt := st.chat_input("AI 챗봇에게 이야기해보세요 (무엇이든 괜찮아요 😊)", key="chat_input"): # 입력창 placeholder 변경, 이모티콘 추가
-                    st.session_state["counseling_messages"].append({"role": "user", "content": prompt})
-                    st.session_state["counseling_history"].append({"role": "user", "content": prompt}) # 상담 history에 사용자 메시지 추가
-                    with st.chat_message("user"):
-                        st.write(prompt)
-
-                    with st.chat_message("assistant"):
-                        with st.spinner("AI 챗봇이 생각 중..."):
-                            chat_session = st.session_state["chat_session"]
-                            # 이전 상담 로그를 프롬프트에 포함하여 전달
-                            history_text = "\n".join([f"{msg['role']}: {msg['content']}" for msg in st.session_state["counseling_history"]]) # 상담 history를 텍스트로 변환
-                            enhanced_prompt = f"""**이전 상담 기록:**\n{history_text}\n\n**사용자 입력:** {prompt}\n\n**AI 상담사 역할:**
-- 30대 여성 심리 상담 전문가
-- 내담자의 감정에 깊이 공감하고, 따뜻하게 위로와 격려를 건네는 역할
-- 친절하고 부드러운 말투 사용 (예: ~군요, ~요, ~게요)
-- 단답형, 명령형, 비판적인 말투는 지양
-- 질문을 통해 내담자의 감정을 더 깊이 이해하려고 노력
-- 해결책 제시보다는 공감과 경청, 지지를 통해 내담자 스스로 답을 찾도록 돕는 역할
-
-**응답:**""" # AI 상담사 역할 및 지시사항 추가
-                            response = chat_session.send_message(enhanced_prompt) # 향상된 프롬프트 사용
-                            ai_response = response.text.strip()
-                            st.session_state["counseling_messages"].append(
-                                {"role": "assistant", "content": ai_response}
-                            )
-                            st.session_state["counseling_history"].append({"role": "assistant", "content": ai_response}) # 상담 history에 챗봇 응답 추가
-                            st.write(ai_response)
-
-
-                if "counseling_messages" not in st.session_state:
-                    st.session_state["counseling_messages"] = []
-
-                # 챗봇 시작 메시지 (분석 결과가 있을 때, 최초 1회만)
-                if not st.session_state["counseling_messages"] and st.session_state["analysis_to_chatbot"]:
-                    initial_chatbot_message = f"안녕하세요! 😊 오늘 {st.session_state['daily_emotions'][str(datetime.date.today())]['emotion']} 감정을 느끼셨군요.  AI 감정 분석 결과는 이렇습니다.\n\n{st.session_state['analysis_to_chatbot']}\n\n 이전 상담 기록을 바탕으로, {st.session_state['daily_emotions'][str(datetime.date.today())]['emotion']}님에게 더 **따뜻하고 편안한** 상담을 제공해드릴 수 있을 것 같아요.  **지금 기분이 어떠신가요?** 저에게 **편하게** 이야기해주세요. 🥰"
-                    st.session_state["counseling_messages"].append({"role": "assistant", "content": initial_chatbot_message})
-                    st.session_state["counseling_history"].append({"role": "assistant", "content": initial_chatbot_message}) # 상담 history에도 시작 메시지 추가
-
-                for msg in st.session_state["counseling_messages"]: # 메시지 순서대로 표시 (reversed 삭제)
-                    with st.chat_message(msg["role"]):
-                        st.write(msg["content"])
-                if st.button("📊 감정 이력 확인", use_container_width=True): # 다음 섹션 이동 버튼
-                    st.session_state.section_index += 1
-                    st.rerun()
-
-
-        # --- 📊 감정 이력 섹션 ---
-        if st.session_state.section_index == 3:
-            with st.container():
-                st.markdown('<p class="section-title">📊 감정 이력</p>', unsafe_allow_html=True) # 섹션 제목
-                st.markdown('<p class="section-subtitle">지난 감정 기록을 통해 마음의 변화를 살펴보세요.</p>', unsafe_allow_html=True) # 섹션 부제목
-                display_emotion_history()
-                if st.button("📢 상담 안내 확인", use_container_width=True): # 다음 섹션 이동 버튼
-                    st.session_state.section_index += 1
-                    st.rerun()
-
-
-        # --- 📢 상담 안내 섹션 ---
-        if st.session_state.section_index == 4:
-            with st.container():
-                st.markdown('<p class="section-title">📢 상담 안내</p>', unsafe_allow_html=True) # 섹션 제목
-                st.markdown('<p class="section-subtitle">혼자 힘든가요? 언제든 전문가의 도움을 요청하세요.</p>', unsafe_allow_html=True) # 섹션 부제목
-                display_counseling_info()
-                if st.button("💖 처음으로 돌아가기", use_container_width=True): # 처음 섹션 이동 버튼
-                    st.session_state.section_index = 0
-                    st.rerun()
+# 사용자 입력 받기
+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'>{prompt}<img src='{user_icon_url}' class='chat-avatar chat-avatar-user'></div>", unsafe_allow_html=True)
 
-if __name__ == "__main__":
-    main()
+    # OpenAI API 호출
+    response = openai_chat(prompt, grade_level)
+    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)