Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import matplotlib.pyplot as plt
 
 class MBTIAnalyzer:
     def __init__(self):
@@ -35,10 +36,13 @@ class MBTIAnalyzer:
         self.scores = {'E': 0, 'I': 0, 'S': 0, 'N': 0, 'T': 0, 'F': 0, 'J': 0, 'P': 0}
 
     def ask_question(self, category, answer):
-        if answer == '1':
-            self.scores[category[0]] += 1
-        elif answer == '2':
-            self.scores[category[1]] += 1
+        # 점수를 5단계로 계산합니다. 매우 그렇다: 2, 그렇다: 1, 보통이다: 0, 아니다: -1, 매우 아니다: -2
+        scale = {"매우 그렇다": 2, "그렇다": 1, "보통이다": 0, "아니다": -1, "매우 아니다": -2}
+        score = scale[answer]
+        if score > 0:
+            self.scores[category[0]] += score
+        elif score < 0:
+            self.scores[category[1]] -= score
 
     def analyze(self, answers):
         idx = 0
@@ -67,25 +71,36 @@ class MBTIAnalyzer:
         result += 'T' if percentages['T'] >= 50 else 'F'
         result += 'J' if percentages['J'] >= 50 else 'P'
 
-        output = f"MBTI 유형: {result}\n\n세부 지수:\n"
-        for trait, percentage in percentages.items():
-            output += f"{trait}: {percentage}%\n"
-        return output
+        return result, percentages
+
+    def plot_result(self, percentages):
+        labels = list(percentages.keys())
+        sizes = list(percentages.values())
+
+        fig, ax = plt.subplots()
+        ax.barh(labels, sizes, color='skyblue')
+        ax.set_xlim(0, 100)
+        ax.set_xlabel('Percentage (%)')
+        ax.set_title('MBTI Result Distribution')
+
+        return fig
 
 def mbti_interface(*answers):
     analyzer = MBTIAnalyzer()
-    return analyzer.analyze(answers)
+    result, percentages = analyzer.analyze(answers)
+    fig = analyzer.plot_result(percentages)
+    return result, fig
 
 # Gradio 인터페이스 설정
 questions = []
 for key in ['EI', 'SN', 'TF', 'JP']:
     for q in MBTIAnalyzer().questions[key]:
-        questions.append(gr.Radio(["1", "2"], label=q))
+        questions.append(gr.Radio(["매우 그렇다", "그렇다", "보통이다", "아니다", "매우 아니다"], label=q))
 
 iface = gr.Interface(
     fn=mbti_interface,
     inputs=questions,
-    outputs="text",
+    outputs=[gr.Textbox(label="MBTI 유형 결과"), gr.Plot(label="결과 차트")],
     title="MBTI 분석기",
     description="20개 이상의 질문을 통해 당신의 MBTI 유형을 분석합니다."
 )