Update app.py

app.py

@@ -21,31 +21,63 @@ model = genai.GenerativeModel(
     generation_config=generation_config,
 )
 
-def generate_date_scenario(my_name, my_gender, my_mbti, partner_name, partner_gender, partner_mbti, situation):
+# MBTI 유형 정보
+mbti_types = {
+    "ISTJ": "청렴결백한 논리주의자",
+    "ISFJ": "용감한 수호자",
+    "INFJ": "선의의 옹호자",
+    "INTJ": "용의주도한 전략가",
+    "ISTP": "만능 재주꾼",
+    "ISFP": "호기로운 예술가",
+    "INFP": "열정적인 중재자",
+    "INTP": "논리적인 사색가",
+    "ESTP": "모험을 즐기는 사업가",
+    "ESFP": "자유로운 영혼의 연예인",
+    "ENFP": "재기발랄한 활동가",
+    "ENTP": "박학다식한 발명가",
+    "ESTJ": "엄격한 관리자",
+    "ESFJ": "사교적인 외교관",
+    "ENFJ": "정의로운 사회운동가",
+    "ENTJ": "대담한 통솔자"
+}
+
+# 관계 유형
+relationship_types = {
+    "연인": "Romantic Couple",
+    "부부": "Married Couple",
+    "친구": "Friends",
+    "가족": "Family",
+    "직장 동료": "Coworkers",
+    "기타": "Others"
+}
+
+
+def generate_relationship_scenario(my_name, my_gender, my_mbti, partner_name, partner_gender, partner_mbti, relationship, situation):
+    my_type_info = f"{my_name} ({my_gender}): {my_mbti} ({mbti_types.get(my_mbti, '알 수 없는 유형')})"
+    partner_type_info = f"{partner_name} ({partner_gender}): {partner_mbti} ({mbti_types.get(partner_mbti, '알 수 없는 유형')})"
 
     system_prompt = f"""
-    당신은 MBTI 전문가이자 관계 코치입니다. 아래 요청에 따라 데이트 시나리오를 작성해주세요.
-    ---
+    당신은 MBTI 전문가이자 관계 코치입니다. 아래 정보를 바탕으로 관계 시나리오를 작성해주세요.
+
     [대상자 정보]
-    - {my_name} ({my_gender}): {my_mbti}
-    - {partner_name} ({partner_gender}): {partner_mbti}
+    - {my_type_info}
+    - {partner_type_info}
+    [관계]
+    - {relationship}
     [상황]
     - {situation}
-    ---
+
     [요청 사항]
-    1. **{my_name} ({my_gender})의 MBTI 유형({my_mbti})에 대한 상세 설명:** 
-       - 핵심 가치, 성격적 특징, 강점과 약점, 주의할 점 등을 포함하여 자세히 설명해주세요.
-    2. **{partner_name} ({partner_gender})의 MBTI 유형({partner_mbti})에 대한 상세 설명:**
+    1. **각 대상자의 MBTI 유형에 대한 상세 설명:**
        - 핵심 가치, 성격적 특징, 강점과 약점, 주의할 점 등을 포함하여 자세히 설명해주세요.
-    3. **MBTI 유형을 바탕으로 한 구체적인 데이트 시나리오 작성:**
-       - 만남, 대화 시작, 식사, 영화 관람, 갈등 발생 및 해결 등 데이트의 각 단계를 구체적인 대화 장면으로 묘사해주세요.
-       - 각 단계에서 {my_name}과 {partner_name}의 감정 변화와 반응을 상세히 묘사하고, 각 인물의 말과 행동이 MBTI 특성과 어떻게 연결되는지 부가 설명해주세요.
-       - 시나리오에 {my_name}과 {partner_name}의 이름, 성별, MBTI 유형이 모두 반영되어야 합니다.
-    4. **상대방을 서로 이해하기 위한 실질적인 조언:**
+    2. **관계 유형과 상황을 고려한 구체적인 시나리오 작성:**
+       - 주어진 관계 유형과 상황에 맞는 현실적인 상호작용을 묘사해주세요.
+       - 각 인물의 MBTI 유형이 어떻게 반영되는지 명확하게 보여주세요.
+       - 시나리오에 모든 입력 정보(이름, 성별, 유형, 관계, 상황)가 포함되어야 합니다.
+    3. **서로 이해하고 관계를 개선하기 위한 조언:**
        - 각 대상자가 상대방의 MBTI 유형을 이해하고, 더 나은 관계를 형성하기 위한 구체적인 상황과 행동을 예시로 제시해주세요.
        - 조언에 {my_name}과 {partner_name}의 이름, 성별, MBTI 유형이 모두 반영되어야 합니다.
        - 그리고 조언의 내용에 상대방에게 하면 좋은 말과 행동의 구체적 예시를 포함해야 합니다.
-    ---
     """
 
     full_text = ""
