Update app.py

app.py

@@ -2,7 +2,7 @@ import time
 import streamlit as st
 import google.generativeai as genai
 from streamlit_extras.colored_header import colored_header
-import markdown
+import markdown # markdown 라이브러리 임포트 확인
 
 # --- Must be the first Streamlit command ---
 st.set_page_config(page_title="MBTI 관계 시뮬레이터", page_icon="🤝", layout="wide")
@@ -11,34 +11,50 @@ st.set_page_config(page_title="MBTI 관계 시뮬레이터", page_icon="🤝", l
 
 # Google Gemini API Key 설정 (Streamlit secrets 사용)
 try:
-    genai.configure(api_key=st.secrets["GEMINI_API_KEY"])
-except KeyError:
-    st.error("GEMINI_API_KEY secrets 설정을 확인해주세요.")
-    st.stop()
+    # secrets에서 키 로드 시도
+    api_key = st.secrets.get("GEMINI_API_KEY")
+    if not api_key:
+        st.error("Streamlit secrets에 'GEMINI_API_KEY'를 설정해주세요.")
+        st.info("Secrets 설정 방법: [Streamlit Docs](https://docs.streamlit.io/library/advanced-features/secrets-management)")
+        st.stop()
+    genai.configure(api_key=api_key)
 except Exception as e:
-    st.error(f"API 키 설정 중 오류 발생: {e}")
+    st.error(f"API 키 설정 중 예상치 못한 오류 발생: {e}")
     st.stop()
 
+
 # 모델 설정
 generation_config = {
     "temperature": 0.75,
     "top_p": 0.8,
     "top_k": 40,
-    "max_output_tokens": 8192,
+    "max_output_tokens": 15000,
 }
 
 # 특정 모델 이름으로 설정 (사용자 요청 반영)
-target_model_name = "gemini-2.0-flash-thinking-exp-01-21"
+# 참고: 'gemini-2.0-flash-thinking-exp-01-21'는 실험적 모델일 수 있습니다.
+# 안정적인 최신 모델을 원하시면 'gemini-1.5-flash-latest' 또는 'gemini-1.5-pro-latest' 사용을 고려하세요.
+target_model_name = "gemini-2.0-flash-thinking-exp-01-21" # 안정적인 모델로 변경 권장 (필요시 원래 모델명 사용)
 
 try:
     model = genai.GenerativeModel(
         model_name=target_model_name,
         generation_config=generation_config,
+        # safety_settings 설정 추가 (콘텐츠 필터링 관련)
+        safety_settings=[
+            {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
+            {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
+            {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
+            {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
+        ]
     )
-    # Model display removed - st.info(f"Using model: {target_model_name}")
+    # 모델 로딩 성공 시 간단한 정보 표시 (선택 사항)
+    # st.sidebar.caption(f"Using model: {target_model_name}")
 except Exception as e:
     st.error(f"Gemini 모델 '{target_model_name}' 로딩 중 오류 발생: {e}")
-    st.error("모델 이름을 다시 확인하거나, 사용 가능한 모델 목록을 확인하여 다른 모델을 시도해보세요.")
+    st.error("모델 이름을 다시 확인하거나, 사용 가능한 다른 모델(예: 'gemini-1.5-flash-latest')을 시도해보세요.")
+    # 사용 가능한 모델 목록 확인 링크 (선택 사항)
+    st.info("사용 가능한 모델 목록은 Google AI Studio 또는 Gemini API 문서를 참조하세요.")
     st.stop()
 
 
@@ -110,11 +126,23 @@ def generate_relationship_scenario(people, relationship, situation):
 
     # 참여자 정보 문자열 생성 함수
     def create_type_info(person):
-        type_description = mbti_descriptions.get(person['type'], '알 수 없는 유형') # 상세 설명 사용
-        return f"{person['name']} ({person['gender']}, {person['type']}): {type_description}"
+        # 입력값이 없을 경우 기본값 처리
+        name = person.get('name', '이름없음')
+        gender = person.get('gender', '미지정')
+        mbti_type = person.get('type', '미지정')
+        type_description = mbti_descriptions.get(mbti_type, '알 수 없는 유형') # 상세 설명 사용
+        return f"{name} ({gender}, {mbti_type}): {type_description}"
+
+    # 참여자 정보 목록 생성 (people 리스트가 비어있지 않을 때만 생성)
+    if people:
+        people_info = "\n".join([f"- {create_type_info(person)}" for person in people])
+    else:
+        # people 리스트가 비어있는 예외적인 경우 처리
+        st.error("참여자 정보가 올바르게 전달되지 않았습니다. 사이드바 입력을 확인해주세요.")
+        return "" # 오류 발생 시 빈 문자열 반환
 
-    # 참여자 정보 목록 생성
-    people_info = "\n".join([f"- {create_type_info(person)}" for person in people])
+    # 관계 유형 영문명 가져오기
+    relationship_en = relationship_types_two.get(relationship, relationship_types_multiple.get(relationship, ''))
 
     # 시스템 프롬프트 (개선됨: 속마음/의도, 상호작용 역학, 오해 지점, 명확한 구조 강조, 테이블 형식 지양)
     system_prompt = f"""
@@ -124,7 +152,7 @@ def generate_relationship_scenario(people, relationship, situation):
 
     *   **참여자:**
     {people_info}
-    *   **관계 유형:** {relationship} ({relationship_types_two.get(relationship, relationship_types_multiple.get(relationship, ''))})
+    *   **관계 유형:** {relationship} ({relationship_en})
     *   **주어진 상황:** {situation}
 
     **[요청 사항]**
@@ -147,7 +175,7 @@ def generate_relationship_scenario(people, relationship, situation):
 
     **4. 관계 개선을 위한 실질적 조언:**
         *   **각 참여자에게 맞춤화된 조언**을 제공해주세요. 이 조언은 시나리오에서 드러난 **구체적인 상호작용, 속마음, 오해 지점**을 직접적으로 다루어야 합니다.
-        *   서로를 더 잘 이해하고 **건강한 관계**를 구축하기 위해 각자가 **시도해볼 수 있는 구체적인 말과 행동**을 제안해주세요. (예: "A(INTJ)님, B(ENFP)님이 아이디어를 낼 때 즉시 분석하기보다, 먼저 '흥미로운 생각인데!'라고 반응하며 B님의 열정을 인정해주세요. 그 후에 함께 현실적인 부분을 논의하는 것이 좋습니다.")
+        *   서로를 더 잘 이해하고 **건강한 관계**를 구축하기 위해 각자가 **시도해볼 수 있는 구체적인 말과 행동**을 제안해주세요. (예: "{people[0]['name']}({people[0]['type']})님, {people[1]['name']}({people[1]['type']})님이 아이디어를 낼 때 즉시 분석하기보다, 먼저 '흥미로운 생각인데!'라고 반응하며 {people[1]['name']}님의 열정을 인정해주세요. 그 후에 함께 현실적인 부분을 논의하는 것이 좋습니다.")
         *   조언에는 반드시 **모든 참여자의 이름, 성별, MBTI 유형**이 명시되어야 합니다.
 
     **[출력 형식]**
@@ -164,18 +192,25 @@ def generate_relationship_scenario(people, relationship, situation):
         response = model.generate_content([system_prompt], stream=True)
         response_container = st.empty() # 응답 표시 영역 미리 확보
         for chunk in response:
-            # chunk.text가 None이 아닌 경우에만 처리
+            # chunk.text가 None이 아닌지 확인
             if chunk.text:
-                full_text += chunk.text
-                # Markdown으로 변환하여 실시간 업데이트
-                html_text = markdown.markdown(full_text)
-                response_container.markdown(html_text, unsafe_allow_html=True)
-            time.sleep(0.02) # 약간의 딜레이로 부드러운 스트리밍 효과
+                full_text += chunk.text # Append chunk text to full_text
+                # Markdown으로 실시간 업데이트 (unsafe_allow_html=False가 더 안전)
+                response_container.markdown(full_text, unsafe_allow_html=False) # HTML 대신 마크다운 직접 렌더링
+            time.sleep(0.01) # 딜레이 약간 줄임
         return full_text # 최종 전체 텍스트 반환
 
     except Exception as e:
         st.error(f"시나리오 생성 중 오류 발생: {e}")
+        # 오류 유형과 메시지를 함께 출력하여 디버깅에 도움
         st.error(f"오류 상세 정보: {type(e).__name__} - {e}")
+        # API 호출 관련 오류일 수 있으므로 추가 정보 제공
+        if "API key" in str(e):
+            st.error("API 키가 유효하지 않거나 할당량이 초과되었을 수 있습니다. Streamlit secrets 설정을 확인하세요.")
+        elif "model" in str(e).lower():
+             st.error(f"모델 '{target_model_name}'을 찾을 수 없거나 접근 권한이 없을 수 있습니다. 모델 이름을 확인하거나 다른 모델을 시도해보세요.")
+        elif "safety" in str(e).lower() or "filtered" in str(e).lower():
+             st.error("콘텐츠 안전 설정에 의해 응답이 필터링되었습니다. 입력 내용을 수정하거나 안전 설정을 조정해보세요.")
         return ""
 
 
@@ -183,7 +218,7 @@ def generate_relationship_scenario(people, relationship, situation):
 
 # 헤더 색상 적용
 colored_header(
-    label="🤝 MBTI 관계 시뮬레이터 v2.0",
+    label="🤝 MBTI 관계 시뮬레이터 v2.1", # 버전 업데이트
     description="참여자들의 MBTI, 관계, 상황을 입력하여 심층적인 관계 시나리오와 조언을 받아보세요.",
     color_name="blue-70"
 )
@@ -200,14 +235,23 @@ with st.sidebar:
     num_people = st.number_input("참여자 수", min_value=2, max_value=5, value=2, key="num_people",
                                  help="최소 2명, 최대 5명까지 설정할 수 있습니다.")
 
-    people = []
+    people = [] # 리스트 초기화를 루프 바깥에서 수행
     # 참여자 수에 맞춰 동적으로 입력 필드 생성
     for i in range(num_people):
-        with st.expander(f"👤 참여자 {i+1}", expanded=False): # 기본적으로 expander 닫힘
+        # 첫 번째 참여자만 기본으로 확장되도록 수정
+        with st.expander(f"👤 참여자 {i+1}", expanded=(i == 0)):
             person = {}
-            person['name'] = st.text_input(f"이름/닉네임", key=f"name_{i}", placeholder=f"예: 철수")
+            # 이름 예시 플레이스홀더 추가
+            default_names = ['철수', '영희', '민준', '서연', '지우']
+            person['name'] = st.text_input(f"이름/닉네임", key=f"name_{i}", placeholder=f"예: {default_names[i % len(default_names)]}")
             person['gender'] = st.radio("성별", ["남성", "여성", "기타"], key=f"gender_{i}", horizontal=True)
-            person['type'] = st.selectbox(f"MBTI 유형", list(mbti_types.keys()), key=f"type_{i}", index=i % len(mbti_types))
+            # 기본 MBTI 선택 인덱스 계산 (오류 방지)
+            default_mbti_index = i % len(mbti_types)
+            person['type'] = st.selectbox(f"MBTI 유형", list(mbti_types.keys()), format_func=lambda x: f"{x} ({mbti_types[x]})", key=f"type_{i}", index=default_mbti_index)
+
+            # --- ★★★ 수정된 부분 ★★★ ---
+            people.append(person) # 입력받은 참여자 정보를 리스트에 추가
+            # ---------------------------
 
     st.divider()
 
@@ -222,7 +266,7 @@ with st.sidebar:
         relationship_options = relationship_types_multiple
     relationship = st.selectbox("관계 유형", list(relationship_options.keys()), key="relationship")
 
-    situation = st.text_area("구체적인 상황", key="situation", height=100, # 높이 줄임
+    situation = st.text_area("구체적인 상황", key="situation", height=100, # 높이 유지
                              placeholder="예: 중요한 프로젝트 마감일을 앞두고 의견 충돌 발생")
 
     st.divider()
@@ -234,34 +278,55 @@ with st.sidebar:
 
 # Main area - Scenario output
 st.header("💡 생성된 시나리오")
+# Use st.container() for better layout control if needed, otherwise direct markdown is fine
 scenario_output_area = st.container()
+# scenario_output_area = st.empty() # Use empty() if you want to replace content completely on each run
 
 # Button click logic
 if generate_button:
-    # Input validation
+    # Input validation 강화
     valid_input = True
-    if not all(p.get('name') for p in people):
+    # 이름 입력 확인 (공백만 입력된 경우도 방지)
+    if not all(p.get('name', '').strip() for p in people):
         st.error("모든 참여자의 이름을 입력해주세요.")
         valid_input = False
-    if not situation:
+    # 상황 입력 확인 (공백만 입력된 경우도 방지)
+    if not situation.strip():
         st.error("구체적인 상황을 입력해주세요.")
         valid_input = False
+    # 관계 유형 선택 확인 (selectbox는 기본값이 있으므로 일반적으로 문제는 없으나 명시적으로 확인)
     if not relationship:
         st.error("관계 유형을 선택해주세요.")
         valid_input = False
+    # people 리스트 자체 확인 (만약을 대비)
+    if not people:
+        st.error("참여자 정보가 없습니다. 페이지를 새로고침하거나 참여자 수를 다시 설정해보세요.")
+        valid_input = False
+
 
     if valid_input:
-        with st.spinner("🧠 MBTI 전문가가 시나리오 분석 및 생성 중..."):
-            full_text_result = generate_relationship_scenario(people, relationship, situation)
-
-        if full_text_result:
-            scenario_output_area.markdown(markdown.markdown(full_text_result), unsafe_allow_html=True)
-            st.success("🎉 시나리오 생성 완료!")
-        else:
-            pass # Error message already shown in generate_relationship_scenario
+        # 시나리오 생성 영역 초기화 (이전 결과 제거)
+        scenario_output_area.empty()
+        with scenario_output_area: # 컨테이너 내에서 스피너와 결과 표시
+            with st.spinner("🧠 MBTI 전문가가 시나리오 분석 및 생성 중... 잠시만 기다려주세요."):
+                # people 리스트가 제대로 채워졌는지 확인 (디버깅용)
+                # st.write("전달되는 참여자 정보:", people)
+                full_text_result = generate_relationship_scenario(people, relationship, situation)
+
+            if full_text_result:
+                # 최종 결과를 Markdown으로 렌더링 (unsafe_allow_html=True 사용 시 주의)
+                # html_output = markdown.markdown(full_text_result)
+                # scenario_output_area.markdown(html_output, unsafe_allow_html=True)
+                scenario_output_area.markdown(full_text_result, unsafe_allow_html=False) # 마크다운 직접 렌더링 권장
+                st.success("🎉 시나리오 생성 완료!")
+            else:
+                # generate_relationship_scenario 함수 내에서 이미 오류 메시지가 표시됨
+                st.warning("시나리오 생성에 실패했습니다. 오류 메시지를 확인해주세요.")
     else:
         st.warning("입력값을 다시 확인해주세요.")
 
-# Initial message
+# Initial message when the app loads or before generation
+# Check if the button has been clicked at least once using session state if needed
+# For simplicity, just show if the button wasn't clicked in this run
 if not generate_button:
      scenario_output_area.info("왼쪽 사이드바에서 참여자 정보, 관계, 상황을 설정 후 '시나리오 생성하기' 버튼을 클릭하세요.")