Update app.py

app.py

@@ -7,7 +7,7 @@ import json
 from io import BytesIO
 import base64
 
-# TCI_TYPES 데이터 (기존에 정의한 내용과 통합)
+# TCI_TYPES 데이터
 TCI_TYPES = {
     "Type1": {
         "name": "활동적인 도전자",
@@ -146,32 +146,32 @@ def calculate_tci(responses):
     return selected_type, percentages[selected_type]
 
 def generate_report(tci_data):
-    개인정보 = tci_data["개인정보"]
-    프로파일 = tci_data["TCI_RS_프로파일"]
-    하위척도 = tci_data["TCI_RS_하위척도"]
+    개인정보 = tci_data.get("개인정보", {})
+    프로파일 = tci_data.get("TCI_RS_프로파일", {})
+    하위척도 = tci_data.get("TCI_RS_하위척도", {})
     
     # 개인 정보 섹션
     personal_info = f"""
     ## 개인 정보
-    - **이름**: {개인정보['이름']}
-    - **개인 고유번호**: {개인정보['개인고유번호']}
-    - **성별**: {개인정보['성별']}
-    - **연령**: {개인정보['연령']}
-    - **소속기관 1**: {개인정보['소속기관1']}
-    - **소속기관 2**: {개인정보['소속기관2']}
-    - **규준집단**: {개인정보['규준집단']}
-    - **무응답수**: {개인정보['무응답수']}
+    - **이름**: {개인정보.get('이름', 'N/A')}
+    - **개인 고유번호**: {개인정보.get('개인고유번호', 'N/A')}
+    - **성별**: {개인정보.get('성별', 'N/A')}
+    - **연령**: {개인정보.get('연령', 'N/A')}
+    - **소속기관 1**: {개인정보.get('소속기관1', 'N/A')}
+    - **소속기관 2**: {개인정보.get('소속기관2', 'N/A')}
+    - **규준집단**: {개인정보.get('규준집단', 'N/A')}
+    - **무응답수**: {개인정보.get('무응답수', 'N/A')}
     """
 
     # 기질 섹션
     temperament = "## 기질\n"
-    for key, value in 프로파일["기질"].items():
-        temperament += f"- **{key}**: 원점수={value['원점수']}, T점수={value['T점수']}, 백분위={value['백분위']}\n"
+    for key, value in 프로파일.get("기질", {}).items():
+        temperament += f"- **{key}**: 원점수={value.get('원점수', 'N/A')}, T점수={value.get('T점수', 'N/A')}, 백분위={value.get('백분위', 'N/A')}\n"
 
     # 성격 섹션
     personality = "## 성격\n"
-    for key, value in 프로파일["성격"].items():
-        personality += f"- **{key}**: 원점수={value['원점수']}, T점수={value['T점수']}, 백분위={value['백분위']}\n"
+    for key, value in 프로파일.get("성격", {}).items():
+        personality += f"- **{key}**: 원점수={value.get('원점수', 'N/A')}, T점수={value.get('T점수', 'N/A')}, 백분위={value.get('백분위', 'N/A')}\n"
 
     # 하위척도 섹션
     subscales = "## TCI-RS 하위척도\n"
@@ -188,12 +188,12 @@ def generate_report(tci_data):
 def plot_scores(tci_data):
     scales = []
     scores = []
-    for scale, data in tci_data["TCI_RS_프로파일"]["기질"].items():
+    for scale, data in tci_data.get("TCI_RS_프로파일", {}).get("기질", {}).items():
         scales.append(scale)
-        scores.append(data["백분위"])
-    for scale, data in tci_data["TCI_RS_프로파일"]["성격"].items():
+        scores.append(data.get("백분위", 0))
+    for scale, data in tci_data.get("TCI_RS_프로파일", {}).get("성격", {}).items():
         scales.append(scale)
-        scores.append(data["백분위"])
+        scores.append(data.get("백분위", 0))
     
     plt.figure(figsize=(12, 6))
     bars = plt.bar(scales, scores, color='skyblue')
@@ -239,83 +239,13 @@ def display_profile(tci_data_json):
 def main():
     with gr.Blocks() as demo:
         gr.Markdown("# TCI-RS 프로파일 보고서")
-        gr.Markdown("아래에 TCI-RS 프로파일 데이터를 JSON 형식으로 입력해주세요.\n\n**예시 JSON 데이터**\n```json\n" + json.dumps({
-            "개인정보": {
-                "이름": "홍길동",
-                "개인고유번호": 123,
-                "성별": "남자",
-                "연령": "만 20 세",
-                "소속기관1": "마음사랑",
-                "소속기관2": "",
-                "규준집단": "일반성인",
-                "무응답수": 0
-            },
-            "TCI_RS_프로파일": {
-                "기질": {
-                    "자극추구(NS)": {"원점수": 34, "T점수": 57, "백분위": 74},
-                    "위험회피(HA)": {"원점수": 40, "T점수": 55, "백분위": 66},
-                    "사회적 민감성(RD)": {"원점수": 41, "T점수": 48, "백분위": 39},
-                    "인내력(PS)": {"원점수": 32, "T점수": 38, "백분위": 11}
-                },
-                "성격": {
-                    "자율성(SD)": {"원점수": 47, "T점수": 49, "백분위": 44},
-                    "연대감(CO)": {"원점수": 43, "T점수": 35, "백분위": 5},
-                    "자기초월(ST)": {"원점수": 27, "T점수": 51, "백분위": 56},
-                    "자율성+연대감(SC)": {"원점수": 90, "T점수": 41, "백분위": 18}
-                }
-            },
-            "TCI_RS_하위척도": {
-                "자극추구(NS)": {
-                    "NS1": {"원점수": 9, "규준집단_M": 9.5, "SD": 3.2, "특성": "관습적안정성 탐색적흥분"},
-                    "NS2": {"원점수": 7, "규준집단_M": 7.0, "SD": 3.3, "특성": "심사숙고 충동성"},
-                    "NS3": {"원점수": 9, "규준집단_M": 6.0, "SD": 3.2, "특성": "절제 무절제"},
-                    "NS4": {"원점수": 9, "규준집단_M": 5.2, "SD": 3.2, "특성": "질서정연 자유분방"}
-                },
-                "위험회피(HA)": {
-                    "HA1": {"원점수": 8, "규준집단_M": 7.8, "SD": 4.1, "특성": "낙천성 예기불안"},
-                    "HA2": {"원점수": 9, "규준집단_M": 9.9, "SD": 3.0, "특성": "(낮은)불확실성에대한두려움 (높은)불확실성에대한두려움"},
-                    "HA3": {"원점수": 12, "규준집단_M": 8.7, "SD": 3.6, "특성": "(낮은)낯선사람에대한수줍음 (높은)낯선사람에대한수줍음"},
-                    "HA4": {"원점수": 11, "규준집단_M": 8.7, "SD": 3.4, "특성": "활기넘침 쉽게지침"}
-                },
-                "사회적 민감성(RD)": {
-                    "RD1": {"원점수": 8, "규준집단_M": 11.1, "SD": 2.9, "특성": "(낮은)정서적감수성 (높은)정서적감수성"},
-                    "RD2": {"원점수": 10, "규준집단_M": 10.5, "SD": 3.0, "특성": "(낮은)정서적개방성 (높은)정서적개방성"},
-                    "RD3": {"원점수": 13, "규준집단_M": 11.6, "SD": 3.3, "특성": "거리두기 친밀감"},
-                    "RD4": {"원점수": 10, "규준집단_M": 9.4, "SD": 2.6, "특성": "독립 의존"}
-                },
-                "인내력(PS)": {
-                    "PS1": {"원점수": 12, "규준집단_M": 12.7, "SD": 3.0, "특성": "(낮은)근면 (높은)근면"},
-                    "PS2": {"원점수": 6, "규준집단_M": 10.7, "SD": 3.0, "특성": "(낮은)끈기 (높은)끈기"},
-                    "PS3": {"원점수": 7, "규준집단_M": 10.4, "SD": 3.7, "특성": "(낮은)성취에대한야망 (높은)성취에대한야망"},
-                    "PS4": {"원점수": 7, "규준집단_M": 9.9, "SD": 3.5, "특성": "(낮은)완벽주의 (높은)완벽주의"}
-                },
-                "자율성(SD)": {
-                    "SD1": {"원점수": 12, "규준집단_M": 12.7, "SD": 2.9, "특성": "(낮은)책임감 (높은)책임감"},
-                    "SD2": {"원점수": 12, "규준집단_M": 11.6, "SD": 3.1, "특성": "(낮은)목적의식 (높은)목적의식"},
-                    "SD3": {"원점수": 6, "규준집단_M": 6.8, "SD": 1.9, "특성": "(낮은)유능감 (높은)유능감"},
-                    "SD4": {"원점수": 5, "규준집단_M": 4.2, "SD": 1.7, "특성": "(낮은)자기수용 (높은)자기수용"},
-                    "SD5": {"원점수": 12, "규준집단_M": 12.5, "SD": 3.2, "특성": "(낮은)자기일치 (높은)자기일치"}
-                },
-                "연대감(CO)": {
-                    "CO1": {"원점수": 11, "규준집단_M": 12.6, "SD": 2.8, "특성": "(낮은)타인수용 (높은)타인수용"},
-                    "CO2": {"원점수": 6, "규준집단_M": 9.6, "SD": 2.5, "특성": "(낮은)공감 (높은)공감"},
-                    "CO3": {"원점수": 7, "규준집단_M": 9.9, "SD": 2.5, "특성": "(낮은)이타성 (높은)이타성"},
-                    "CO4": {"원점수": 9, "규준집단_M": 8.9, "SD": 2.6, "특성": "(낮은)관대함 (높은)관대함"},
-                    "CO5": {"원점수": 10, "규준집단_M": 15.1, "SD": 2.5, "특성": "(낮은)공평 (높은)공평"}
-                },
-                "자기초월(ST)": {
-                    "ST1": {"원점수": 10, "규준집단_M": 9.1, "SD": 4.0, "특성": "자의식 창조적자기망각"},
-                    "ST2": {"원점수": 8, "규준집단_M": 7.4, "SD": 4.1, "특성": "(낮은)우주만물과의일체감 (높은)우주만물과의일체감"},
-                    "ST3": {"원점수": 9, "규준집단_M": 9.2, "SD": 5.4, "특성": "합리적유물론 영성수용"}
-                }
-            }
-        }, ensure_ascii=False, indent=4) + "\n```")
-        
-        input_json = gr.Textbox(
-            label="TCI-RS 데이터 (JSON)",
-            lines=15,
-            placeholder="TCI-RS 데이터를 JSON 형식으로 입력해주세요.",
-            value=json.dumps({
+        gr.Markdown(
+            """
+            아래에 TCI-RS 프로파일 데이터를 JSON 형식으로 입력해주세요.
+
+            **예시 JSON 데이터**
+            ```json
+            {
                 "개인정보": {
                     "이름": "홍길동",
                     "개인고유번호": 123,
@@ -386,7 +316,9 @@ def main():
                     }
                 }
             }
-        }, ensure_ascii=False, indent=4) + "\n```")
+            ```
+            """
+        )
         
         input_json = gr.Textbox(
             label="TCI-RS 데이터 (JSON)",
@@ -462,8 +394,7 @@ def main():
                         "ST3": {"원점수": 9, "규준집단_M": 9.2, "SD": 5.4, "특성": "합리적유물론 영성수용"}
                     }
                 }
-            }
-        }, ensure_ascii=False, indent=4))
+            }, ensure_ascii=False, indent=4))
         
         submit = gr.Button("보고서 생성")
         
@@ -472,8 +403,5 @@ def main():
         
         submit.click(fn=display_profile, inputs=input_json, outputs=[report_output, image_output])
     
-    demo.launch()
-
-if __name__ == "__main__":
-    main()
-
+    if __name__ == "__main__":
+        main()