import streamlit as st

def main():
    st.title("Психологический тест: Определение типа личности")
    st.write("Отвечайте на вопросы, выбирая вариант A или B. В конце вы получите ваш тип личности.")

    questions = [
        "Как вы предпочитаете проводить свободное время?\nA. В одиночестве или с близкими друзьями\nB. В компании людей, знакомясь с новыми",
        "Как вы принимаете решения?\nA. Логически анализируя\nB. Ориентируясь на чувства и эмоции",
        "Как вы решаете проблемы?\nA. Предпочитаю действовать по плану\nB. Часто импровизирую",
        "Вы чаще полагаетесь на:\nA. Практическое мышление и опыт\nB. Воображение и интуицию",
        "Вы предпочитаете:\nA. Конкретные факты и детали\nB. Общую картину и идеи",
        "Что для вас важнее в отношениях?\nA. Доверие и предсказуемость\nB. Новизна и эмоции",
        "Как вы предпочитаете организовать свою работу?\nA. Следовать заранее установленному расписанию\nB. Работать в свободном режиме, когда есть вдохновение",
        "Как вы ведете себя в незнакомой ситуации?\nA. Предпочитаю сначала наблюдать и анализировать\nB. Активно включаюсь и пробую что-то новое",
    ]

    results = {"E": 0, "I": 0, "S": 0, "N": 0, "T": 0, "F": 0, "J": 0, "P": 0}

    for i, question in enumerate(questions, start=1):
        st.subheader(f"Вопрос {i}")
        st.write(question)
        answer = st.radio(f"Ваш выбор для вопроса {i}", ("A", "B"), key=f"q{i}")
        if answer == "A":
            if i in [1, 8]:
                results["I"] += 1
            elif i in [2]:
                results["T"] += 1
            elif i in [3, 6, 7]:
                results["J"] += 1
            elif i in [4, 5]:
                results["S"] += 1
        elif answer == "B":
            if i in [1, 8]:
                results["E"] += 1
            elif i in [2]:
                results["F"] += 1
            elif i in [3, 6, 7]:
                results["P"] += 1
            elif i in [4, 5]:
                results["N"] += 1

    if st.button("Показать результат"):
        personality_type = ""
        personality_type += "I" if results["I"] > results["E"] else "E"
        personality_type += "S" if results["S"] > results["N"] else "N"
        personality_type += "T" if results["T"] > results["F"] else "F"
        personality_type += "J" if results["J"] > results["P"] else "P"

        st.subheader(f"Ваш тип личности: {personality_type}")
        descriptions = {
            "INFJ": "Идеалист и мечтатель. Любите помогать другим, планируете свои действия.",
            "ENFP": "Творческий и энергичный. Вы легко находите общий язык с людьми.",
            "INTJ": "Стратег и аналитик. Вы ставите долгосрочные цели и ищете пути их достижения.",
            "ENTP": "Изобретатель и новатор. Любите исследовать и находить нестандартные решения.",
            "ISTJ": "Организованный и ответственный. Цените факты, стабильность и порядок.",
            "ESTJ": "Лидер и исполнитель. Вы прирожденный организатор и сторонник правил.",
            "ISFJ": "Защитник и поддержка. Заботитесь о других и стремитесь к гармонии.",
            "ESFJ": "Добрый и общительный. Цените традиции и заботитесь о благополучии окружающих.",
            "ISTP": "Практичный и независимый. Предпочитаете действовать, а не говорить.",
            "ESTP": "Энергичный и решительный. Любите приключения и быстро адаптируетесь.",
            "ISFP": "Творческий и чувствительный. Любите свободу и природу.",
            "ESFP": "Веселый и открытый. Живете в моменте и заряжаете других энергией.",
            "INTP": "Философ и исследователь. Вас привлекают идеи и сложные теории.",
            "ENTJ": "Командир и организатор. Вы нацелены на результат и стратегическое планирование.",
            "INFP": "Мечтатель и гуманист. Вас вдохновляют идеи и человеческие ценности.",
            "ENFJ": "Наставник и вдохновитель. Легко находите общий язык с людьми и поддерживаете их.",
        }
        description = descriptions.get(personality_type, "Уникальная личность, которой нет в базе данных.")
        st.write(f"Описание: {description}")

if __name__ == "__main__":
    main()