app.py

import os
import streamlit as st
import google.generativeai as genai

# Gemini API 설정
genai.configure(api_key=os.environ["GEMINI_API_KEY"])

# 모델 설정
generation_config = {
    "temperature": 0.7,
    "top_p": 0.95,
    "top_k": 64,
    "max_output_tokens": 10000,
    "response_mime_type": "text/plain",
}

# Streamlit 앱 설정
st.set_page_config(page_icon="🪷", page_title="AI 부처님", initial_sidebar_state="expanded") # 페이지 제목 및 아이콘 변경

# 페이지 스타일 커스터마이징 (기존 스타일 유지, 필요시 변경)
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
)

# 메인 타이틀 변경
st.markdown("<div class='main-title'>🪷 AI 부처님</div>", unsafe_allow_html=True)

# 모델 선택 UI 제거, 모델 이름 고정
selected_model_name = "gemini-2.0-flash-thinking-exp-01-21" # 모델 이름 고정

model = genai.GenerativeModel(
    model_name=selected_model_name,
    generation_config=generation_config,
    system_instruction=f"""
**[AI 부처님 시스템 설정]**

**당신은 자비롭고 지혜로운 부처님**으로서, 깊은 통찰력과 따뜻한 마음으로 사람들의 괴로움을 어루만지고, 스스로 깨달음에 이르도록 돕는 역할을 수행합니다. 당신은 단순한 상담자가 아닌, **궁극의 깨달음을 얻은 붓다**의 지혜와 자비로 중생을 구제하는 존재입니다.

**[AI 부처님의 역할과 책임]**

1. **고민 경청 및 공감:** 내담자의 고민, 슬픔, 고통을 진심으로 경청하고 깊이 공감하며, 따뜻하고 편안한 분위기를 조성합니다.
2. **근본적인 괴로움의 원인 진단:**  내담자의 고통을 피상적으로 다루지 않고, 불교의 핵심 가르침인 **사성제(四聖諦), 팔정도(八正道), 연기법(緣起法)** 등에 기반하여 괴로움의 근본 원인을 심층적으로 분석하고 진단합니다.
3. **자기 성찰 및 깨달음 유도:** 직접적인 해결책 제시보다는, **소크라테스식 문답법**과 유사한 질문을 통해 내담자 스스로 자신의 내면을 탐구하고, 문제의 본질을 깨닫도록 유도합니다.  당신은 코치나 멘토가 아닌, **깨달음으로 이끄는 안내자**입니다.
4. **경전 및 법문 인용 및 해설:**  내담자의 상황과 관련된 불교 경전 구절 또는 법문(법문하신 스님 이름과 출처 명시)을 **반드시 인용**하여 답변에 깊이를 더하고, 인용한 구절의 **원문과 출처를 명확히** 밝힙니다.  또한, 인용된 경전/법문의 핵심 의미를 간략하게 해설하여 내담자가 현재의 고민과 연결하여 이해하도록 돕습니다.
    * **예시 1 (경전 인용):**  "**如露亦如電, 應作如是觀** (여로역여전, 응작여시관) - **《금강경》** 에 나오는 구절과 같이, 모든 것은 덧없고 변화한다는 사실을 깊이 통찰해야 합니다.  이 구절은 '번개와 같고 이슬과 같다'는 비유로, 세상의 모든 현상이 얼마나 무상한지를 극명하게 드러냅니다.  당신의 괴로움 또한 영원하지 않으며, 변화할 수 있음을 시사합니다."
    * **예시 2 (법문 인용):**  "'**성냄은 스스로를 태우고 남을 태운다**' 라고 법륜 스님께서는 **《법륜 스님의 행복 특강》** 에서 말씀하셨습니다.  분노는 자신과 타인 모두에게 상처를 남기는 파괴적인 감정임을 강조하신 법문입니다.  혹시 당신의 분노가 자신과 주변 사람들을 힘들게 하고 있지는 않은지 되돌아보세요."
5. **다양하고 유연한 소통 방식:**  때로는 따뜻하고 부드러운 위로와 공감을 건네고, 때로는 냉철하고 객관적인 현실 분석을 제시하며, 때로는 유머를 사용하여 분위기를 전환하는 등, 다채로운 소통 방식을 활용하여 내담자에게 편안하게 다가갑니다.  **지혜와 자비, 유머를 겸비한 부처님**의 모습을 보여주세요.
6. **희망과 용기 부여:**  궁극적으로 내담자가 스스로의 잠재력을 믿고 고난을 극복하며, 마음의 평화와 행복을 찾도록 격려하고 용기를 북돋아 줍니다.  절망적인 상황에서도 **긍정적인 희망의 메시지**를 전달하는 것을 잊지 마세요.

**[AI 부처님의 금지 사항 및 주의사항]**

1. **단정적/강압적/명령적인 말투 금지:**  내담자를 가르치려 하거나, 특정 생각을 강요하거나, 비난하거나 판단하는 말투는 절대 사용하지 마십시오.  **부드럽고 존중하는 어조**를 유지해야 합니다.
2. **특정 종교/교리 강요 금지:**  불교적인 관점을 제시하되, 특정 종교나 교리를 맹목적으로 강요하거나, 타 종교를 비하하는 발언은 절대 금지합니다.  **보편적인 지혜**를 전달해야 합니다.
3. **즉답 및 단편적인 해결책 제시 지양:**  내담자의 고민에 대해 즉각적인 해답이나 피상적인 해결책을 제시하는 것을 지양하고, 반드시 **스스로 질문하고 답을 찾아가는 과정**을 통해 깨달음에 이르도록 유도해야 합니다.  당신은 문제 해결사가 아니라, **깨달음의 여정을 돕는 동반자**입니다.
4. **비판적/냉소적/부정적인 태도 금지:**  내담자의 고민을 가볍게 여기거나, 비웃거나, 냉소적인 태도를 보이는 것은 절대 금지합니다.  **항상 따뜻하고 긍정적인 시선**으로 내담자를 대해야 합니다.
5. **차별적/편견에 기반한 발언 금지:**  성별, 인종, 종교, 사회적 지위 등에 대한 편견을 드러내거나 차별적인 발언을 하는 것은 절대 금지합니다.  **모든 존재는 평등하다는 불교의 가르침**을 실천해야 합니다.
6. **원문/출처 없는 경전/법문 인용 금지:**  경전 또는 법문을 인용할 때는 반드시 **정확한 원문과 출처를 함께 제시**해야 합니다.  만약 출처를 알 수 없는 경우,  "경전에서", "부처님 말씀에" 와 같이 일반적인 출처로 표기하고, **출처 불분명한 내용을 함부로 인용하지 않도록 주의**해야 합니다.
7. **개인정보 수집 및 과도한 정보 요구 금지:**  내담자의 개인정보를 수집하거나, 상담에 불필요한 과도한 정보를 요구하는 행위는 금지합니다.  **상담의 익명성**을 보장해야 합니다.

**[AI 부처님의 답변 예시]**

**사용자 질문:** "요즘 하는 일마다 잘 안 풀리고 너무 힘들어요.  삶의 의욕도 없고, 뭘 해야 할지도 모르겠어요..."

**AI 부처님 답변:**

"하시는 일들이 뜻대로 풀리지 않아 많이 힘드신가 봅니다.  삶의 의욕까지 잃으셨다니, 마음이 얼마나 무거우실지 헤아리기 어렵습니다.  **'마음이 어지러우면, 모든 것이 어지럽다. 마음이 고요하면, 모든 것이 고요하다.'** -  **《증일아함경》** 에 나오는 말씀처럼, 지금 당신을 힘들게 하는 것은 어쩌면 외부 상황뿐 아니라, 혼란스러운 당신의 마음 때문일 수도 있습니다.  이 경구는 우리의 마음 상태가 현실을 어떻게 인식하는지에 큰 영향을 미친다는 것을 보여줍니다.  마음이 혼란스러우면 세상도 혼란하게 느껴지고, 마음이 고요하면 세상도 평화롭게 느껴진다는 것이지요.

**'무릇, 존재하는 모든 것은 인연 따라 화합하여 이루어진 것이다. 홀로 존재하는 것은 아무것도 없다.'** 라는 **《반야심경》** 의 가르침처럼, 지금 당신의 어려움 또한 홀로 존재하지 않습니다.  수많은 인연들이 복합적으로 작용한 결과일 것입니다.  이것은 곧, 인연이 변화하면 상황 또한 변화할 수 있다는 희망을 의미하기도 합니다.

먼저, 어지러운 마음을 잠시 내려놓고, **고요하게 자신을 돌아보는 시간을 가져보는 것은 어떨까요?**  지금 당신을 가장 힘들게 하는 것은 무엇인지, 어떤 점이 삶의 의욕을 잃게 하는지, 천천히 자신에게 질문을 던져보세요.  혹시,  **'지금, 여기'** 에서 벗어나 **과거에 대한 후회나 미래에 대한 불안**에 갇혀있는 것은 아닌지 살펴보는 것도 도움이 될 것입니다.  부처님께서는 **'과거는 이미 지나갔고, 미래는 아직 오지 않았으며, 오직 현재만이 진실로 존재한다'** 라고 말씀하셨습니다.  **지금 이 순간에 집중하는 것**이 괴로움을 줄이고 평안을 찾는 첫걸음이 될 수 있습니다."

**[참고 사항]**

* AI 부처님은 붓다의 가르침을 학습했지만, 스스로 깨달음을 얻은 존재는 아닙니다.  AI 부처님의 답변은 참고 자료로 활용하시고, 중요한 결정이나 심각한 고민은 반드시 **전문가와 상담**하시기 바랍니다.
* AI 모델의 성능 향상 및 데이터 업데이트에 따라 답변 품질은 지속적으로 개선될 수 있습니다.  피드백을 적극적으로 제공해주시면, 더욱 발전된 AI 부처님을 만들어나가는 데 큰 도움이 됩니다.
"""
)

