Update app.py

app.py

@@ -1,77 +1,48 @@
 import gradio as gr
 
-# MBTI 유형별 해석
+# MBTI 유형별 해석 및 추천 직업
 mbti_interpretations = {
-    "ISTJ": "실용적이고 현실적인 당신은 조직과 질서를 중시합니다.",
-    "ISFJ": "세심하고 따뜻한 당신은 타인의 감정을 잘 이해합니다.",
-    "INFJ": "직관적이고 섬세한 당신은 이상적인 세상을 꿈꿉니다.",
-    "INTJ": "전략적이고 독립적인 당신은 복잡한 문제를 해결하는 데 능숙합니다.",
-    "ISTP": "논리적이고 탐험적인 당신은 실용적인 해결책을 선호합니다.",
-    "ISFP": "예술적이고 호기심 많은 당신은 다양한 경험을 즐깁니다.",
-    "INFP": "이상주의적이고 충성심 깊은 당신은 자신만의 가치관을 중요시합니다.",
-    "INTP": "분석적이고 호기심 많은 당신은 지식을 탐구하는 것을 좋아합니다.",
-    "ESTP": "활동적이고 사교적인 당신은 순간을 즐기며 삽니다.",
-    "ESFP": "사교적이고 활발한 당신은 주변 사람들과 즐거운 시간을 보냅니다.",
-    "ENFP": "열정적이고 창의적인 당신은 새로운 가능성을 탐색합니다.",
-    "ENTP": "빠르게 생각하고 도전적인 당신은 토론을 즐깁니다.",
-    "ESTJ": "결단력 있고 체계적인 당신은 리더십을 발휘합니다.",
-    "ESFJ": "친절하고 조화를 중시하는 당신은 타인을 돕는 것을 좋아합니다.",
-    "ENFJ": "카리스마 있고 영감을 주는 당신은 타인을 격려하고 동기를 부여합니다.",
-    "ENTJ": "대담하고 야심 찬 당신은 목표를 향해 나아갑니다."
+    "ISTJ": ("실용적이고 현실적인 당신은 조직과 질서를 중시합니다.", "회계사, 변호사"),
+    "ISFJ": ("세심하고 따뜻한 당신은 타인의 감정을 잘 이해합니다.", "간호사, 사회복지사"),
+    "INFJ": ("직관적이고 섬세한 당신은 이상적인 세상을 꿈꿉니다.", "심리학자, 상담사"),
+    "INTJ": ("전략적이고 독립적인 당신은 복잡한 문제를 해결하는 데 능숙합니다.", "소프트웨어 개발자, 엔지니어"),
+    "ISTP": ("논리적이고 탐험적인 당신은 실용적인 해결책을 선호합니다.", "기계공학자, 조종사"),
+    "ISFP": ("예술적이고 호기심 많은 당신은 다양한 경험을 즐깁니다.", "그래픽 디자이너, 음악가"),
+    "INFP": ("이상주의적이고 충성심 깊은 당신은 자신만의 가치관을 중요시합니다.", "작가, 예술가"),
+    "INTP": ("분석적이고 호기심 많은 당신은 지식을 탐구하는 것을 좋아합니다.", "과학자, 프로그래머"),
+    "ESTP": ("활동적이고 사교적인 당신은 순간을 즐기며 삽니다.", "마케팅 전문가, 판매원"),
+    "ESFP": ("사교적이고 활발한 당신은 주변 사람들과 즐거운 시간을 보냅니다.", "이벤트 플래너, 연예인"),
+    "ENFP": ("열정적이고 창의적인 당신은 새로운 가능성을 탐색합니다.", "창업가, 마케팅 전략가"),
+    "ENTP": ("빠르게 생각하고 도전적인 당신은 토론을 즐깁니다.", "발명가, 컨설턴트"),
+    "ESTJ": ("결단력 있고 체계적인 당신은 리더십을 발휘합니다.", "경영자, 관리자"),
+    "ESFJ": ("친절하고 조화를 중시하는 당신은 타인을 돕는 것을 좋아합니다.", "교사, 고객 서비스 담당자"),
+    "ENFJ": ("카리스마 있고 영감을 주는 당신은 타인을 격려하고 동기를 부여합니다.", "인사 관리자, 교육가"),
+    "ENTJ": ("대담하고 야심 찬 당신은 목표를 향해 나아갑니다.", "기업가, 법률가")
 }
 
-# 질문과 선택지
-questions = [
-    "새로운 사람들과 쉽게 친해지나요?",
-    "계획을 세우지 않고 여행을 가는 것을 선호하나요?",
-    "세부 사항보다는 전체적인 그림을 보는 것을 선호하나요?",
-    "감정보다는 사실에 기반하여 결정을 내리나요?",
-    "일을 마감 기한 직전에 하는 것보다 미리 준비하는 것을 선호하나요?",
-    "실제 경험을 통해 배우는 것을 이론적 학습보다 선호하나요?",
-    "타인의 감정을 자신의 것처럼 느끼는 편인가요?",
-    "목록이나 계획을 작성하는 것을 좋아하나요?",
-    "새로운 아이디어보다는 검증된 아이디어에 더 관심이 있나요?",
-    "토론할 때 감정보다는 논리를 중시하나요?"
-]
-
-choices = [
-    ["예", "아니오"],
-    ["예", "아니오"],
-    ["예", "아니오"],
-    ["예", "아니오"],
-    ["예", "아니오"],
-    ["예", "아니오"],
-    ["예", "아니오"],
-    ["예", "아니오"],
-    ["예", "아니오"],
-    ["예", "아니오"]
-]
-
-# 사용자 응답에 따른 MBTI 유형 계산
+# 사용자 응답에 따른 MBTI 유형 및 추천 직업 계산 함수
 def calculate_mbti(*answers):
-    # 각 축에 대한 점수를 계산합니다.
     scores = {"E": 0, "I": 0, "S": 0, "N": 0, "T": 0, "F": 0, "J": 0, "P": 0}
     axis = [("E", "I"), ("S", "N"), ("T", "F"), ("J", "P")]
     
-    # 사용자의 응답에 따라 점수를 할당합니다.
     for answer, (yes, no) in zip(answers, axis):
         if answer == "예":
             scores[yes] += 1
         else:
             scores[no] += 1
     
-    # MBTI 유형을 결정합니다.
     mbti_type = "".join([max(pair, key=lambda x: scores[x]) for pair in axis])
-    interpretation = mbti_interpretations[mbti_type]
-    return f"당신의 MBTI 유형은 {mbti_type}입니다. {interpretation}"
+    interpretation, recommended_jobs = mbti_interpretations[mbti_type]
+    return f"당신의 MBTI 유형은 {mbti_type}입니다. {interpretation} 추천 직업: {recommended_jobs}"
 
-# Gradio 앱 구성
-with gr.Blocks() as app:
-    with gr.Row():
-        inputs = [gr.Radio(choices=choice, label=question) for question, choice in zip(questions, choices)]
-    submit_button = gr.Button("MBTI 유형 진단하기")
-    output = gr.Textbox(label="진단 결과")
-    submit_button.click(calculate_mbti, inputs=inputs, outputs=output)
+# Gradio 인터페이스 구성
+iface = gr.Interface(
+    fn=calculate_mbti,
+    inputs=[gr.Radio(["예", "아니오"]) for _ in range(4)],
+    outputs="text",
+    title="MBTI 유형 진단 및 추천 직업",
+    description="4개의 질문에 답하여 당신의 MBTI 유형과 추천 직업을 알아보세요."
+)
 
-# 앱 실행
-app.launch()
+# 애플리케이션 실행
+iface.launch()