Update app.py

app.py

@@ -1,27 +1,261 @@
 import gradio as gr
 import numpy as np
 
-# 질문 리스트 (예시)
+# 질문 리스트
 questions = [
     "당신은 혼자 있을 때 에너지를 얻나요?",
     "당신은 계획을 세우는 것을 좋아하나요?",
     "당신은 새로운 상황에 쉽게 적응하나요?",
-    # 더 많은 질문을 추가하세요.
+    "당신은 감정보다는 논리에 더 의존하나요?",
+    "당신은 다른 사람을 돕는 것을 좋아하나요?",
+    "당신은 사교적인 성격을 가지고 있나요?",
+    "당신은 일을 끝내기 전에 계획을 세우는 편인가요?",
+    "당신은 사실과 데이터를 중시하나요?",
+    "당신은 감정 표현을 잘 하나요?",
+    "당신은 문제를 해결할 때 창의적인 방법을 사용하나요?",
+    "당신은 다른 사람들과의 협업을 선호하나요?",
+    "당신은 결정을 내릴 때 직감을 신뢰하나요?",
+    "당신은 규칙과 절차를 따르는 것을 좋아하나요?",
+    "당신은 주변 환경에 민감하게 반응하나요?",
+    "당신은 일을 처리할 때 효율성을 중시하나요?",
+    "당신은 남들이 생각하는 것에 대해 많이 신경 쓰나요?",
+    "당신은 자연스럽게 리더 역할을 맡게 되나요?",
+    "당신은 새로운 아이디어를 제안하는 것을 좋아하나요?",
+    "당신은 대인 관계에서 갈등을 피하려 하나요?",
+    "당신은 목표를 설정하고 이를 달성하기 위해 노력하나요?"
 ]
 
 # MBTI 유형 예시 데이터
 mbti_types = {
-    "INTJ": {"성격": "독창적이고 전략적인 사고를 가진 조용한 리더", "직업": "과학자, 엔지니어", "궁합": ["ENFP", "ENTP"], "나쁜 궁합": ["ESFP"]},
-    "ENFP": {"성격": "열정적이고 창의적인 사고를 가진 사람", "직업": "예술가, 작가", "궁합": ["INTJ", "INFJ"], "나쁜 궁합": ["ISTJ"]},
-    # 다른 유형도 추가
+    "INTJ": {
+        "성격": "독창적이고 전략적인 사고를 가진 조용한 리더",
+        "직업": "과학자, 엔지니어, 데이터 분석가",
+        "궁합": ["ENFP", "ENTP"],
+        "나쁜 궁합": ["ESFP", "ESTP"],
+        "궁합 설명": {
+            "ENFP": "ENFP는 INTJ의 이상주의를 자극하고, 서로의 강점과 약점을 보완합니다.",
+            "ENTP": "ENTP는 INTJ의 창의성과 전략적 사고를 자극하여 생산적인 파트너십을 형성합니다."
+        },
+        "나쁜 궁합 설명": {
+            "ESFP": "ESFP는 INTJ와 정반대의 성향을 가지고 있어 갈등이 잦을 수 있습니다.",
+            "ESTP": "ESTP는 실용적인 접근을 선호하는 반면, INTJ는 장기적인 계획을 중시합니다."
+        }
+    },
+    "INTP": {
+        "성격": "논리적이고 분석적인 사고를 가진 혁신가",
+        "직업": "연구원, 프로그래머, 철학자",
+        "궁합": ["ENTJ", "ESTJ"],
+        "나쁜 궁합": ["ESFJ", "ESTP"],
+        "궁합 설명": {
+            "ENTJ": "ENTJ는 INTP의 분석적 사고를 촉진하고 함께 전략적인 목표를 달성할 수 있습니다.",
+            "ESTJ": "ESTJ는 INTP의 논리적 접근을 존중하며 실용적인 해결책을 찾는 데 도움을 줍니다."
+        },
+        "나쁜 궁합 설명": {
+            "ESFJ": "ESFJ는 감정적인 결정을 중시하는 반면, INTP는 논리를 중시하여 갈등이 생길 수 있습니다.",
+            "ESTP": "ESTP는 즉각적인 행동을 선호하며, INTP의 깊은 사고와는 충돌할 수 있습니다."
+        }
+    },
+    "ENTJ": {
+        "성격": "대담하고 결단력 있는 리더",
+        "직업": "기업가, 경영자, 변호사",
+        "궁합": ["INTP", "INTJ"],
+        "나쁜 궁합": ["INFP", "ISFP"],
+        "궁합 설명": {
+            "INTP": "INTP는 ENTJ의 전략적 사고를 보완하며, 논리적 접근을 통해 문제 해결에 도움을 줍니다.",
+            "INTJ": "INTJ는 ENTJ와 함께 장기적인 목표를 설정하고, 실행할 수 있는 강력한 파트너가 됩니다."
+        },
+        "나쁜 궁합 설명": {
+            "INFP": "INFP는 감정적 접근을 선호하며, ENTJ의 냉정한 논리적 접근과 충돌할 수 있습니다.",
+            "ISFP": "ISFP는 즉흥적이고 감정적인 경향이 있어, ENTJ의 체계적이고 논리적인 성향과 맞지 않을 수 있습니다."
+        }
+    },
+    "ENTP": {
+        "성격": "창의적이고 논쟁을 즐기는 비평가",
+        "직업": "마케터, 변호사, 기자",
+        "궁합": ["INFJ", "INTJ"],
+        "나쁜 궁합": ["ISFJ", "ISTJ"],
+        "궁합 설명": {
+            "INFJ": "INFJ는 ENTP의 창의적 사고를 이해하며, 이상적인 목표 설정을 도와줍니다.",
+            "INTJ": "INTJ는 ENTP와 함께 혁신적인 아이디어를 실현하는 데 강력한 파트너가 됩니다."
+        },
+        "나쁜 궁합 설명": {
+            "ISFJ": "ISFJ는 전통적이고 보수적인 ���향이 강해, ENTP의 혁신적이고 비판적인 사고와 충돌할 수 있습니다.",
+            "ISTJ": "ISTJ는 실용적인 접근을 선호하며, ENTP의 실험적이고 모험적인 성향과 갈등이 있을 수 있습니다."
+        }
+    },
+    "INFJ": {
+        "성격": "통찰력 있고 헌신적인 이상주의자",
+        "직업": "상담사, 심리학자, 작가",
+        "궁합": ["ENFP", "ENTP"],
+        "나쁜 궁합": ["ESTP", "ESFP"],
+        "궁합 설명": {
+            "ENFP": "ENFP는 INFJ의 이상주의와 깊은 감정을 이해하고 이를 격려합니다.",
+            "ENTP": "ENTP는 INFJ의 창의성을 자극하고, 혁신적인 아이디어를 실현하는 데 도움을 줍니다."
+        },
+        "나쁜 궁합 설명": {
+            "ESTP": "ESTP는 즉각적인 행동을 선호하며, INFJ의 깊은 사고와 감정적 접근과 충돌할 수 있습니다.",
+            "ESFP": "ESFP는 즉흥적이고 외향적인 성향이 강해, INFJ의 내성적이고 계획적인 성향과 맞지 않을 수 있습니다."
+        }
+    },
+    "INFP": {
+        "성격": "이상적이고 충실한 중재자",
+        "직업": "예술가, 작가, 상담사",
+        "궁합": ["ENFJ", "ENTJ"],
+        "나쁜 궁합": ["ESTJ", "ESFJ"],
+        "궁합 설명": {
+            "ENFJ": "ENFJ는 INFP의 이상주의를 격려하고, 함께 더 나은 세상을 만들기 위해 노력합니다.",
+            "ENTJ": "ENTJ는 INFP의 이상을 현실로 만들기 위해 논리적이고 전략적인 도움을 줍니다."
+        },
+        "나쁜 궁합 설명": {
+            "ESTJ": "ESTJ는 실용적이고 조직적인 성향이 강해, INFP의 이상적이고 감정적인 성향과 충돌할 수 있습니다.",
+            "ESFJ": "ESFJ는 사회적 규범을 중시하며, INFP의 개성과 독립성을 존중하지 않을 수 있습니다."
+        }
+    },
+    "ENFJ": {
+        "성격": "카리스마 있고 사교적인 리더",
+        "직업": "교사, 정치인, 사회운동가",
+        "궁합": ["INFP", "INFJ"],
+        "나쁜 궁합": ["ISTP", "INTP"],
+        "궁합 설명": {
+            "INFP": "INFP는 ENFJ의 감정적 깊이와 이상주의를 이해하며, 함께 의미 있는 목표를 추구합니다.",
+            "INFJ": "INFJ는 ENFJ와 함께 사람들을 돕고, 세상을 더 나은 곳으로 만드는 데 협력할 수 있습니다."
+        },
+        "나쁜 궁합 설명": {
+            "ISTP": "ISTP는 논리적이고 즉흥적인 성향이 강해, ENFJ의 감정적이고 계획적인 성향과 충돌할 수 있습니다.",
+            "INTP": "INTP는 논리적이고 분석적인 성향이 강해, ENFJ의 감정적이고 사교적인 성향과 맞지 않을 수 있습니다."
+        }
+    },
+    "ENFP": {
+        "성격": "열정적이고 창의적인 사고를 가진 사람",
+        "직업": "예술가, 작가, 마케터",
+        "궁합": ["INTJ", "INFJ"],
+        "나쁜 궁합": ["ISTJ", "ISFJ"],
+        "궁합 설명": {
+            "INTJ": "INTJ는 ENFP의 창의적인 아이디어를 현실로 실현하는 데 도움을 줍니다.",
+            "INFJ": "INFJ는 ENFP의 열정을 이해하며, 함께 이상적인 목표를 추구할 수 있습니다."
+        },
+        "나쁜 궁합 설명": {
+            "ISTJ": "ISTJ는 전통적이고 실용적인 성향이 강해, ENFP의 창의적이고 모험적인 성향과 충돌할 수 있습니다.",
+            "ISFJ": "ISFJ는 사회적 규범을 중시하며, ENFP의 자유롭고 개방적인 성향과 맞지 않을 수 있습니다."
+        }
+    },
+    "ISTJ": {
+        "성격": "실용적이고 신뢰할 수 있는 관리인",
+        "직업": "회계사, 관리자, 공무원",
+        "궁합": ["ESFP", "ESTP"],
+        "나쁜 궁합": ["ENFP", "ENTP"],
+        "궁합 설명": {
+            "ESFP": "ESFP는 ISTJ의 실용적인 성향을 존중하며, 함께 현실적인 목표를 달성할 수 있습니다.",
+            "ESTP": "ESTP는 ISTJ의 조직적이고 계획적인 성향을 보완하며, 빠른 문제 해결에 도움을 줍니다."
+        },
+        "나쁜 궁합 설명": {
+            "ENFP": "ENFP는 창의적이고 자유로운 성향이 강해, ISTJ의 전통적이고 보수적인 성향과 충돌할 수 있습니다.",
+            "ENTP": "ENTP는 실험적이고 모험적인 성향이 강해, ISTJ의 실용적이고 신중한 성향과 맞지 않을 수 있습니다."
+        }
+    },
+    "ISFJ": {
+        "성격": "성실하고 배려심 깊은 보호자",
+        "직업": "간호사, 교사, 사회복지사",
+        "궁합": ["ESFP", "ESTP"],
+        "나쁜 궁합": ["ENTP", "ENFP"],
+        "궁합 설명": {
+            "ESFP": "ESFP는 ISFJ의 배려심과 성실함을 존중하며, 함께 따뜻한 인간관계를 유지할 수 있습니다.",
+            "ESTP": "ESTP는 ISFJ의 세심한 성향을 보완하며, 현실적인 문제 해결에 도움을 줍니다."
+        },
+        "나쁜 궁합 설명": {
+            "ENTP": "ENTP는 실험적이고 비판적인 성향이 강해, ISFJ의 전통적이고 보호적인 성향과 충돌할 수 있습니다.",
+            "ENFP": "ENFP는 자유롭고 창의적인 성향이 강해, ISFJ의 조직적이고 보수적인 성향과 맞지 않을 수 있습니다."
+        }
+    },
+    "ESTJ": {
+        "성격": "실용적이고 조직적인 관리자",
+        "직업": "경영자, 군인, 공무원",
+        "궁합": ["ISTP", "INTP"],
+        "나쁜 궁합": ["INFP", "INFJ"],
+        "궁합 설명": {
+            "ISTP": "ISTP는 ESTJ의 조직적이고 실용적인 성향을 보완하며, 함께 효율적인 문제 해결을 도모할 수 있습니다.",
+            "INTP": "INTP는 ESTJ의 논리적 사고를 보완하며, 전략적인 목표 달성에 도움을 줍니다."
+        },
+        "나쁜 궁합 설명": {
+            "INFP": "INFP는 이상적이고 감정적인 성향이 강해, ESTJ의 실용적이고 논리적인 성향과 충돌할 수 있습니다.",
+            "INFJ": "INFJ는 감정적이고 내성적인 성향이 강해, ESTJ의 외향적이고 실용적인 성향과 맞지 않을 수 있습니다."
+        }
+    },
+    "ESFJ": {
+        "성격": "사교적이고 협력적인 지원자",
+        "직업": "교사, 간호사, 이벤트 플래너",
+        "궁합": ["ISFP", "ISTP"],
+        "나쁜 궁합": ["INTP", "INTJ"],
+        "궁합 설명": {
+            "ISFP": "ISFP는 ESFJ의 배려심을 이해하며, 함께 따뜻한 인간관계를 유지할 수 있습니다.",
+            "ISTP": "ISTP는 ESFJ의 세심함을 보완하며, 현실적인 문제 해결에 도움을 줍니다."
+        },
+        "나쁜 궁합 설명": {
+            "INTP": "INTP는 논리적이고 분석적인 성향이 강해, ESFJ의 감정적이고 사교적인 성향과 충돌할 수 있습니다.",
+            "INTJ": "INTJ는 전략적이고 장기적인 계획을 중시하여, ESFJ의 즉각적인 행동과 맞지 않을 수 있습니다."
+        }
+    },
+    "ISTP": {
+        "성격": "유연하고 창의적인 문제 해결사",
+        "직업": "엔지니어, 기술자, 파일럿",
+        "궁합": ["ESFJ", "ESTJ"],
+        "나쁜 궁합": ["ENFJ", "ESFP"],
+        "궁합 설명": {
+            "ESFJ": "ESFJ는 ISTP의 창의적인 문제 해결 능력을 존중하며, 함께 실용적인 목표를 달성할 수 있습니다.",
+            "ESTJ": "ESTJ는 ISTP의 유연함을 보완하며, 체계적인 문제 해결에 도움을 줍니다."
+        },
+        "나쁜 궁합 설명": {
+            "ENFJ": "ENFJ는 감정적이고 사교적인 성향이 강해, ISTP의 논리적이고 독립적인 성향과 충돌할 수 있습니다.",
+            "ESFP": "ESFP는 즉흥적이고 외향적인 성향이 강해, ISTP의 신중하고 분석적인 성향과 맞지 않을 수 있습니다."
+        }
+    },
+    "ISFP": {
+        "성격": "예술적이고 감각적인 장인",
+        "직업": "디자이너, 예술가, 음악가",
+        "궁합": ["ESFJ", "ESTJ"],
+        "나쁜 궁합": ["ENTJ", "ENTP"],
+        "궁합 설명": {
+            "ESFJ": "ESFJ는 ISFP의 감정적이고 예술적인 성향을 이해하며, 함께 따뜻한 인간관계를 유지할 수 있습니다.",
+            "ESTJ": "ESTJ는 ISFP의 유연함을 보완하며, 실용적인 문제 해결에 도움을 줍니다."
+        },
+        "나쁜 궁합 설명": {
+            "ENTJ": "ENTJ는 논리적이고 계획적인 성향이 강해, ISFP의 즉흥적이고 감정적인 성향과 충돌할 수 있습니다.",
+            "ENTP": "ENTP는 비판적이고 실험적인 성향이 강해, ISFP의 감정적이고 예술적인 성향과 맞지 않을 수 있습니다."
+        }
+    },
+    "ESTP": {
+        "성격": "모험적이고 즉흥적인 활동가",
+        "직업": "기업가, 운동선수, 경찰",
+        "궁합": ["ISFJ", "ISTJ"],
+        "나쁜 궁합": ["INFJ", "INTJ"],
+        "궁합 설명": {
+            "ISFJ": "ISFJ는 ESTP의 활동적이고 모험적인 성향을 이해하며, 함께 현실적인 목표를 달성할 수 있습니다.",
+            "ISTJ": "ISTJ는 ESTP의 즉흥적인 성향을 보완하며, 체계적인 문제 해결에 도움을 줍니다."
+        },
+        "나쁜 궁합 설명": {
+            "INFJ": "INFJ는 감정적이고 내성적인 성향이 강해, ESTP의 외향적이고 즉흥적인 성향과 충돌할 수 있습니다.",
+            "INTJ": "INTJ는 전략적이고 장기적인 계획을 중시하여, ESTP의 즉각적인 행동과 맞지 않을 수 있습니다."
+        }
+    },
+    "ESFP": {
+        "성격": "사교적이고 열정적인 연예인",
+        "직업": "배우, 연예인, 이벤트 플래너",
+        "궁합": ["ISFJ", "ISTJ"],
+        "나쁜 궁합": ["INTJ", "INFJ"],
+        "궁합 설명": {
+            "ISFJ": "ISFJ는 ESFP의 활발하고 사교적인 성향을 이해하며, 함께 따��한 인간관계를 유지할 수 있습니다.",
+            "ISTJ": "ISTJ는 ESFP의 즉흥적인 성향을 보완하며, 현실적인 문제 해결에 도움을 줍니다."
+        },
+        "나쁜 궁합 설명": {
+            "INTJ": "INTJ는 논리적이고 계획적인 성향이 강해, ESFP의 즉흥적이고 감정적인 성향과 충돌할 수 있습니다.",
+            "INFJ": "INFJ는 감정적이고 내성적인 성향이 강해, ESFP의 외향적이고 활동적인 성향과 맞지 않을 수 있습니다."
+        }
+    }
 }
 
