Upload app.py

app.py

@@ -197,7 +197,7 @@ forms_questions = {
     ],
     "Algorithmic Aversion": [
         # Trust in LLM
-        {"question": "I trust the answers provided by AI chatbots (e.g., ChatGPT, Grok) to be accurate.", "options": ["Strongly Disagree", "Disagree", "Neutral", "Agree", "Strongly Agree"]},
+        {"question": "I trust the answers provided by AI chatbots (e.g., ChatGPT) to be accurate.", "options": ["Strongly Disagree", "Disagree", "Neutral", "Agree", "Strongly Agree"]},
         {"question": "I feel confident relying on an AI chatbot for important tasks.", "options": ["Strongly Disagree", "Disagree", "Neutral", " Agree", "Strongly Agree"]},
         {"question": "I worry that AI chatbots might give me incorrect information.", "options": ["Strongly Disagree", "Disagree", "Neutral", " Agree", "Strongly Agree"]},
 
@@ -266,7 +266,7 @@ def create_interface():
 
         # New Forms Section
         with gr.Column(visible=False, elem_id="forms_section") as forms_section:
-            gr.Markdown("## Post-Test Questions")
+            gr.Markdown("## Pre-Test Questions")
             gr.Markdown("Please answer the following questions to complete the study.")
             form_radios = []
             for form_name, questions in forms_questions.items():
@@ -299,8 +299,8 @@ def create_interface():
                 if next_key is None:
                     return {
                         id_section: gr.update(visible=True),
-                        main_interface: gr.update(visible=False),
                         forms_section: gr.update(visible=False),
+                        main_interface: gr.update(visible=False),
                         completion_section: gr.update(visible=False),
                         id_message: gr.update(value="The study is full. Thank you for your interest.", visible=True)
                     }
@@ -313,47 +313,47 @@ def create_interface():
             state.selected_indices = assignments[assignment_key]
             data = load_latest_data(prolific_id)
             if data:
-                if state.forms_completed:
+                if not state.forms_completed:
                     return {
                         id_section: gr.update(visible=False),
+                        forms_section: gr.update(visible=True),
                         main_interface: gr.update(visible=False),
-                        forms_section: gr.update(visible=False),
-                        completion_section: gr.update(visible=True),
+                        completion_section: gr.update(visible=False),
                         id_message: gr.update(value="", visible=False)
                     }
-                elif state.current_idx >= 28:
+                elif state.current_idx < 28:
                     return {
                         id_section: gr.update(visible=False),
-                        main_interface: gr.update(visible=False),
-                        forms_section: gr.update(visible=True),
+                        forms_section: gr.update(visible=False),
+                        main_interface: gr.update(visible=True),
                         completion_section: gr.update(visible=False),
-                        id_message: gr.update(value="", visible=False)
+                        id_message: gr.update(value="", visible=False),
+                        **update_interface(state.current_idx)
                     }
                 else:
                     return {
                         id_section: gr.update(visible=False),
-                        main_interface: gr.update(visible=True),
                         forms_section: gr.update(visible=False),
-                        completion_section: gr.update(visible=False),
-                        id_message: gr.update(value="", visible=False),
-                        **update_interface(state.current_idx)
+                        main_interface: gr.update(visible=False),
+                        completion_section: gr.update(visible=True),
+                        id_message: gr.update(value="", visible=False)
                     }
             else:
                 state.annotations = [None] * 28
                 state.current_idx = 0
+                state.forms_completed = False
+                state.form_responses = {}
                 return {
                     id_section: gr.update(visible=False),
-                    main_interface: gr.update(visible=True),
-                    forms_section: gr.update(visible=False),
+                    forms_section: gr.update(visible=True),
+                    main_interface: gr.update(visible=False),
                     completion_section: gr.update(visible=False),
-                    id_message: gr.update(value="", visible=False),
-                    **update_interface(0)
+                    id_message: gr.update(value="", visible=False)
                 }
-
         # Updated update_interface to use selected_indices
         def update_interface(current_idx):
             if current_idx >= 28:
-                current_idx = 39
+                current_idx = 27
             actual_idx = state.selected_indices[current_idx]
             current_data = response_pairs[actual_idx]
             progress = f"**Progress:** {current_idx/28:.0%} ({min(current_idx, 28)}/28)"
@@ -380,7 +380,6 @@ def create_interface():
                 gr.Warning(error_msg)
                 return {
                     main_interface: gr.update(visible=True),
-                    forms_section: gr.update(visible=False),
                     completion_section: gr.update(visible=False),
                     **update_interface(state.current_idx)
                 }
@@ -404,14 +403,12 @@ def create_interface():
             if new_idx >= 28:
                 return {
                     main_interface: gr.update(visible=False),
-                    forms_section: gr.update(visible=True),
-                    completion_section: gr.update(visible=False),
-                    **update_interface(39)  # Keep last question state
+                    completion_section: gr.update(visible=True),
+                    **update_interface(27)
                 }
             else:
                 return {
                     main_interface: gr.update(visible=True),
-                    forms_section: gr.update(visible=False),
                     completion_section: gr.update(visible=False),
                     **update_interface(new_idx)
                 }
@@ -430,44 +427,48 @@ def create_interface():
         # New function to handle form submission
         def handle_forms_submit(*form_inputs):
             if any(input_val is None for input_val in form_inputs):
-                gr.Warning("Please answer all questions before submitting.")
+                gr.Warning("Please answer all questions before proceeding.")
                 return {
                     forms_section: gr.update(visible=True),
+                    main_interface: gr.update(visible=False),
                     completion_section: gr.update(visible=False)
                 }
             state.form_responses = {}
             idx = 0
             for form_name, questions in forms_questions.items():
                 for q in questions:
-                    key = f"{form_name}_{q['question']}"  # Convert tuple to string
+                    key = f"{form_name}_{q['question']}"
                     state.form_responses[key] = form_inputs[idx]
                     idx += 1
             state.forms_completed = True
             save_annotations()
+            state.current_idx = 0
             return {
                 forms_section: gr.update(visible=False),
-                completion_section: gr.update(visible=True)
+                main_interface: gr.update(visible=True),
+                completion_section: gr.update(visible=False),
+                **update_interface(0)
             }
 
         # Event bindings
         id_submit_btn.click(
             handle_id_submit,
             inputs=prolific_id,
-            outputs=[id_section, main_interface, forms_section, completion_section, id_message, prompt_box, 
+            outputs=[id_section, forms_section, main_interface, completion_section, id_message, prompt_box, 
                      response_a, response_b, progress_md, feedback, confidence, selection_radio]
         )
 
         prev_btn.click(
             handle_navigation,
             inputs=[gr.State("prev"), selection_radio, confidence, feedback],
-            outputs=[main_interface, forms_section, completion_section, prompt_box, response_a, 
+            outputs=[main_interface, completion_section, prompt_box, response_a, 
                      response_b, progress_md, feedback, confidence, selection_radio]
         )
 
         next_btn.click(
             handle_navigation,
             inputs=[gr.State("next"), selection_radio, confidence, feedback],
-            outputs=[main_interface, forms_section, completion_section, prompt_box, response_a,
+            outputs=[main_interface, completion_section, prompt_box, response_a,
                      response_b, progress_md, feedback, confidence, selection_radio]
         )
 
@@ -481,7 +482,8 @@ def create_interface():
         submit_forms_btn.click(
             handle_forms_submit,
             inputs=form_radios,
-            outputs=[forms_section, completion_section]
+            outputs=[forms_section, main_interface, completion_section, prompt_box, response_a, 
+                     response_b, progress_md, feedback, confidence, selection_radio]
         )
 
     return demo