relax the requirements of db key mapping in loading examples

backend/examples.py

@@ -142,7 +142,12 @@ def build_infos(abs_path, datatype: str, dataset_name: str, db_key: str):
         except Exception as e:
             print(f"Failed to load local file: {e}")
             return {}
-    print(f"datasets: {results_data['eval'].keys()}")
+
+    if db_key and db_key in results_data["eval"]:
+        dataset = results_data["eval"][db_key]
+    else:
+        # Get the first entry of results_data["eval"] if db_key is not found
+        dataset = next(iter(results_data["eval"].values()))
     dataset = results_data["eval"][db_key]
     first_model = next(iter(dataset.keys()))
     first_attack = next(iter(dataset[first_model].keys()))