Update app.py

app.py

@@ -9,12 +9,7 @@ QUESTIONS = [
     "음악을 듣거나 산책을 할 때 그 속에 빠져들어 나 자신을 잊는다.",
     "종종 실제보다 일이 더 어렵거나 위험할 것이라고 예상한다.",
     "내가 할 수 있는 한 더 잘 하고 싶기 때문에, 열심을 다해 나 자신을 몰아붙인다.",
-    "주머니가 바닥날 때까지 돈을 쓰거나, 신용카드를 너무 많이 사용해서 빚을 지는 때가 있다.",
-    "내가 가지고 있는 많은 습관들은 내가 가치 있는 목표를 달성하는 데 방해가 된다.",
-    "문제가 생기면 혼자서 해결하려 한다.",
-    "미래에 어떤 일이 잘못될까봐 자주 걱정한다.",
-    "별로 힘들이지 않고 하루 종일 활동할 수 있다.",
-    # 40개 이상의 질문을 여기에 추가
+    # 추가 질문 45개
 ]
 
 # 규준 점수 데이터
@@ -30,18 +25,25 @@ STANDARD_SCORES = {
 
 # 점수 계산 함수
 def calculate_scores(responses):
-    scores = {}
-    for scale in STANDARD_SCORES.keys():
-        raw_score = sum(responses.get(scale, []))  # 원점수 합계
-        mean = STANDARD_SCORES[scale]["M"]
-        sd = STANDARD_SCORES[scale]["SD"]
+    # 질문 응답은 5점 척도(0~4)
+    # 각 척도에 대한 점수를 계산
+    scores = {"NS": 0, "HA": 0, "RD": 0, "PS": 0, "SD": 0, "CO": 0, "ST": 0}
+    for i, response in enumerate(responses):
+        scale = list(STANDARD_SCORES.keys())[i % len(STANDARD_SCORES)]  # 척도 순환
+        scores[scale] += response  # 원점수 합계
+
+    # 각 척도별 T 점수 및 백분위 계산
+    results = {}
+    for scale, raw_score in scores.items():
+        mean, sd = STANDARD_SCORES[scale]["M"], STANDARD_SCORES[scale]["SD"]
         t_score = 50 + 10 * (raw_score - mean) / sd
-        scores[scale] = {
+        percentile = round((raw_score - mean + 2 * sd) / (4 * sd) * 100, 2)
+        results[scale] = {
             "원점수": raw_score,
             "T점수": round(t_score, 2),
-            "백분위": round((raw_score - mean + 2 * sd) / (4 * sd) * 100, 2),
+            "백분위": percentile,
         }
-    return scores
+    return results
 
 # 결과 시각화
 def visualize_results(scores):
@@ -55,21 +57,22 @@ def visualize_results(scores):
     plt.xlabel("척도")
     plt.ylabel("T 점수")
     plt.legend()
+    plt.tight_layout()
     plt.savefig("results.png")
     return "results.png"
 
-# Gradio 앱 인터페이스 정의
+# Gradio 앱 함수
 def tci_app(name, gender, age, responses):
+    """
+    설문 데이터를 처리하고 결과를 반환
+    """
     scores = calculate_scores(responses)
     graph_path = visualize_results(scores)
 
     report = f"이름: {name}\n성별: {gender}\n나이: {age}\n\n"
     report += "TCI 결과:\n"
-    for scale, score_data in scores.items():
-        report += (
-            f"{scale}: 원점수={score_data['원점수']}, "
-            f"T점수={score_data['T점수']}, 백분위={score_data['백분위']}\n"
-        )
+    for scale, data in scores.items():
+        report += f"{scale}: 원점수={data['원점수']}, T점수={data['T점수']}, 백분위={data['백분위']}\n"
 
     return report, graph_path
 
@@ -80,15 +83,15 @@ inputs = [
     gr.Number(label="나이"),
 ]
 
-# 설문 항목을 라디오 버튼으로 추가
-response_inputs = []
-for i, question in enumerate(QUESTIONS):
-    response_inputs.append(
-        gr.Radio(
-            label=f"질문 {i+1}: {question}",
-            choices=["매우 아니다", "아니다", "보통이다", "그렇다", "매우 그렇다"],
-        )
+# 질문 항목에 대한 라디오 버튼 입력 생성
+response_inputs = [
+    gr.Radio(
+        label=f"질문 {i+1}: {question}",
+        choices=["매우 아니다", "아니다", "보통이다", "그렇다", "매우 그렇다"],
+        type="index",
     )
+    for i, question in enumerate(QUESTIONS)
+]
 
 inputs.extend(response_inputs)
 
@@ -99,7 +102,7 @@ outputs = [
 
 # Gradio 앱 실행
 app = gr.Interface(
-    fn=tci_app,
+    fn=lambda name, gender, age, *responses: tci_app(name, gender, age, responses),
     inputs=inputs,
     outputs=outputs,
     title="TCI 기질 분석기",