@@ -62,36 +94,35 @@ def generate_date_scenario(my_name, my_gender, my_mbti, partner_name, partner_ge
 
     return full_text
 
-# Streamlit 인터페이스
-st.set_page_config(page_title="MBTI 데이트 시뮬레이터", page_icon="❤️")
 
-colored_header(
-    label="💖 MBTI 데이트 시뮬레이터 💖",
-    description="나와 상대방의 MBTI 유형을 기반으로 데이트 시나리오를 미리 체험해보세요!",
-    color_name="red-70"
-)
+# Streamlit UI
+st.set_page_config(page_title="MBTI 관계 시뮬레이터", page_icon="🤝")
+
+st.title("🤝 MBTI 관계 시뮬레이터")
+st.write("두 사람의 MBTI 유형과 관계, 상황을 입력하여 관계 시나리오를 생성해보세요.")
+
+tab1, tab2 = st.tabs(["👤 내 정보", "👥 상대방 정보"])
 
-# 사이드바 입력 폼
-with st.sidebar:
-    st.subheader("내 정보")
-    my_name = st.text_input("내 이름(닉네임)", "나")
-    my_gender = st.selectbox("내 성별", ["남성", "여성"])
-    my_mbti = st.selectbox("내 MBTI 유형", ["ISTJ", "ISFJ", "INFJ", "INTJ", "ISTP", "ISFP", "INFP", "INTP", "ESTP", "ESFP", "ENFP", "ENTP", "ESTJ", "ESFJ", "ENFJ", "ENTJ"])
+with tab1:
+    my_name = st.text_input("내 이름/닉네임", "나")
+    my_gender = st.radio("내 성별", ["남성", "여성"])
+    my_mbti = st.selectbox("내 MBTI 유형", list(mbti_types.keys()))
 
-    st.subheader("상대방 정보")
-    partner_name = st.text_input("상대방 이름(닉네임)", "상대방")
-    partner_gender = st.selectbox("상대방 성별", ["남성", "여성"])
-    partner_mbti = st.selectbox("상대방 MBTI 유형", ["ISTJ", "ISFJ", "INFJ", "INTJ", "ISTP", "ISFP", "INFP", "INTP", "ESTP", "ESFP", "ENFP", "ENTP", "ESTJ", "ESFJ", "ENFJ", "ENTJ"])
+with tab2:
+    partner_name = st.text_input("상대방 이름/닉네임", "상대방")
+    partner_gender = st.radio("상대방 성별", ["남성", "여성"])
+    partner_mbti = st.selectbox("상대방 MBTI 유형", list(mbti_types.keys()))
 
-    st.subheader("데이트 상황")
-    situation = st.text_area("예: 첫 만남, 영화 관람 후 저녁 식사", height=150)
+relationship = st.selectbox("관계 유형", list(relationship_types.keys()))
+situation = st.text_area("상황 (예: 갈등 상황, 특별한 이벤트 등)", height=150)
 
-    generate_scenario_button = st.button("💕 시나리오 생성! 💕")
+generate_button = st.button("시나리오 생성 ➡️")
 
-# 메인 영역 출력
 scenario_output_area = st.empty()
 
-if generate_scenario_button:
-    full_text = generate_date_scenario(my_name, my_gender, my_mbti, partner_name, partner_gender, partner_mbti, situation)
+if generate_button:
+    full_text = generate_relationship_scenario(
+        my_name, my_gender, my_mbti, partner_name, partner_gender, partner_mbti, relationship, situation
+    )
     if full_text:
         scenario_output_area.markdown(markdown.markdown(full_text), unsafe_allow_html=True)