Update app.py

app.py

@@ -1,27 +1,32 @@
 import gradio as gr
 
-# 사용자의 응답을 분석하여 MBTI 유형을 계산하는 함수
 def mbti_diagnosis(*answers):
     dimension_scores = {'E': 0, 'I': 0, 'N': 0, 'S': 0, 'T': 0, 'F': 0, 'J': 0, 'P': 0}
-    dimensions = ['E', 'I', 'N', 'S', 'T', 'F', 'J', 'P']
     
-    # 각 문항에 따라 차원 점수를 업데이트
+    # 각 문항에 따라 차원 점수를 업데이트 (보정된 로직)
     for i, answer in enumerate(answers):
-        # '예' 응답은 1로, '아니오' 응답은 0으로 계산
         score = 1 if answer == "예" else 0
-        if i % 4 == 0 or i % 4 == 1:
-            dimension_scores[dimensions[i % 8]] += score
-        else:
-            dimension_scores[dimensions[i % 8 + 1]] += score
-    
+        if i in [0, 1, 4, 5, 8, 9, 12, 13, 16, 17]: # E, N, T, J 차원에 해당하는 문항 인덱스
+            if i % 2 == 0: # E, N
+                dimension_scores['E' if i < 8 else 'N'] += score
+            else: # I, S
+                dimension_scores['I' if i < 8 else 'S'] += score
+        else: # T, F, J, P 차원에 해당하는 문항 인덱스
+            if i % 2 == 0: # T, J
+                dimension_scores['T' if i < 16 else 'J'] += score
+            else: # F, P
+                dimension_scores['F' if i < 16 else 'P'] += score
+
     # MBTI 유형 결정
     mbti_type = ''
-    mbti_type += 'E' if dimension_scores['E'] >= dimension_scores['I'] else 'I'
-    mbti_type += 'N' if dimension_scores['N'] >= dimension_scores['S'] else 'S'
-    mbti_type += 'T' if dimension_scores['T'] >= dimension_scores['F'] else 'F'
-    mbti_type += 'J' if dimension_scores['J'] >= dimension_scores['P'] else 'P'
+    mbti_type += 'E' if dimension_scores['E'] > dimension_scores['I'] else 'I'
+    mbti_type += 'N' if dimension_scores['N'] > dimension_scores['S'] else 'S'
+    mbti_type += 'T' if dimension_scores['T'] > dimension_scores['F'] else 'F'
+    mbti_type += 'J' if dimension_scores['J'] > dimension_scores['P'] else 'P'
     
     return mbti_type
+
+
 questions = [
     "사람들과 어울리는 것을 좋아하나요?",
     "구체적인 사실보다는 아이디어에 더 관심이 많나요?",