Update app.py

app.py

@@ -56,59 +56,35 @@ instinct_variants = {
 }
 
 def generate_date_scenario(my_name, my_gender, my_type, my_wing, my_instinct, partner_name, partner_gender, partner_type, partner_wing, partner_instinct, situation):
-    my_type_full = my_type
-    if my_wing:
-        my_type_full += my_wing
-    if my_instinct:
-        my_type_full += f" ({instinct_variants.get(my_instinct, '')})"
-
-    partner_type_full = partner_type
-    if partner_wing:
-        partner_type_full += partner_wing
-    if partner_instinct:
-        partner_type_full += f" ({instinct_variants.get(partner_instinct, '')})"
+
+    # 에니어그램 유형 정보 문자열 생성 (날개/본능 포함)
+    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)}"
+        return type_info
+
+    my_type_info = create_type_info(my_name, my_gender, my_type, my_wing, my_instinct)
+    partner_type_info = create_type_info(partner_name, partner_gender, partner_type, partner_wing, partner_instinct)
+
+
 
     system_prompt = f"""
     당신은 에니어그램 전문가이자 관계 코치입니다. 아래 요청에 따라 데이트 시나리오를 작성해주세요.
-    ---
+
     [대상자 정보]
-    - {my_name} ({my_gender}): {my_type}"""
-
-    if my_wing:
-        system_prompt += f" {my_wing}"
-    if my_instinct:
-        system_prompt += f" ({instinct_variants.get(my_instinct, '')})"
-    
-    system_prompt += f" ({enneagram_types.get(my_type, '알 수 없는 유형')})"
-
-    system_prompt += f"""
-    - {partner_name} ({partner_gender}): {partner_type}"""
-
-    if partner_wing:
-        system_prompt += f" {partner_wing}"
-    if partner_instinct:
-        system_prompt += f" ({instinct_variants.get(partner_instinct, '')})"
-    
-    system_prompt += f" ({enneagram_types.get(partner_type, '알 수 없는 유형')})"
-
-    system_prompt += f"""
+    - {my_type_info}
+    - {partner_type_info}
+
     [상황]
     - {situation}
-    ---
+
     [요청 사항]
-    1. **{my_name} ({my_gender})의 에니어그램 유형에 대한 상세 설명:**"""
-    
-    if my_wing or my_instinct:
-        system_prompt += f" (날개: {my_wing if my_wing else '없음'}, 본능: {instinct_variants.get(my_instinct, '없음')})"
-    
-    system_prompt += """ 
+    1. **{my_name} ({my_gender})의 에니어그램 유형에 대한 상세 설명:**
        - 핵심 가치, 성격적 특징, 강점과 약점, 주의할 점 등을 포함하여 자세히 설명해주세요.
-    2. **{partner_name} ({partner_gender})의 에니어그램 유형에 대한 상세 설명:**"""
-
-    if partner_wing or partner_instinct:
-        system_prompt += f" (날개: {partner_wing if partner_wing else '없음'}, 본능: {instinct_variants.get(partner_instinct, '없음')})"
-
-    system_prompt += """
+    2. **{partner_name} ({partner_gender})의 에니어그램 유형에 대한 상세 설명:**
        - 핵심 가치, 성격적 특징, 강점과 약점, 주의할 점 등을 포함하여 자세히 설명해주세요.
     3. **에니어그램 유형을 바탕으로 한 구체적인 데이트 시나리오 작성:**
        - 만남, 대화 시작, 식사, 영화 관람, 갈등 발생 및 해결 등 데이트의 각 단계를 구체적인 대화 장면으로 묘사해주세요.
@@ -117,7 +93,6 @@ def generate_date_scenario(my_name, my_gender, my_type, my_wing, my_instinct, pa
     4. **상대방을 서로 이해하기 위한 실질적인 조언:**
        - 각 대상자가 상대방의 에니어그램 유형을 이해하고, 더 나은 관계를 형성하기 위한 구체적인 상황과 행동을 예시로 제시해주세요.
        - 조언에 {my_name}과 {partner_name}의 이름, 성별, 에니어그램 유형이 모두 반영되어야 합니다.
-    ---
     """
 
     full_text = ""
@@ -127,37 +102,40 @@ def generate_date_scenario(my_name, my_gender, my_type, my_wing, my_instinct, pa
             full_text += chunk.text
             html_text = markdown.markdown(full_text)
             scenario_output_area.markdown(html_text, unsafe_allow_html=True)
-            time.sleep(0.05)
+            time.sleep(0.05)  # 출력 속도 조절
     except Exception as e:
         st.error(f"Error: {str(e)}")
         return ""
 
     return full_text
 
+
 # Streamlit 인터페이스
 st.set_page_config(page_title="에니어그램 데이트 시뮬레이터", page_icon="❤️")
 
 colored_header(
     label="💖 에니어그램 데이트 시뮬레이터 💖",
     description="나와 상대방의 에니어그램 유형을 기반으로 데이트 시나리오를 미리 체험해보세요!",
-    color_name="red-70"
+    color_name="red-70",
 )
 
+
 # 사이드바 입력 폼
 with st.sidebar:
     st.subheader("내 정보")
     my_name = st.text_input("내 이름(닉네임)", "나")
     my_gender = st.selectbox("내 성별", ["남성", "여성"])
     my_type = st.selectbox("내 에니어그램 유형", list(enneagram_types.keys()))
-    my_wing = st.selectbox("내 날개 유형 (선택)", [""] + list(wing_types.keys()))
-    my_instinct = st.selectbox("내 본능 유형 (선택)", [""] + list(instinct_variants.keys()))
+    my_wing = st.selectbox("내 날개 유형 (선택)", [""] + list(wing_types.keys()))  # 빈 문자열 추가
+    my_instinct = st.selectbox("내 본능 유형 (선택)", [""] + list(instinct_variants.keys()))  # 빈 문자열 추가
 
     st.subheader("상대방 정보")
     partner_name = st.text_input("상대방 이름(닉네임)", "상대방")
     partner_gender = st.selectbox("상대방 성별", ["남성", "여성"])
     partner_type = st.selectbox("상대방 에니어그램 유형", list(enneagram_types.keys()))
-    partner_wing = st.selectbox("상대방 날개 유형 (선택)", [""] + list(wing_types.keys()))
-    partner_instinct = st.selectbox("상대방 본능 유형 (선택)", [""] + list(instinct_variants.keys()))
+    partner_wing = st.selectbox("상대방 날개 유형 (선택)", [""] + list(wing_types.keys()))  # 빈 문자열 추가
+    partner_instinct = st.selectbox("상대방 본능 유형 (선택)", [""] + list(instinct_variants.keys()))  # 빈 문자열 추가
+
 
     st.subheader("데이트 상황")
     situation = st.text_area("예: 첫 만남, 영화 관람 후 저녁 식사", height=150)
@@ -168,7 +146,7 @@ with st.sidebar:
 scenario_output_area = st.empty()
 
 if generate_scenario_button:
-    # 날개와 본능 유형이 선택되지 않은 경우
+    # 날개/본능 유형 선택 안한 경우 None으로 처리
     my_wing = my_wing if my_wing != "" else None
     my_instinct = my_instinct if my_instinct != "" else None
     partner_wing = partner_wing if partner_wing != "" else None