Update app.py

app.py

@@ -112,9 +112,10 @@ questionnaire = {
     },
 }
 
+import json
+import numpy as np
 import gradio as gr
 import plotly.graph_objects as go
-import numpy as np
 from scipy.stats import percentileofscore
 
 # Define TRAIT_COLORS
@@ -257,7 +258,22 @@ def start_test():
         gr.update(visible=False),
     )
 
-# Next question
+
+
+
+
+# Save Plotly chart as HTML
+def save_plotly_html(chart, file_name="results_chart.html"):
+    chart.write_html(file_name)
+    print(f"Chart saved to {file_name}")
+
+# Save test results as a JSON file
+def save_results(responses, file_name="test_results.json"):
+    with open(file_name, "w") as f:
+        json.dump({"responses": responses}, f, indent=4)
+    print(f"Results saved to {file_name}")
+
+# Modified next_question to save results and chart
 def next_question(response):
     state["responses"].append(int(response))
     state["current_question"] += 1
@@ -265,6 +281,11 @@ def next_question(response):
     if state["current_question"] >= len(questions):
         scores, z_scores, percentiles = compute_scores_and_percentiles(state["responses"])
         result_chart = create_chart(scores, z_scores, percentiles)
+
+        # Save results and chart
+        save_results(state["responses"], "test_results.json")
+        save_plotly_html(result_chart, "results_chart.html")
+
         return "Test Complete! Your results:", result_chart, gr.update(visible=False), gr.update(visible=True)
 
     question = questions[state["current_question"]][2]
@@ -278,22 +299,50 @@ def next_question(response):
         gr.update(visible=False),
     )
 
-# Create Gradio app
 def create_gradio_app():
     with gr.Blocks() as app:
         gr.Markdown("## Extended Big Five Personality Test")
 
+        # UI Elements
         start_button = gr.Button("Start Test")
         question_text = gr.Textbox(label="Question", interactive=False, visible=True)
         button_group = gr.Radio([str(i) for i in range(1, 11)], label="Your Response (1-10)", visible=True)
         progress_gauge = gr.Plot()
         result_output = gr.Textbox(label="Results", visible=True)
+        download_results = gr.File(label="Download Results JSON", visible=False)
+        download_chart = gr.File(label="Download Chart HTML", visible=True)
+
+        # Start Test Logic
+        start_button.click(
+            start_test,
+            outputs=[question_text, progress_gauge, button_group, result_output]
+        )
+
+        # Question Logic
+        def handle_next_question(response):
+            output = next_question(response)
 
-        start_button.click(start_test, outputs=[question_text, progress_gauge, button_group, result_output])
-        button_group.change(next_question, inputs=button_group, outputs=[question_text, progress_gauge, button_group, result_output])
+            # Check if the test is complete to enable downloads
+            if state["current_question"] >= len(questions):
+                return (*output, "test_results.json", "results_chart.html")
+            else:
+                return (*output, None, None)
+
+        # Respond to button changes and enable downloads after test completion
+        button_group.change(
+            handle_next_question,
+            inputs=button_group,
+            outputs=[
+                question_text, progress_gauge, button_group, result_output,
+                download_results, download_chart
+            ]
+        )
 
     return app
 
-# Launch Gradio app
+
+# Launch the Gradio app
 app = create_gradio_app()
-app.launch()
+
+# Launch the app locally
+app.launch()