Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py (Final, Robust, and Workable Version with Chained Dropdowns)
+# app.py 
 import pandas as pd
 import numpy as np
 import matplotlib.pyplot as plt
@@ -84,7 +84,7 @@ class EnhancedAIvsRealGazeAnalyzer:
         self.combined_data['Answer_Correctness'] = self.combined_data['Correct'].map({True: 'Correct', False: 'Incorrect'})
         
         self.numeric_cols = self.combined_data.select_dtypes(include=np.number).columns.tolist()
-        self.time_metrics = [c for c in self.numeric_cols if any(k in c.lower() for k in ['time', 'duration', 'fixation'])]
+        self.time_ metrics = [c for c in self.numeric_cols if any(k in c.lower() for k in ['time', 'duration', 'fixation'])]
         self.valid_playback_participants = sorted(list(self.valid_playback_trials.keys()))
         print(f"--- Data Loading Successful. Found {len(self.valid_playback_participants)} participants with fixation data. ---")
         return self
@@ -161,7 +161,7 @@ class EnhancedAIvsRealGazeAnalyzer:
         trial_info = self.combined_data[(self.combined_data['participant_id'] == str(participant)) & (self.combined_data['Question'] == question)].iloc[0]
         summary_text = f"**Actual Answer:** `{trial_info['Answer_Correctness']}`"
         
-        return summary_text, fig, gr.Slider(maximum=slider_max, value=fixation_num, interactive=True)
+        return summary_text, fig, gr.Slider(maximum=slider_max, value=fixation_num, interactive=True, step=1, minimum=0)
     
     def analyze_rq1_metric(self, metric):
         if not metric or metric not in self.combined_data.columns: return None, "Metric not found."
@@ -176,7 +176,23 @@ class EnhancedAIvsRealGazeAnalyzer:
     def update_question_dropdown(self, participant):
         """Dynamically updates the question dropdown based on the selected participant."""
         valid_questions = self.valid_playback_trials.get(participant, [])
-        return gr.Dropdown(choices=sorted(valid_questions), interactive=True, value=None)
+        return gr.Dropdown(choices=sorted(valid_questions), interactive=True, value=None, label="2. Select a Question")
+
+    # <<< FIX START: Step 1 -> Add a new handler method for when a question is selected. >>>
+    # This function provides a clean, single entry point to reset the visualization.
+    def handle_new_trial_selection(self, participant, question):
+        """Called when a new trial is selected. Resets the view to the first fixation."""
+        if not participant or not question:
+            # Return a default/empty state for all outputs if nothing is selected
+            return "Select a trial to begin.", None, gr.Slider(value=0, interactive=False)
+        
+        # Start the playback at the very first fixation for a better user experience.
+        # You can change '1' to '0' if you prefer it to start completely empty.
+        initial_fixation_num = 1
+        
+        # Call the main generation function with this fixed initial value
+        return self.generate_gaze_playback(participant, question, initial_fixation_num)
+    # <<< FIX END >>>
 
 # --- DATA SETUP & GRADIO APP ---
 def setup_and_load_data():
@@ -235,10 +251,29 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     rq2_test_size_slider.release(fn=analyzer.run_prediction_model, inputs=[rq2_test_size_slider, rq2_estimators_slider], outputs=outputs_rq2)
     rq2_estimators_slider.release(fn=analyzer.run_prediction_model, inputs=[rq2_test_size_slider, rq2_estimators_slider], outputs=outputs_rq2)
     
+    # <<< FIX START: Step 2 -> Update the event wiring for Tab 3. >>>
     # Chained dropdown logic for Tab 3
-    playback_participant.change(fn=analyzer.update_question_dropdown, inputs=playback_participant, outputs=playback_question)
-    playback_question.change(lambda: 0, None, playback_slider).then(fn=analyzer.generate_gaze_playback, inputs=[playback_participant, playback_question, playback_slider], outputs=outputs_playback)
-    playback_slider.release(fn=analyzer.generate_gaze_playback, inputs=[playback_participant, playback_question, playback_slider], outputs=outputs_playback)
+    playback_participant.change(
+        fn=analyzer.update_question_dropdown, 
+        inputs=playback_participant, 
+        outputs=playback_question
+    )
+
+    # When a question is selected, call the new handler to reset the entire playback view.
+    # This is much cleaner and more reliable than the previous .then() chain.
+    playback_question.change(
+        fn=analyzer.handle_new_trial_selection, 
+        inputs=[playback_participant, playback_question], 
+        outputs=outputs_playback
+    )
+    
+    # The slider event handler remains the same and works correctly for scrubbing through fixations.
+    playback_slider.release(
+        fn=analyzer.generate_gaze_playback, 
+        inputs=[playback_participant, playback_question, playback_slider], 
+        outputs=outputs_playback
+    )
+    # <<< FIX END >>>
 
     demo.load(fn=analyzer.analyze_rq1_metric, inputs=rq1_metric_dropdown, outputs=[rq1_plot_output, rq1_summary_output])
     demo.load(fn=analyzer.run_prediction_model, inputs=[rq2_test_size_slider, rq2_estimators_slider], outputs=outputs_rq2)