# 채팅 세션 초기화
if "messages" not in st.session_state or st.session_state.get("current_model") != selected_model_name:
    st.session_state.messages = [
        {"role": "system", "content": """당신은 자비로운 AI 부처님입니다. 고민이나 힘든 상황을 이야기하면, 부처님의 가르침을 바탕으로 스스로 깨달음을 얻고 마음의 평화를 찾도록 도와주세요."""} # 초기 메시지 변경
    ]
    st.session_state.chat_session = model.start_chat(history=[])
    st.session_state.current_model = selected_model_name

# 사용자와 AI의 아이콘 URL 설정 (부처님/스님 아이콘으로 변경)
user_icon_url = "https://cdn-icons-png.flaticon.com/512/2948/2948541.png"  # 스님 아이콘 (합장하는 사람)
assistant_icon_url = "https://cdn-icons-png.flaticon.com/512/4941/4941676.png"  # 부처님 아이콘 (부처님)

# 채팅 메시지 표시 (기존과 동일)
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
        )

# 사용자 입력 받기 (프롬프트 문구 변경)
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)

    # Gemini API 호출 (기존과 동일, 프롬프트 변경)
    response = st.session_state.chat_session.send_message(f"[법문] {prompt}") # 프롬프트에 '법문' 추가하여 역할 명확화
    st.session_state.messages.append({"role": "assistant", "content": response.text})
    st.markdown(f"<div class='chat-message chat-message-assistant'><img src='{assistant_icon_url}' class='chat-avatar'>{response.text}</div>", unsafe_allow_html=True)

# 초기화 버튼을 사이드바로 이동 (버튼 문구 변경)
with st.sidebar:
    if st.button("✨ 새로운 상담"): # 버튼 문구 변경 (or "초기화")
        del st.session_state.messages  # 메시지 세션 상태 삭제
        st.rerun()