Update app.py

app.py

@@ -82,6 +82,18 @@ def save_responses_to_json(username, responses):
     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')
@@ -127,14 +139,26 @@ 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 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",
+                    label="Download UserChoices.json",
                     data=json_data,
                     file_name="UserChoices.json",
                     mime="application/json"