import gradio as gr

# MBTI 유형별 해석
mbti_interpretations = {
    "ISTJ": "실용적이고 현실적인 당신은 조직과 질서를 중시합니다.",
    "ISFJ": "세심하고 따뜻한 당신은 타인의 감정을 잘 이해합니다.",
    "INFJ": "직관적이고 섬세한 당신은 이상적인 세상을 꿈꿉니다.",
    "INTJ": "전략적이고 독립적인 당신은 복잡한 문제를 해결하는 데 능숙합니다.",
    "ISTP": "논리적이고 탐험적인 당신은 실용적인 해결책을 선호합니다.",
    "ISFP": "예술적이고 호기심 많은 당신은 다양한 경험을 즐깁니다.",
    "INFP": "이상주의적이고 충성심 깊은 당신은 자신만의 가치관을 중요시합니다.",
    "INTP": "분석적이고 호기심 많은 당신은 지식을 탐구하는 것을 좋아합니다.",
    "ESTP": "활동적이고 사교적인 당신은 순간을 즐기며 삽니다.",
    "ESFP": "사교적이고 활발한 당신은 주변 사람들과 즐거운 시간을 보냅니다.",
    "ENFP": "열정적이고 창의적인 당신은 새로운 가능성을 탐색합니다.",
    "ENTP": "빠르게 생각하고 도전적인 당신은 토론을 즐깁니다.",
    "ESTJ": "결단력 있고 체계적인 당신은 리더십을 발휘합니다.",
    "ESFJ": "친절하고 조화를 중시하는 당신은 타인을 돕는 것을 좋아합니다.",
    "ENFJ": "카리스마 있고 영감을 주는 당신은 타인을 격려하고 동기를 부여합니다.",
    "ENTJ": "대담하고 야심 찬 당신은 목표를 향해 나아갑니다."
}

# 질문과 선택지
questions = [
    "새로운 사람들과 쉽게 친해지나요?",
    "계획을 세우지 않고 여행을 가는 것을 선호하나요?",
    "세부 사항보다는 전체적인 그림을 보는 것을 선호하나요?",
    "감정보다는 사실에 기반하여 결정을 내리나요?",
    "일을 마감 기한 직전에 하는 것보다 미리 준비하는 것을 선호하나요?",
    "실제 경험을 통해 배우는 것을 이론적 학습보다 선호하나요?",
    "타인의 감정을 자신의 것처럼 느끼는 편인가요?",
    "목록이나 계획을 작성하는 것을 좋아하나요?",
    "새로운 아이디어보다는 검증된 아이디어에 더 관심이 있나요?",
    "토론할 때 감정보다는 논리를 중시하나요?"
]

choices = [
    ["예", "아니오"],
    ["예", "아니오"],
    ["예", "아니오"],
    ["예", "아니오"],
    ["예", "아니오"],
    ["예", "아니오"],
    ["예", "아니오"],
    ["예", "아니오"],
    ["예", "아니오"],
    ["예", "아니오"]
]

# 사용자 응답에 따른 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}"

# 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)

# 앱 실행
app.launch()