Create app.py

app.py

@@ -0,0 +1,130 @@
+import streamlit as st
+import openai
+from dotenv import load_dotenv
+import os
+
+# Load the OpenAI API Key
+api_key = st.text_input('Enter your OpenAI API Key', type="password")
+
+# Set the OpenAI API key
+if api_key:
+    openai.api_key = api_key
+
+# English-translated version of the questions (MBTI-related questions)
+questions = [
+    {"text": "Do you enjoy being spontaneous and keeping your options open?", "trait": "P"},
+    {"text": "Do you prefer spending weekends quietly at home rather than going out?", "trait": "I"},
+    {"text": "Do you feel more energized when you are around people?", "trait": "E"},
+    {"text": "Do you easily set and meet deadlines?", "trait": "J"},
+    {"text": "Are your decisions often influenced by how they will affect others emotionally?", "trait": "F"},
+    {"text": "Do you like discussing symbolic or metaphorical interpretations of a story?", "trait": "N"},
+    {"text": "Do you strive to maintain harmony in group settings, even if it means compromising?", "trait": "F"},
+    {"text": "When a friend is upset, is your first instinct to offer emotional support rather than solutions?", "trait": "F"},
+    {"text": "In arguments, do you focus more on being rational than on people's feelings?", "trait": "T"},
+    {"text": "When you learn something new, do you prefer hands-on experience over theory?", "trait": "S"},
+    {"text": "Do you often think about how today's actions will affect the future?", "trait": "N"},
+    {"text": "Are you comfortable adapting to new situations as they happen?", "trait": "P"},
+    {"text": "Do you prefer exploring different options before making a decision?", "trait": "P"},
+    {"text": "At parties, do you start conversations with new people?", "trait": "E"},
+    {"text": "When faced with a problem, do you prefer discussing it with others?", "trait": "E"},
+    {"text": "When making decisions, do you prioritize logic over personal considerations?", "trait": "T"},
+    {"text": "Do you find solitude more refreshing than social gatherings?", "trait": "I"},
+    {"text": "Do you prefer having a clear plan and dislike unexpected changes?", "trait": "J"},
+    {"text": "Do you find satisfaction in finishing tasks and making final decisions?", "trait": "J"},
+    {"text": "Do you tend to process your thoughts internally before speaking?", "trait": "I"},
+    {"text": "Are you more interested in exploring abstract theories and future possibilities?", "trait": "N"},
+    {"text": "When planning a vacation, do you prefer to have a detailed plan?", "trait": "S"},
+    {"text": "Do you often rely on objective criteria to assess situations?", "trait": "T"},
+    {"text": "Do you focus more on details and facts in your surroundings?", "trait": "S"}
+]
+
+# Function to calculate MBTI scores based on responses
+def calculate_weighted_mbti_scores(responses):
+    weights = {
+        "Strongly Agree": 2,
+        "Agree": 1,
+        "Neutral": 0,
+        "Disagree": -1,
+        "Strongly Disagree": -2
+    }
+
+    scores = {'E': 0, 'I': 0, 'S': 0, 'N': 0, 'T': 0, 'F': 0, 'J': 0, 'P': 0}
+
+    for i, response in enumerate(responses):
+        weight = weights.get(response, 0)
+        trait = questions[i]["trait"]
+        if trait in scores:
+            scores[trait] += weight
+
+    return scores
+
+# Function to determine MBTI type based on weighted scores
+def classic_mbti_weighted(responses):
+    scores = calculate_weighted_mbti_scores(responses)
+    mbti_type = ""
+    for trait_pair in ['EI', 'SN', 'TF', 'JP']:
+        trait1, trait2 = trait_pair
+        if scores[trait1] >= scores[trait2]:
+            mbti_type += trait1
+        else:
+            mbti_type += trait2
+    return mbti_type
+
+# Streamlit component to display the quiz and handle responses
+def show_mbti_quiz():
+    st.title('FlexTemp Personality Test')
+
+    # Step 1: Input name
+    participant_name = st.text_input("Enter your name")
+
+    if participant_name:
+        responses = []
+        st.subheader(f"Hello {participant_name}, let's start the quiz!")
+
+        for i, question in enumerate(questions):
+            response = st.radio(
+                question["text"],
+                ["Strongly Agree", "Agree", "Neutral", "Disagree", "Strongly Disagree"]
+            )
+            if response:
+                responses.append(response)
+
+        if len(responses) == len(questions):
+            st.subheader("Your MBTI Personality Type:")
+            mbti_type_classic = classic_mbti_weighted(responses)
+            st.write(f"Your MBTI type based on weighted answers: {mbti_type_classic}")
+
+            # You can add LLM-based prediction if needed here (example OpenAI-based model)
+            if api_key:
+                # Run the LLM (GPT-4, for example) model to generate a personality type.
+                prompt = f"""
+                Determine a person's personality type based on their answers to the following Myers-Briggs Type Indicator (MBTI) questions:
+                The person has answered the following questions:
+
+                {', '.join([f"{question['text']} {response}" for question, response in zip(questions, responses)])}
+
+                What is the MBTI personality type based on these answers?
+                """
+                try:
+                    response = openai.ChatCompletion.create(
+                        model="gpt-4o",
+                        messages=[{"role": "system", "content": "You are a helpful assistant."},
+                                  {"role": "user", "content": prompt}]
+                    )
+                    mbti_type_llm = response['choices'][0]['message']['content']
+                    st.write(f"Your MBTI type according to AI: {mbti_type_llm}")
+                except Exception as e:
+                    st.error(f"Error occurred: {e}")
+
+        else:
+            st.warning("Please answer all the questions!")
+
+# Main function to display the app
+def main():
+    if api_key:
+        show_mbti_quiz()
+    else:
+        st.info("Please enter your OpenAI API Key to begin the quiz.")
+
+if __name__ == "__main__":
+    main()