added pucks and samples

backend/app/database.py

@@ -3,7 +3,7 @@ from sqlalchemy import create_engine
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import sessionmaker
 
-SQLALCHEMY_DATABASE_URL = "sqlite:///./test.db"
+SQLALCHEMY_DATABASE_URL = "sqlite:///./test.db"  # Use appropriate path or database URL
 
 engine = create_engine(SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False})
 SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
@@ -21,19 +21,20 @@ def get_db():
 
 
 def init_db():
-    # Import inside function to avoid circular dependency
-    from app import models  
+    # Import models inside function to avoid circular dependency
+    from app import models
     Base.metadata.create_all(bind=engine)
 
 
 def load_sample_data(session: Session):
-    # Import inside function to avoid circular dependency
-    from app.data import contacts, return_addresses, dewars, proposals, shipments
+    # Import models inside function to avoid circular dependency
+    from app.data import contacts, return_addresses, dewars, proposals, shipments, pucks, samples
 
-    from app import models  # Ensure these imports are correct
+    from app import models
 
+    # If any data already exists, skip seeding
     if session.query(models.ContactPerson).first():
         return
 
-    session.add_all(contacts + return_addresses + dewars + proposals + shipments)
-    session.commit()
+    session.add_all(contacts + return_addresses + dewars + proposals + shipments + pucks + samples)
+    session.commit()