import gradio as gr

# MBTI 유형별 해석 및 추천 직업
mbti_interpretations = {
    "ISTJ": ("실용적이고 현실적인 당신은 조직과 질서를 중시합니다.", "회계사, 변호사"),
    "ISFJ": ("세심하고 따뜻한 당신은 타인의 감정을 잘 이해합니다.", "간호사, 사회복지사"),
    "INFJ": ("직관적이고 섬세한 당신은 이상적인 세상을 꿈꿉니다.", "심리학자, 상담사"),
    "INTJ": ("전략적이고 독립적인 당신은 복잡한 문제를 해결하는 데 능숙합니다.", "소프트웨어 개발자, 엔지니어"),
    "ISTP": ("논리적이고 탐험적인 당신은 실용적인 해결책을 선호합니다.", "기계공학자, 조종사"),
    "ISFP": ("예술적이고 호기심 많은 당신은 다양한 경험을 즐깁니다.", "그래픽 디자이너, 음악가"),
    "INFP": ("이상주의적이고 충성심 깊은 당신은 자신만의 가치관을 중요시합니다.", "작가, 예술가"),
    "INTP": ("분석적이고 호기심 많은 당신은 지식을 탐구하는 것을 좋아합니다.", "과학자, 프로그래머"),
    "ESTP": ("활동적이고 사교적인 당신은 순간을 즐기며 삽니다.", "마케팅 전문가, 판매원"),
    "ESFP": ("사교적이고 활발한 당신은 주변 사람들과 즐거운 시간을 보냅니다.", "이벤트 플래너, 연예인"),
    "ENFP": ("열정적이고 창의적인 당신은 새로운 가능성을 탐색합니다.", "창업가, 마케팅 전략가"),
    "ENTP": ("빠르게 생각하고 도전적인 당신은 토론을 즐깁니다.", "발명가, 컨설턴트"),
    "ESTJ": ("결단력 있고 체계적인 당신은 리더십을 발휘합니다.", "경영자, 관리자"),
    "ESFJ": ("친절하고 조화를 중시하는 당신은 타인을 돕는 것을 좋아합니다.", "교사, 고객 서비스 담당자"),
    "ENFJ": ("카리스마 있고 영감을 주는 당신은 타인을 격려하고 동기를 부여합니다.", "인사 관리자, 교육가"),
    "ENTJ": ("대담하고 야심 찬 당신은 목표를 향해 나아갑니다.", "기업가, 법률가")
}

# 사용자 응답에 따른 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_type = "".join([max(pair, key=lambda x: scores[x]) for pair in axis])
    interpretation, recommended_jobs = mbti_interpretations[mbti_type]
    return f"당신의 MBTI 유형은 {mbti_type}입니다. {interpretation} 추천 직업: {recommended_jobs}"

# Gradio 인터페이스 구성
iface = gr.Interface(
    fn=calculate_mbti,
    inputs=[gr.Radio(["예", "아니오"]) for _ in range(4)],
    outputs="text",
    title="MBTI 유형 진단 및 추천 직업",
    description="4개의 질문에 답하여 당신의 MBTI 유형과 추천 직업을 알아보세요."
)

# 애플리케이션 실행
iface.launch()