fix: ab data saving

backend/session_manager.py

@@ -283,8 +283,6 @@ class SessionManager:
             return
 
         try:
-            # Get all AB pairs that were presented to the user from the submitted data
-            # The comparison data will tell us which type based on the IDs
             print(f"[DEBUG] Received ratings for {len(comparisons_data)} comparisons")
             
             # Build a set of already saved comparison IDs to avoid duplicates
@@ -299,25 +297,41 @@ class SessionManager:
             }
             print(f"[DEBUG] Already have {len(existing_pairs)} AB comparisons saved for this session")
             
-            # Process each comparison from the submitted data
-            for comp_id, comparison in comparisons_data.items():
-                clip_a_id = comparison.get("clip_a_id")
-                clip_b_id = comparison.get("clip_b_id")
-                
-                if not clip_a_id or not clip_b_id:
-                    print(f"[WARN] Skipping comparison {comp_id} - missing clip IDs")
-                    continue
+            # Determine which pairs we're processing based on comparison_type in the data
+            # Check first comparison to see if this is model or gender
+            first_comparison = next(iter(comparisons_data.values()), {}) if comparisons_data else {}
+            comparison_type = first_comparison.get("comparison_type", "")
+            
+            # Get the appropriate pairs from session
+            if comparison_type == "model_vs_model":
+                presented_pairs = session.get("ab_model_pairs", [])
+                print(f"[DEBUG] Processing model-vs-model pairs: {len(presented_pairs)} pairs presented")
+            elif comparison_type == "gender_vs_gender":
+                presented_pairs = session.get("ab_gender_pairs", [])
+                print(f"[DEBUG] Processing gender-vs-gender pairs: {len(presented_pairs)} pairs presented")
+            else:
+                print(f"[WARN] Unknown comparison type: {comparison_type}")
+                presented_pairs = []
+            
+            # Process all presented pairs
+            for clip_a, clip_b in presented_pairs:
+                clip_a_id = clip_a.id
+                clip_b_id = clip_b.id
                 
                 # Skip if we've already saved this pair
                 if (clip_a_id, clip_b_id) in existing_pairs:
                     print(f"[DEBUG] Skipping duplicate comparison: {clip_a_id} vs {clip_b_id}")
                     continue
                 
+                # Get user's rating for this pair from the submitted data
+                comp_id = f"{clip_a_id}_vs_{clip_b_id}"
+                comparison = comparisons_data.get(comp_id, {})
+                
                 ab_response = {
                     "session_id": session_id,
                     "clip_a_id": clip_a_id,
                     "clip_b_id": clip_b_id,
-                    "comparison_type": comparison.get("comparison_type"),
+                    "comparison_type": comparison_type,
                     "choice": comparison.get("choice"),  # Can be None if not rated
                     "comment": comparison.get("comment", ""),
                     # Support both model_vs_model (gender_mismatch_a/b)