Update app.py

app.py

@@ -1,7 +1,8 @@
 import streamlit as st
 import openai
-from dotenv import load_dotenv
+import json
 import os
+from dotenv import load_dotenv
 
 # Load the OpenAI API Key
 api_key = st.text_input('Enter your OpenAI API Key', type="password")
@@ -70,6 +71,17 @@ def classic_mbti_weighted(responses):
             mbti_type += trait2
     return mbti_type
 
+# Function to save responses to a JSON file
+def save_responses_to_json(username, responses):
+    user_data = {
+        "username": username,
+        "responses": [{"text": question["text"], "answer": response} for question, response in zip(questions, responses)]
+    }
+
+    # Save to UserChoices.json
+    with open("UserChoices.json", "w") as json_file:
+        json.dump(user_data, json_file, indent=4)
+
 # Streamlit component to display the quiz and handle responses
 def show_mbti_quiz():
     st.title('FlexTemp Personality Test')
@@ -107,7 +119,7 @@ def show_mbti_quiz():
                     """
                     try:
                         response = openai.ChatCompletion.create(
-                            model="gpt-4o",
+                            model="gpt-4",
                             messages=[{"role": "system", "content": "You are a helpful assistant."},
                                       {"role": "user", "content": prompt}]
                         )
@@ -115,6 +127,19 @@ def show_mbti_quiz():
                         st.write(f"Your MBTI type according to AI: {mbti_type_llm}")
                     except Exception as e:
                         st.error(f"Error occurred: {e}")
+                
+                # Save responses and allow download of UserChoices.json
+                save_responses_to_json(participant_name, responses)
+                with open("UserChoices.json", "r") as json_file:
+                    json_data = json_file.read()
+                
+                st.download_button(
+                    label="Download UserChoices",
+                    data=json_data,
+                    file_name="UserChoices.json",
+                    mime="application/json"
+                )
+
         else:
             st.warning("Please answer all the questions!")
 
@@ -125,12 +150,10 @@ def main():
         st.write("""
             ### FlexTemp Personality Test
             This app is designed to help you determine your MBTI personality type based on your answers to a series of questions. The process works as follows:
-
             1. **Weighted MBTI Scoring**: 
             - Each question corresponds to a trait in the MBTI system. 
             - Your responses are scored on a scale from "Strongly Agree" to "Strongly Disagree", with each level being assigned a weight. 
             - These weights are used to calculate your MBTI type by comparing the scores of trait pairs (E/I, S/N, T/F, J/P).
-
             2. **LLM-Based Prediction**:
             - Optionally, you can also get your MBTI type based on the answers using a language model (LLM) like GPT-4. This provides an additional prediction that may offer insights into your personality.
             - The LLM is trained on vast amounts of data and can generate responses based on patterns from psychological research and real-world interactions.