Update sample_benchmarks.py

sample_benchmarks.py

@@ -4,12 +4,12 @@ Sample benchmarks initialization for Dynamic Highscores system.
 This script adds sample benchmarks to the database to provide initial options for users.
 """
 
-from database_schema import init_db
+from database_schema import DynamicHighscoresDB
 
 def add_sample_benchmarks():
     """Add sample benchmarks to the database."""
     # Initialize database
-    db = init_db()
+    db = DynamicHighscoresDB()
     
     # Sample benchmarks to add
     sample_benchmarks = [
@@ -46,21 +46,27 @@ def add_sample_benchmarks():
     ]
     
     # Add each benchmark to the database
+    added_count = 0
     for benchmark in sample_benchmarks:
-        benchmark_id = db.add_benchmark(
-            name=benchmark["name"],
-            dataset_id=benchmark["dataset_id"],
-            description=benchmark["description"],
-            metrics=benchmark["metrics"]
-        )
-        
-        print(f"Added benchmark '{benchmark['name']}' with ID: {benchmark_id}")
+        try:
+            benchmark_id = db.add_benchmark(
+                name=benchmark["name"],
+                dataset_id=benchmark["dataset_id"],
+                description=benchmark["description"],
+                metrics=benchmark["metrics"]
+            )
+            
+            if benchmark_id:
+                print(f"Added benchmark '{benchmark['name']}' with ID: {benchmark_id}")
+                added_count += 1
+        except Exception as e:
+            print(f"Error adding benchmark '{benchmark['name']}': {e}")
     
     # Close database connection
     db.close()
     
-    return len(sample_benchmarks)
+    return added_count
 
 if __name__ == "__main__":
     num_added = add_sample_benchmarks()
-    print(f"Added {num_added} sample benchmarks to the database.")
+    print(f"Added {num_added} sample benchmarks to the database.")