Update app.py

app.py

@@ -29,40 +29,28 @@ mbti_types = [
     "ESTP", "ESFP", "ENFP", "ENTP",
 ]
 
-def generate_date_scenario(my_name, my_gender, my_mbti, partner_name, partner_gender, partner_mbti, situation):
-
+def generate_date_scenario(my_name, my_gender, my_type, partner_name, partner_gender, partner_type, situation):
     system_prompt = f"""
     당신은 MBTI 전문가이자 관계 코치입니다. 아래 요청에 따라 데이트 시나리오를 작성해주세요.
-
     ---
-
     [대상자 정보]
-
-    - {my_name} ({my_gender}): {my_mbti}
-    - {partner_name} ({partner_gender}): {partner_mbti}
-
+    - {my_name} ({my_gender}): {my_type} ({mbti_types.get(my_type, '알 수 없는 유형')})
+    - {partner_name} ({partner_gender}): {partner_type} ({mbti_types.get(partner_type, '알 수 없는 유형')})
     [상황]
-
     - {situation}
-
     ---
-
     [요청 사항]
-
-    1. **{my_name} ({my_gender})의 MBTI 유형({my_mbti})에 대한 상세 설명:** 
-       - 핵심 가치, 성격적 특징, 강점과 약점, 주의할 점 등을 포함하여 자세히 설명해주세요.
-
-    2. **{partner_name} ({partner_gender})의 MBTI 유형({partner_mbti})에 대한 상세 설명:**
-       - 핵심 가치, 성격적 특징, 강점과 약점, 주의할 점 등을 포함하여 자세히 설명해주세요.
-
-    3. **MBTI 유형을 바탕으로 한 데이트 시나리오 작성:**
-       - 상호작용, 내면, 대화, 갈등 해결, 상대방 이해, 궁합 평가 및 조언, 유형 특성 반영 등의 요소를 포함하여 시나리오를 작성해주세요.
+    1. **{my_name} ({my_gender})의 MBTI 유형({my_type})에 대한 상세 설명:** 
+       - 주요 특징, 강점과 약점, 의사소통 스타일 등을 포함하여 자세히 설명해주세요.
+    2. **{partner_name} ({partner_gender})의 MBTI 유형({partner_type})에 대한 상세 설명:**
+       - 주요 특징, 강점과 약점, 의사소통 스타일 등을 포함하여 자세히 설명해주세요.
+    3. **MBTI 유형을 바탕으로 한 구체적인 데이트 시나리오 작성:**
+       - 만남, 대화 시작, 식사, 영화 관람, 갈등 발생 및 해결 등 데이트의 각 단계를 구체적으로 묘사해주세요.
+       - 각 단계에서 {my_name}과 {partner_name}의 감정 변화와 반응을 상세히 묘사해주세요.
        - 시나리오에 {my_name}과 {partner_name}의 이름, 성별, MBTI 유형이 모두 반영되어야 합니다.
-
-    4. **상대방을 서로 이해하기 위한 조언:**
-       - 각 대상자가 상대방의 MBTI 유형을 이해하고, 더 나은 관계를 형성하기 위한 실질적인 조언을 제공해주세요.
+    4. **상대방을 서로 이해하기 위한 실질적인 조언:**
+       - 각 대상자가 상대방의 MBTI 유형을 이해하고, 더 나은 관계를 형성하기 위한 구체적인 상황과 행동을 예시로 제시해주세요.
        - 조언에 {my_name}과 {partner_name}의 이름, 성별, MBTI 유형이 모두 반영되어야 합니다.
-
     ---
     """
 
@@ -81,7 +69,7 @@ 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="❤️")
+st.set_page_config(page_title="MBTI 데이트 시뮬레이터", page_icon="💖")
 
 colored_header(
     label="💖 MBTI 데이트 시뮬레이터 💖",
@@ -94,12 +82,12 @@ with st.sidebar:
     st.subheader("내 정보")
     my_name = st.text_input("내 이름(닉네임)", "나")
     my_gender = st.selectbox("내 성별", ["남성", "여성"])
-    my_mbti = st.selectbox("내 MBTI 유형", mbti_types)
+    my_type = st.selectbox("내 MBTI 유형", list(mbti_types.keys()))
 
     st.subheader("상대방 정보")
     partner_name = st.text_input("상대방 이름(닉네임)", "상대방")
     partner_gender = st.selectbox("상대방 성별", ["남성", "여성"])
-    partner_mbti = st.selectbox("상대방 MBTI 유형", mbti_types)
+    partner_type = st.selectbox("상대방 MBTI 유형", list(mbti_types.keys()))
 
     st.subheader("데이트 상황")
     situation = st.text_area("예: 첫 만남, 영화 관람 후 저녁 식사", height=150)
@@ -107,4 +95,9 @@ with st.sidebar:
     generate_scenario_button = st.button("💕 시나리오 생성! 💕")
 
 # 메인 영역 출력
-scenario_output_area = st.empty()
+scenario_output_area = st.empty()
+
+if generate_scenario_button:
+    full_text = generate_date_scenario(my_name, my_gender, my_type, partner_name, partner_gender, partner_type, situation)
+    if full_text:
+        scenario_output_area.markdown(markdown.markdown(full_text), unsafe_allow_html=True)