Update app.py

app.py

@@ -76,14 +76,18 @@ relationship_types_multiple = {
 }
 
 def generate_relationship_scenario(my_name, my_gender, my_type, my_wing, my_instinct, partner_infos, relationship, situation):
+
+    # 에니어그램 유형 정보 문자열 생성 (날개/본능 포함)
     def create_type_info(name, gender, type, wing, instinct):
         type_info = f"{name} ({gender}): {type} ({enneagram_types.get(type, '알 수 없는 유형')})"
-        if wing:  # wing 값이 있는 경우에만 추가
+        if wing:
             type_info += f" 날개: {wing_types.get(wing)}"
-        if instinct:  # instinct 값이 있는 경우에만 추가
+        if instinct:
             type_info += f", 본능: {instinct_variants.get(instinct)}"
         return type_info
 
+    my_type_info = create_type_info(my_name, my_gender, my_type, my_wing, my_instinct)
+
     full_text = ""
     for partner_info in partner_infos:
         partner_name = partner_info["name"]
@@ -96,7 +100,6 @@ def generate_relationship_scenario(my_name, my_gender, my_type, my_wing, my_inst
 
         system_prompt = f"""
         당신은 에니어그램 전문가이자 관계 코치입니다. 아래 정보를 바탕으로 관계 시나리오를 작성해주세요.
-
         [대상자 정보]
         - {my_type_info}
         - {partner_type_info}
@@ -104,7 +107,6 @@ def generate_relationship_scenario(my_name, my_gender, my_type, my_wing, my_inst
         - {relationship}
         [상황]
         - {situation}
-
         [요청 사항]
         1. **{my_name} ({my_gender})의 에니어그램 유형에 대한 상세 설명:**
            - 핵심 가치, 성격적 특징, 강점과 약점, 주의할 점 등을 포함하여 자세히 설명해주세요.
@@ -185,4 +187,4 @@ 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)
     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)