Update app.py

app.py

@@ -36,17 +36,27 @@ enneagram_types = {
 
 # 날개 유형 정보
 wing_types = {
-    "w1": "개혁가 날개",
-    "w2": "조력가 날개",
-    "w3": "성취자 날개",
-    "w4": "예술가 날개",
-    "w5": "사색가 날개",
-    "w6": "충실가 날개",
-    "w7": "낙천가 날개",
-    "w8": "도전가 날개",
-    "w9": "평화주의자 날개"
+    "1w9": "1w9",
+    "1w2": "1w2",
+    "2w1": "2w1",
+    "2w3": "2w3",
+    "3w2": "3w2",
+    "3w4": "3w4",
+    "4w3": "4w3",
+    "4w5": "4w5",
+    "5w4": "5w4",
+    "5w6": "5w6",
+    "6w5": "6w5",
+    "6w7": "6w7",
+    "7w6": "7w6",
+    "7w8": "7w8",
+    "8w7": "8w7",
+    "8w9": "8w9",
+    "9w8": "9w8",
+    "9w1": "9w1",
 }
 
+
 # 본능 유형 정보
 instinct_variants = {
     "sp": "자기보존",
@@ -54,7 +64,7 @@ instinct_variants = {
     "sx": "성적"
 }
 
-# 2인 관계 유형
+# 관계 유형 (2인 관계와 다인 관계 분리)
 relationship_types_two = {
     "연인": "Romantic Couple",
     "부부": "Married Couple",
@@ -64,120 +74,93 @@ relationship_types_two = {
     "기타": "Others"
 }
 
-# 다자 관계 유형
 relationship_types_multiple = {
+    "친구": "Friends",
+    "가족": "Family",
     "직장 동료": "Coworkers",
-    "친구들": "Friends Group",
     "프로젝트 팀": "Project Team",
     "스터디 그룹": "Study Group",
-    "동호회": "Club",
-    "가족 모임": "Family Gathering",
-    "네트워킹 모임": "Networking Event"
+    "기타": "Others"
 }
 
-def generate_relationship_scenario(my_name, my_gender, my_type, my_wing, my_instinct, partner_infos, relationship, situation):
+
+def generate_relationship_scenario(people, relationship, situation):
 
     # 에니어그램 유형 정보 문자열 생성 (날개/본능 포함)
-    def create_type_info(name, gender, type, wing, instinct):
-        type_info = f"{name} ({gender}): {type} ({enneagram_types.get(type, '알 수 없는 유형')})"
-        if wing:
-            type_info += f" 날개: {wing_types.get(wing)}"
-        if instinct:
-            type_info += f", 본능: {instinct_variants.get(instinct)}"
+    def create_type_info(person):
+        type_info = f"{person['name']} ({person['gender']}): {person['type']} ({enneagram_types.get(person['type'], '알 수 없는 유형')})"
+        if person['wing']:
+            type_info += f" 날개: {person['wing']}"
+        if person['instinct']:
+            type_info += f", 본능: {person['instinct_variants'].get(person['instinct'])}"
         return type_info
 
-    my_type_info = create_type_info(my_name, my_gender, my_type, my_wing, my_instinct)
+    people_info = "\n".join([f"- {create_type_info(person)}" for person in people])
+
+
+    system_prompt = f"""
+    당신은 에니어그램 전문가이자 관계 코치입니다. 아래 정보를 바탕으로 관계 시나리오를 작성해주세요.
+    [대상자 정보]
+    {people_info}
+    [관계]
+    - {relationship}
+    [상황]
+    - {situation}
+    [요청 사항]
+    1. **각 대상자의 에니어그램 유형에 대한 상세 설명:**
+       - 각 대상자의 핵심 가치, 성격적 특징, 강점과 약점, 주의할 점 등을 포함하여 자세히 설명해주세요.
+    2. **에니어그램 유형을 바탕으로 한 구체적인 시나리오 작성:**
+       - 상황에 맞는 시나리오를 작성하고, 각 단계를 구체적인 대화 장면으로 묘사해주세요.
+       - 각 단계에서 등장인물들의 감정 변화와 반응을 상세히 묘사하고, 각 인물의 말과 행동이 에니어그램 특성과 어떻게 연결되는지 부가 설명해주세요.
+       - 시나리오에 모든 등장인물의 이름, 성별, 에니어그램 유형이 모두 반영되어야 합니다.
+    3. **서로 이해하기 위한 실질적인 조언:**
+       - 각 대상자가 서로의 에니어그램 유형을 이해하고, 더 나은 관계를 형성하기 위한 구체적인 조언을 제시해주세요.
+       - 조언에 모든 등장인물의 이름, 성별, 에니어그램 유형이 모두 반영되어야 합니다.
+       - 각 대상자가 다른 대상자에게 해주면 좋은 말과 행동의 구체적인 예와 왜 좋은지에 대한 이유도 설명해주세요.
+    """
 
     full_text = ""
-    for partner_info in partner_infos:
-        partner_name = partner_info["name"]
-        partner_gender = partner_info["gender"]
-        partner_type = partner_info["type"]
-        partner_wing = partner_info["wing"]
-        partner_instinct = partner_info["instinct"]
-
-        partner_type_info = create_type_info(partner_name, partner_gender, partner_type, partner_wing, partner_instinct)
-
-        system_prompt = f"""
-        당신은 에니어그램 전문가이자 관계 코치입니다. 아래 정보를 바탕으로 관계 시나리오를 작성해주세요.
-        [대상자 정보]
-        - {my_type_info}
-        - {partner_type_info}
-        [관계]
-        - {relationship}
-        [상황]
-        - {situation}
-        [요청 사항]
-        1. **{my_name} ({my_gender})의 에니어그램 유형에 대한 상세 설명:**
-           - 핵심 가치, 성격적 특징, 강점과 약점, 주의할 점 등을 포함하여 자세히 설명해주세요.
-        2. **{partner_name} ({partner_gender})의 에니어그램 유형에 대한 상세 설명:**
-           - 핵심 가치, 성격적 특징, 강점과 약점, 주의할 점 등을 포함하여 자세히 설명해주세요.
-        3. **에니어그램 유형을 바탕으로 한 구체적인 시나리오 작성:**
-           - 만남, 대화 시작, 활동, 갈등 발생 및 해결 등 각 단계의 구체적인 대화 장면을 묘사해주세요.
-           - 각 단계에서 {my_name}과 {partner_name}의 감정 변화와 반응을 상세히 묘사하고, 각 인물의 말과 행동이 에니어그램 특성과 어떻게 연결되는지 부가 설명해주세요.
-           - 시나리오에 {my_name}과 {partner_name}의 이름, 성별, 에니어그램 유형이 모두 반영되어야 합니다.
-        4. **상대방을 서로 이해하기 위한 실질적인 조언:**
-           - 각 대상자가 상대방의 에니어그램 유형을 이해하고, 더 나은 관계를 형성하기 위한 구체적인 조언을 제시해주세요.
-           - 조언에 {my_name}과 {partner_name}의 이름, 성별, 에니어그램 유형이 모두 반영되어야 합니다.
-           - 마지막에 상대방에게 해주면 좋은 말과 행동의 구체적인 예와 왜 상대방에게 좋은지에 대한 이유도 설명해주세요.
-        """
-
-        try:
-            response = model.generate_content([system_prompt], stream=True)
-            full_text += f"## {partner_name}과의 관계 시나리오\n\n"
-            for chunk in response:
-                full_text += chunk.text
-                html_text = markdown.markdown(full_text)
-                scenario_output_area.markdown(html_text, unsafe_allow_html=True)
-                time.sleep(0.05)  # 출력 속도 조절
-            full_text += "\n\n---\n\n"  # 시나리오 구분선 추가
-        except Exception as e:
-            st.error(f"Error: {str(e)}")
-            return ""
+    try:
+        response = model.generate_content([system_prompt], stream=True)
+        for chunk in response:
+            full_text += chunk.text
+            html_text = markdown.markdown(full_text)
+            scenario_output_area.markdown(html_text, unsafe_allow_html=True)
+            time.sleep(0.05)
+    except Exception as e:
+        st.error(f"Error: {str(e)}")
+        return ""
 
     return full_text
 
+
 # Streamlit UI
 st.set_page_config(page_title="에니어그램 관계 시뮬레이터", page_icon="🤝")
-
 st.title("🤝 에니어그램 관계 시뮬레이터")
-st.write("여러 사람과의 에니어그램 관계 시나리오를 생성해보세요.")
-
-# 사이드바에 탭 배치
-with st.sidebar:
-    st.subheader("👤 내 정보")
-    my_name = st.text_input("내 이름/닉네임", "나")
-    my_gender = st.radio("내 성별", ["남성", "여성"])
-    my_type = st.selectbox("내 에니어그램 유형", list(enneagram_types.keys()))
-    my_wing = st.selectbox("내 날개 유형 (선택)", [""] + list(wing_types.keys()))
-    my_instinct = st.selectbox("내 본능 유형 (선택)", [""] + list(instinct_variants.keys()))
-
-    st.subheader("👥 상대방 정보")
-    num_partners = st.number_input("상대방 수", min_value=1, max_value=10, value=1, step=1)
-
-    partner_infos = []
-    for i in range(int(num_partners)):
-        st.write(f"**상대방 {i+1} 정보**")
-        partner_name = st.text_input(f"상대방 {i+1} 이름/닉네임", f"상대방{i+1}")
-        partner_gender = st.radio(f"상대방 {i+1} 성별", ["남성", "여성"], key=f"partner_gender_{i}")
-        partner_type = st.selectbox(f"상대방 {i+1} 에니어그램 유형", list(enneagram_types.keys()), key=f"partner_type_{i}")
-        partner_wing = st.selectbox(f"상대방 {i+1} 날개 유형 (선택)", [""] + list(wing_types.keys()), key=f"partner_wing_{i}")
-        partner_instinct = st.selectbox(f"상대방 {i+1} 본능 유형 (선택)", [""] + list(instinct_variants.keys()), key=f"partner_instinct_{i}")
-        partner_infos.append({
-            "name": partner_name,
-            "gender": partner_gender,
-            "type": partner_type,
-            "wing": partner_wing if partner_wing != "" else None,
-            "instinct": partner_instinct if partner_instinct != "" else None
-        })
-
-    relationship_type = st.radio("관계 유형 선택", ["2인 관계", "다자 관계"])
-    if relationship_type == "2인 관계":
-        relationship = st.selectbox("관계 유형", list(relationship_types_two.keys()))
-    else:  # 다자 관계
-        relationship = st.selectbox("관계 유형", list(relationship_types_multiple.keys()))
-
-    situation = st.text_area("상황 (예: 갈등 상황, 특별한 이벤트 등)", height=150)
+st.write("참여자들의 에니어그램 유형, 관계, 상황을 입력하여 관계 시나리오를 생성해보세요.")
+
+# 사람 추가/삭제 기능
+num_people = st.number_input("참여자 수", min_value=2, value=2)
+people = []
+
+for i in range(num_people):
+    with st.expander(f"참여자 {i+1} 정보"):
+        person = {}
+        person['name'] = st.text_input(f"이름/닉네임", key=f"name_{i}")
+        person['gender'] = st.radio("성별", ["남성", "여성"], key=f"gender_{i}")
+        person['type'] = st.selectbox("에니어그램 유형", list(enneagram_types.keys()), key=f"type_{i}")
+        person['wing'] = st.selectbox("날개 유형 (선택)", [""] + list(wing_types.keys()), key=f"wing_{i}")
+        person['instinct'] = st.selectbox("본능 유형 (선택)", [""] + list(instinct_variants.keys()), key=f"instinct_{i}")
+        people.append(person)
+
+
+# 관계 유형 선택 (참여자 수에 따라 다른 옵션 제공)
+if num_people == 2:
+    relationship = st.selectbox("관계 유형", list(relationship_types_two.keys()))
+else:
+    relationship = st.selectbox("관계 유형", list(relationship_types_multiple.keys()
+                                              
+situation = st.text_area("상황 (예: 갈등 상황, 특별한 이벤트 등)", height=150)
 
 generate_button = st.button("시나리오 생성 ➡️")
 
@@ -185,6 +168,6 @@ generate_button = st.button("시나리오 생성 ➡️")
 scenario_output_area = st.empty()
 
 if generate_button:
-    full_text = generate_relationship_scenario(my_name, my_gender, my_type, my_wing, my_instinct, partner_infos, relationship, situation)
+    full_text = generate_relationship_scenario(people, relationship, situation)
     if full_text:
-        scenario_output_area.markdown(markdown.markdown(full_text), unsafe_allow_html=True)
+        scenario_output_area.markdown(markdown.markdown(full_text), unsafe_allow_html=True)