-# 질문에 따른 점수 계산 함수
 def calculate_mbti_responses(responses):
     scores = np.zeros(16)  # 16가지 MBTI 유형에 대한 점수
     for i, response in enumerate(responses):
-        # 점수 계산 로직 (예: 특정 질문이 특정 유형에 어떤 영향을 미치는지)
-        # 각 유형별 점수 계산 (간단히 가정)
         if response == "매우 그렇다":
             scores += np.random.randint(5, 10, size=16)
         elif response == "그렇다":
@@ -32,12 +266,27 @@ def calculate_mbti_responses(responses):
             scores -= np.random.randint(2, 5, size=16)
         elif response == "매우 아니다":
             scores -= np.random.randint(5, 10, size=16)
-
+    
     best_match_index = np.argmax(scores)
+    second_best_match_index = np.argsort(scores)[-2]  # 두 번째로 높은 점수의 인덱스
+    
+    # 가장 높은 점수와 두 번째로 높은 점수의 차이
+    score_difference = scores[best_match_index] - scores[second_best_match_index]
+    
+    # 신뢰도를 점수 차이에 따라 계산
+    if score_difference > 15:
+        percentage = 95 + np.random.randint(0, 5)  # 매우 높은 신뢰도
+    elif score_difference > 10:
+        percentage = 85 + np.random.randint(0, 10)  # 높은 신뢰도
+    elif score_difference > 5:
+        percentage = 75 + np.random.randint(0, 10)  # 중간 신뢰도
+    else:
+        percentage = 60 + np.random.randint(0, 15)  # 낮은 신뢰도
+
     best_match_type = list(mbti_types.keys())[best_match_index]
