app.py

import gradio as gr
import numpy as np

# MBTI 질문 데이터
questions = [
    "나는 사람들과 시간을 보내는 것을 즐긴다.",
    "나는 구체적이고 실용적인 정보를 선호한다.",
    "나는 논리적인 결정을 내리는 것을 중요하게 생각한다.",
    "나는 체계적이고 계획적인 삶을 좋아한다.",
    "나는 새로운 아이디어를 탐구하는 것을 즐긴다.",
    # 추가 질문 작성 (총 20개 이상)
]

# 유형별 설명 데이터
mbti_descriptions = {
    "INTJ": {
        "성격": "독립적이고 전략적인 사고를 가진 성향.",
        "추천 직업": ["데이터 분석가", "소프트웨어 개발자", "연구원"],
        "궁합": {
            "좋은": ("ENTP", "혁신적이고 유연한 사고가 잘 맞음."),
            "나쁜": ("ISFP", "감정과 논리의 차이가 충돌할 가능성.")
        }
    },
    # 다른 유형 데이터 추가
}

# 선택지를 5단계로 설정
options = ["매우 아니다", "아니다", "보통이다", "그렇다", "매우 그렇다"]

def calculate_mbti(answers):
    scores = {"E/I": 0, "S/N": 0, "T/F": 0, "J/P": 0}
    for i, answer in enumerate(answers):
        if i % 4 == 0:
            scores["E/I"] += (answer - 3)
        elif i % 4 == 1:
            scores["S/N"] += (answer - 3)
        elif i % 4 == 2:
            scores["T/F"] += (answer - 3)
        elif i % 4 == 3:
            scores["J/P"] += (answer - 3)
    result = (
        ("I" if scores["E/I"] < 0 else "E") +
        ("N" if scores["S/N"] < 0 else "S") +
        ("F" if scores["T/F"] < 0 else "T") +
        ("P" if scores["J/P"] < 0 else "J")
    )
    percentages = {
        "E/I": abs(scores["E/I"] / 20 * 100),
        "S/N": abs(scores["S/N"] / 20 * 100),
        "T/F": abs(scores["T/F"] / 20 * 100),
        "J/P": abs(scores["J/P"] / 20 * 100)
    }
    return result, percentages

def analyze_mbti(answers):
    result, percentages = calculate_mbti(answers)
    description = mbti_descriptions.get(result, {})
    성격 = description.get("성격", "데이터가 없습니다.")
    추천직업 = description.get("추천 직업", [])
    궁합 = description.get("궁합", {})
    return {
        "MBTI 유형": result,
        "성격": 성격,
        "추천 직업": 추천직업,
        "궁합": 궁합,
        "세부 비율": percentages
    }

# Gradio 인터페이스
def mbti_interface(*answers):
    answers = list(answers)
    result = analyze_mbti(answers)
    return result

# 질문과 선택지를 기반으로 Gradio UI 생성
inputs = [gr.Radio(options, label=q) for q in questions]
outputs = gr.JSON()

iface = gr.Interface(
    fn=mbti_interface,
    inputs=inputs,
    outputs=outputs,
    title="MBTI 분석기",
    description="질문에 답변하고 자신의 MBTI 유형을 분석해 보세요!"
)

if __name__ == "__main__":
    iface.launch()