Update app.py

app.py

@@ -1,97 +1,70 @@
-import gradio as gr
-import numpy as np
-import matplotlib.pyplot as plt
-
-# 질문 데이터
-QUESTIONS = {
-    "NS": [
-        "쉬운 일보다는 도전적인 일이 더 좋다.",
-        "새로운 것을 시도하는 데 흥미를 느낀다.",
-        "변화를 즐긴다.",
-        "계획을 세우지 않고 즉흥적으로 행동하는 경향이 있다.",
-    ],
-    "HA": [
-        "종종 실제보다 일이 더 어렵거나 위험하다고 예상한다.",
-        "미래에 어떤 일이 잘못될까 봐 자주 걱정한다.",
-        "낯선 사람을 만날 때, 매우 수줍어하며 위축된다.",
-        "피로감을 쉽게 느낀다.",
-    ],
-    # 추가 질문 NS, HA, RD, PS, SD, CO, ST 구성
-}
-
-# 규준 점수 데이터
-STANDARD_SCORES = {
-    "NS": {"M": 20, "SD": 5},
-    "HA": {"M": 18, "SD": 4},
-    "RD": {"M": 22, "SD": 6},
-    "PS": {"M": 19, "SD": 4},
-    "SD": {"M": 24, "SD": 6},
-    "CO": {"M": 26, "SD": 5},
-    "ST": {"M": 18, "SD": 5},
-}
-
-# 점수 계산 함수
-def calculate_scores(responses):
-    results = {}
-    for scale, questions in QUESTIONS.items():
-        raw_score = sum(responses[scale])  # 원점수 합계
-        mean = STANDARD_SCORES[scale]["M"]
-        sd = STANDARD_SCORES[scale]["SD"]
-        t_score = 50 + 10 * (raw_score - mean) / sd
-        percentile = (raw_score - mean + 2 * sd) / (4 * sd) * 100
-        results[scale] = {
+scores[scale] = {
             "원점수": raw_score,
             "T점수": round(t_score, 2),
-            "백분위": round(percentile, 2),
+            "백분위": round((raw_score - mean + 2 * sd) / (4 * sd) * 100, 2),
         }
-    return results
+    return scores
 
 # 결과 시각화
-def visualize_results(results):
-    scales = list(results.keys())
-    t_scores = [results[scale]["T점수"] for scale in scales]
-    
-    plt.figure(figsize=(8, 4))
+def visualize_results(scores):
+    scales = scores.keys()
+    t_scores = [scores[scale]["T점수"] for scale in scales]
+
+    plt.figure(figsize=(10, 6))
     plt.bar(scales, t_scores, color="skyblue")
-    plt.axhline(50, color="gray", linestyle="--", label="평균 T점수")
-    plt.title("TCI 척도별 T점수")
+    plt.axhline(50, color="gray", linestyle="--", label="평균 T 점수")
+    plt.title("TCI 척도별 T 점수", fontsize=14)
     plt.xlabel("척도")
-    plt.ylabel("T점수")
+    plt.ylabel("T 점수")
     plt.legend()
-    plt.tight_layout()
-    
-    # 그래프를 이미지로 반환
     plt.savefig("results.png")
     return "results.png"
 
-# Gradio 인터페이스
-def tci_app(name, gender, age, **responses):
-    user_responses = {scale: [responses[f"{scale}_{i}"] for i in range(len(QUESTIONS[scale]))] for scale in QUESTIONS}
-    scores = calculate_scores(user_responses)
+# 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 = f"이름: {name}\n성별: {gender}\n나이: {age}\n\n"
     report += "TCI 결과:\n"
-    for scale, data in scores.items():
-        report += f"{scale}: 원점수={data['원점수']}, T점수={data['T점수']}, 백분위={data['백분위']}\n"
-    
+    for scale, score_data in scores.items():
+        report += (
+            f"{scale}: 원점수={score_data['원점수']}, "
+            f"T점수={score_data['T점수']}, 백분위={score_data['백분위']}\n"
+        )
+
     return report, graph_path
 
 # Gradio UI 구성
 inputs = [
     gr.Textbox(label="이름"),
     gr.Radio(label="성별", choices=["남성", "여성"]),
-    gr.Number(label="연령"),
+    gr.Number(label="나이"),
 ]
 
-for scale, questions in QUESTIONS.items():
-    for i, question in enumerate(questions):
-        inputs.append(gr.Slider(0, 4, step=1, label=f"{scale}_{i+1}: {question}"))
+# 설문 항목을 라디오 버튼으로 추가
+response_inputs = []
+for i, question in enumerate(QUESTIONS):
+    response_inputs.append(
+        gr.Radio(
+            label=f"질문 {i+1}: {question}",
+            choices=["매우 아니다", "아니다", "보통이다", "그렇다", "매우 그렇다"],
+        )
+    )
+
+inputs.extend(response_inputs)
 
 outputs = [
     gr.Textbox(label="결과 보고서", lines=10),
     gr.Image(label="결과 그래프"),
 ]
 
-app = gr.Interface(fn=tci_app, inputs=inputs, outputs=outputs, title="TCI 기질 분석기")
+# Gradio 앱 실행
+app = gr.Interface(
+    fn=tci_app,
+    inputs=inputs,
+    outputs=outputs,
+    title="TCI 기질 분석기",
+    description="TCI 설문을 기반으로 기질 및 성격 유형을 분석합니다.",
+)
 app.launch()