Update app.py

app.py

@@ -28,12 +28,34 @@ options = [
     ["리더십을 발휘한다", "팀원으로서 지원한다"]
 ]
 
-# MBTI 유형 계산 함수
+# MBTI 유형 계산 및 정보 반환 함수
 def calculate_mbti(*answers):
-    # 단순화된 예시 로직: 선택된 옵션의 인덱스를 기반으로 MBTI 유형 결정
-    mbti_type = "INFP"  # 예시 결과
-    mbti_interpretation = "당신은 이상적이고, 열정적이며, 창의적인 사람입니다."  # 예시 해석
-    return mbti_type, mbti_interpretation
+    # MBTI 차원 점수 초기화
+    dimensions = {"E": 0, "I": 0, "S": 0, "N": 0, "T": 0, "F": 0, "J": 0, "P": 0}
+    
+    # 사용자의 선택에 따라 차원 점수 업데이트
+    for answer, (introvert, extrovert) in zip(answers, [(1, 0), (0, 1), (1, 0), (0, 1), (1, 0), (0, 1), (0, 1), (0, 1), (1, 0), (1, 0)]):
+        if answer == 0:  # 첫 번째 옵션 선택
+            dimensions["E"] += extrovert
+            dimensions["I"] += introvert
+        else:  # 두 번째 옵션 선택
+            dimensions["S"] += extrovert
+            dimensions["N"] += introvert
+            dimensions["T"] += extrovert
+            dimensions["F"] += introvert
+            dimensions["J"] += extrovert
+            dimensions["P"] += introvert
+    
+    # MBTI 유형 결정
+    mbti_type = ""
+    mbti_type += "E" if dimensions["E"] >= dimensions["I"] else "I"
+    mbti_type += "S" if dimensions["S"] >= dimensions["N"] else "N"
+    mbti_type += "T" if dimensions["T"] >= dimensions["F"] else "F"
+    mbti_type += "J" if dimensions["J"] >= dimensions["P"] else "P"
+    
+    # MBTI 유형에 따른 성격의 장단점과 유명인 반환
+    mbti_details = mbti_info.get(mbti_type, {"장단점": "정보 없음", "유명인": "정보 없음"})
+    return mbti_type, mbti_details["장단점"], mbti_details["유명인"]
 
 # Gradio 인터페이스 구성
 iface = gr.Interface(
@@ -41,10 +63,11 @@ iface = gr.Interface(
     inputs=[gr.Radio(options[i], label=questions[i]) for i in range(10)],
     outputs=[
         gr.Textbox(label="당신의 MBTI 유형"),
-        gr.Textbox(label="유형 해석")
+        gr.Textbox(label="유형의 장단점"),
+        gr.Textbox(label="대표적인 유명인")
     ],
     title="MBTI 진단 테스트",
-    description="10개의 질문에 답하여 당신의 MBTI 유형을 알아보세요."
+    description="10개의 질문에 답하여 당신의 MBTI 유형과 그 유형의 장단점, 대표적인 유명인을 알아보세요."
 )
 
 # 인터페이스 실행