import gradio as gr

def mbti_test(q1, q2, q3, q4, q5, q6, q7, q8, q9, q10):

    # 각 지표별 점수 초기화 (0점부터 시작)
    e_score = 0
    n_score = 0
    f_score = 0
    p_score = 0

    # [질문 1~10] 선택지에 따른 점수 합산 (단순 if문 구조)

    # E(외향) vs I(내향)
    if q1 == "사람들과 함께 있을 때 에너지가 생긴다": e_score = e_score + 1
    if q2 == "먼저 말을 거는 편이다": e_score = e_score + 1
    if q3 == "주말에는 밖으로 나가는 게 좋다": e_score = e_score + 1

    # S(감각) vs N(직관)
    if q4 == "상상을 자주 하는 편이다": n_score = n_score + 1
    if q5 == "영화 볼 때 숨겨진 의미를 찾으려 한다": n_score = n_score + 1

    # T(사고) vs F(감정)
    if q6 == "고민 상담을 들으면 해결책부터 생각난다": f_score = f_score - 1 # T면 점수 깎기
    else: f_score = f_score + 1 # F면 점수 더하기
    if q7 == "리액션보다는 내용이 중요하다": f_score = f_score - 1
    else: f_score = f_score + 1
    if q8 == "친구가 우울해서 화분 샀다고 하면 '왜 우울해?'라고 묻는다": f_score = f_score + 1

    # J(판단) vs P(계획)
    if q9 == "여행 갈 때 시간 단위로 계획을 짠다": p_score = p_score - 1
    else: p_score = p_score + 1
    if q10 == "벼락치기보다는 미리미리 하는 편이다": p_score = p_score - 1
    else: p_score = p_score + 1

    # 최종 MBTI 글자 조합 (if문을 이용한 판정)
    result = ""

    # E/I 판정
    if e_score >= 2: result = result + "E"
    else: result = result + "I"

    # S/N 판정
    if n_score >= 1: result = result + "N"
    else: result = result + "S"

    # T/F 판정
    if f_score >= 2: result = result + "F"
    else: result = result + "T"

    # J/P 판정
    if p_score >= 1: result = result + "P"
    else: result = result + "J"

​​​​​​​    return f"당신의 MBTI 결과는 [{result}] 입니다!"

# --- 웹 화면 구성 (학생들이 수정하기 쉬운 부분) ---

with gr.Blocks() as demo:
    gr.Markdown("# 🔍 초간단 10문항 MBTI 테스트")
    gr.Markdown("질문에 답하고 아래의 '결과 보기' 버튼을 눌러주세요.")

    # 질문지 구성 (Radio 버튼 사용)
    q1 = gr.Radio(["사람들과 함께 있을 때 에너지가 생긴다", "혼자 있을 때 에너지가 충전된다"], label="1. 당신은 에너지를 어디서 얻나요?")
    q2 = gr.Radio(["먼저 말을 거는 편이다", "상대방이 말을 걸 때까지 기다린다"], label="2. 처음 본 사람에게?")
    q3 = gr.Radio(["주말에는 밖으로 나가는 게 좋다", "집에서 쉬는 게 제일 좋다"], label="3. 쉬는 날에는?")
    q4 = gr.Radio(["현실적인 사실이 중요하다", "상상을 자주 하는 편이다"], label="4. 당신의 생각 스타일은?")
    q5 = gr.Radio(["보이는 그대로 믿는다", "영화 볼 때 숨겨진 의미를 찾으려 한다"], label="5. 사물을 볼 때?")
    q6 = gr.Radio(["해결책 제시", "공감과 위로"], label="6. 친구가 힘들다고 할 때 당신은?")
    q7 = gr.Radio(["리액션보다는 내용이 중요하다", "내용보다는 말투와 반응이 중요하다"], label="7. 대화할 때 더 중요한 것은?")
    q8 = gr.Radio(["'무슨 화분 샀어?' (T)", "'왜 우울해?' (F)"], label="8. 친구가 우울해서 화분 샀다고 하면?")
    q9 = gr.Radio(["시간 단위로 계획을 짠다", "발길 닿는 대로 움직인다"], label="9. 여행 갈 때 스타일은?")
    q10 = gr.Radio(["미리미리 과제를 한다", "마지막에 몰아서 벼락치기를 한다"], label="10. 할 일이 생겼을 때?")

    btn = gr.Button("결과 확인하기", variant="primary")
    output = gr.Textbox(label="테스트 결과")

    # 버튼 클릭 시 함수 호출
    btn.click(
        fn=mbti_test, 
        inputs=[q1, q2, q3, q4, q5, q6, q7, q8, q9, q10], 
        outputs=output
    )

# share=True로 설정하면 외부 링크(gradio.live)가 생성됩니다.
demo.launch(share=True)