fix: database issues

backend/session_manager.py

@@ -298,9 +298,12 @@ class SessionManager:
             print(f"[DEBUG] Already have {len(existing_pairs)} AB comparisons saved for this session")
             
             # 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", "")
+            # Find the first comparison that has a comparison_type (skip incomplete ones)
+            comparison_type = None
+            for comp_data in comparisons_data.values():
+                if comp_data.get("comparison_type"):
+                    comparison_type = comp_data["comparison_type"]
+                    break
             
             # Get the appropriate pairs from session
             if comparison_type == "model_vs_model":
@@ -310,7 +313,7 @@ class SessionManager:
                 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}")
+                print(f"[WARN] Could not determine comparison type from data (found: {comparison_type})")
                 presented_pairs = []
             
             # Process all presented pairs
@@ -323,9 +326,13 @@ class SessionManager:
                     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, {})
+                # Find user's rating for this pair from the submitted data
+                # JS sends numeric keys ("1", "2", etc.), so search by clip IDs
+                comparison = {}
+                for comp_data in comparisons_data.values():
+                    if comp_data.get("clip_a_id") == clip_a_id and comp_data.get("clip_b_id") == clip_b_id:
+                        comparison = comp_data
+                        break
                 
                 ab_response = {
                     "session_id": session_id,
@@ -484,4 +491,4 @@ class SessionManager:
                 for r in self.responses.get("feedback", [])
                 if r.get("session_id") == session_id
             ],
-        }
+        }