-    percentage = (scores[best_match_index] / sum(scores)) * 100
     return best_match_type, percentage
 
+
 # Gradio UI 구성
 def mbti_quiz(*responses):
     mbti_type, percentage = calculate_mbti_responses(responses)
@@ -45,14 +294,33 @@ def mbti_quiz(*responses):
     details = mbti_types[mbti_type]
     result += f"\n\n성격 설명: {details['성격']}"
     result += f"\n추천 직업: {details['직업']}"
-    result += f"\n\n좋은 궁합: {', '.join(details['궁합'])}"
-    result += f"\n나쁜 궁합: {', '.join(details['나쁜 궁합'])}"
-    return result
 
-# Gradio 입력 필드 구성
-inputs = [gr.inputs.Radio(["매우 그렇다", "그렇다", "보통이다", "아니다", "매우 아니다"], label=question) for question in questions]
-output = gr.outputs.Textbox(label="MBTI 결과")
+    # 좋은 궁합 설명 추가
+    good_matches = details['궁합']
+    good_explanations = details['궁합 설명']
+    result += f"\n\n좋은 궁합:"
+    for match in good_matches:
+        result += f"\n- {match}: {good_explanations[match]}"
+
+    # 나쁜 궁합 설명 추가
+    bad_matches = details['나쁜 궁합']
+    bad_explanations = details['나쁜 궁합 설명']
+    result += f"\n\n나쁜 궁합:"
+    for match in bad_matches:
+        result += f"\n- {match}: {bad_explanations[match]}"
+
+    # 이미지 파일 경로 추가
+    image_path = f"./{mbti_type}.webp"
+    
+    return result, image_path
+
+# Gradio UI 구성
+def display_results(text, image):
+    return text, image
+
+inputs = [gr.Radio(["매우 그렇다", "그렇다", "보통이다", "아니다", "매우 아니다"], label=question) for question in questions]
+output_text = gr.Textbox(label="MBTI 결과")
+output_image = gr.Image(label="유형 이미지")
 
-# Gradio 인터페이스 생성
-iface = gr.Interface(fn=mbti_quiz, inputs=inputs, outputs=output, title="MBTI 성격 유형 테스트", description="20개 이상의 질문을 통해 MBTI 성격 유형을 분석하세요.")
+iface = gr.Interface(fn=mbti_quiz, inputs=inputs, outputs=[output_text, output_image], title="MBTI 성격 유형 테스트", description="20개 이상의 질문을 통해 MBTI 성격 유형을 분석하세요.")
 iface.launch()