From d4f1e110262c3d6cc5ecd1ecbe37a376b56e042c Mon Sep 17 00:00:00 2001
From: antoinefalisse <antoinefalisse@gmail.com>
Date: Sun, 15 Feb 2026 09:26:58 +0100
Subject: [PATCH] fix device model order

---
 utils.py | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/utils.py b/utils.py
index c7ee8d0..ce92015 100644
--- a/utils.py
+++ b/utils.py
@@ -254,10 +254,17 @@ def downloadVideosFromServer(session_id,trial_id, isDocker=True,
                 session_desc = getMetadataFromServer(session_id)      
                 
             # Load iPhone models.
-            phoneModel= []
-            for i,video in enumerate(trial["videos"]):    
-                phoneModel.append(video['parameters']['model'])
-            session_desc['iphoneModel'] = {'Cam' + str(i) : phoneModel[i] for i in range(len(phoneModel))}
+            iphoneModel = {}
+            for video in trial["videos"]:
+                device_id = video.get("device_id", "").replace('-', '').upper()
+                if device_id not in mappingCamDevice:
+                    raise ValueError(f"Unknown device_id: {device_id}")
+                k = mappingCamDevice[device_id]
+                model = None
+                if isinstance(video.get("parameters"), dict):
+                    model = video["parameters"].get("model")
+                iphoneModel[f"Cam{k}"] = model
+            session_desc['iphoneModel'] = iphoneModel
         
             # Save metadata.
             with open(sessionYamlPath, 'w') as file: