Update abc.txt

abc.txt

@@ -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,29 @@ 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)
+
+# Function to save personality results to Output.json
+def save_personality_to_output_json(username, mbti_type_classic, mbti_type_llm):
+    output_data = {
+        "username": username,
+        "mbti_type_classic": mbti_type_classic,
+        "mbti_type_llm": mbti_type_llm
+    }
+
+    # Save to Output.json
+    with open("Output.json", "w") as json_file:
+        json.dump(output_data, json_file, indent=4)
+
 # Streamlit component to display the quiz and handle responses
 def show_mbti_quiz():
     st.title('FlexTemp Personality Test')
@@ -115,15 +139,54 @@ 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 personality info to Output.json
+                save_responses_to_json(participant_name, responses)
+                save_personality_to_output_json(participant_name, mbti_type_classic, mbti_type_llm)
+
+                with open("Output.json", "r") as json_file:
+                    json_data = json_file.read()
+
+                st.download_button(
+                    label="Download Output.json",
+                    data=json_data,
+                    file_name="Output.json",
+                    mime="application/json"
+                )
+
+                with open("UserChoices.json", "r") as json_file:
+                    json_data = json_file.read()
+
+                st.download_button(
+                    label="Download UserChoices.json",
+                    data=json_data,
+                    file_name="UserChoices.json",
+                    mime="application/json"
+                )
+
         else:
             st.warning("Please answer all the questions!")
 
 # Main function to display the app
 def main():
+    # Add instructions to the sidebar
+    with st.sidebar.expander("How This App Works", expanded=False):
+        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.
+            """)
+    
     if api_key:
         show_mbti_quiz()
     else:
         st.info("Please enter your OpenAI API Key to begin the quiz.")
 
 if __name__ == "__main__":
-    main() 
+    main()