Update app/models.py

app/models.py

@@ -3,18 +3,32 @@ from app.database import Base
 
 class Customer(Base):
     __tablename__ = "customers"
-    customer_id = Column(Integer, primary_key=True, index=True)
+
-    full_name = Column(String, index=True)
+    customer_id = Column(String(20), primary_key=True, index=True)  # IDs are strings now
-    email = Column(String, unique=True)
+    full_name = Column(String(100), index=True)
-    phone = Column(String)
+    email = Column(String(100), unique=True)
+    phone = Column(String(50))
+    date_of_birth = Column(Date)
+    age = Column(Integer)
+    gender = Column(String(20))
+    street_address = Column(String(100))
+    city = Column(String(50))
+    state = Column(String(10))
+    zip_code = Column(String(20))
     home_branch_id = Column(Integer)
     customer_since = Column(Date)
+    employment_status = Column(String(50))
+    annual_income = Column(Float)
+    credit_score = Column(Integer)
+    preferred_contact_method = Column(String(20))
 
 class Account(Base):
     __tablename__ = "accounts"
-    account_id = Column(Integer, primary_key=True, index=True)
+
-    account_number = Column(String, unique=True, index=True)
+    account_id = Column(String(20), primary_key=True, index=True)  # IDs are strings now
-    customer_id = Column(Integer)
+    account_number = Column(String(20), unique=True, index=True)
-    account_type = Column(String)
+    customer_id = Column(String(20))  # matches Customer.customer_id
+    account_type = Column(String(20))
     open_date = Column(Date)
     balance = Column(Float)
+    branch_id = Column(Integer)