Update app.py

app.py

@@ -90,32 +90,31 @@ def show_mbti_quiz():
                 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}")
-
+            # Add a button to generate personality information
+            if st.button("Generate Personality Trait Information"):
+                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!")