diff --git a/demo.py b/demo.py
new file mode 100644
index 0000000..bdd4df7
--- /dev/null
+++ b/demo.py
@@ -0,0 +1,292 @@
+"""demo."""
+
+from datetime import datetime
+
+from sqlalchemy import (
+    JSON,
+    Boolean,
+    CheckConstraint,
+    Column,
+    DateTime,
+    ForeignKey,
+    Integer,
+    LargeBinary,
+    Numeric,
+    String,
+    Text,
+)
+from sqlalchemy.orm import declarative_base, relationship
+
+from sqlalchemy_data_model_visualizer import add_web_font_and_interactivity, generate_data_model_diagram
+
+Base = declarative_base()
+
+
+class GenericUser(Base):
+    """GenericUser."""
+
+    __tablename__ = "generic_user"
+    email = Column(String, primary_key=True, index=True)
+    external_id = Column(String, unique=True, nullable=False)
+    is_active = Column(Boolean, default=True)
+    is_blocked = Column(Boolean, default=False)
+    last_ip_address = Column(String, nullable=True)
+    last_user_agent = Column(String, nullable=True)
+    last_estimated_location = Column(JSON, nullable=True)
+    preferences = Column(JSON)
+    registered_at = Column(DateTime, default=datetime.utcnow, index=True)
+    last_login = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow, index=True)
+    is_deleted = Column(Boolean, default=False)
+    deleted_at = Column(DateTime, nullable=True)
+    customer = relationship("Customer", uselist=False, back_populates="generic_user")
+    content_creator = relationship("ContentCreator", uselist=False, back_populates="generic_user")
+    user_sessions = relationship("UserSession", back_populates="generic_user")
+    audit_logs = relationship("GenericAuditLog", back_populates="actor")
+    notifications = relationship("GenericNotification", back_populates="recipient")
+
+
+class Customer(Base):
+    """Customer."""
+
+    __tablename__ = "customer"
+    email = Column(String, ForeignKey("generic_user.email"), primary_key=True, index=True)
+    total_purchases = Column(Numeric(10, 10), default=0.0)
+    generic_user = relationship("GenericUser", back_populates="customer")
+    service_requests = relationship("ServiceRequest", back_populates="customer")
+    subscriptions = relationship("GenericSubscription", back_populates="customer")
+    subscription_usages = relationship("GenericSubscriptionUsage", back_populates="customer")
+    billing_infos = relationship("GenericBillingInfo", back_populates="customer")
+    feedbacks_provided = relationship("GenericFeedback", back_populates="customer")
+
+
+class ContentCreator(Base):
+    """ContentCreator."""
+
+    __tablename__ = "content_creator"
+    email = Column(String, ForeignKey("generic_user.email"), primary_key=True, index=True)
+    projects_created = Column(Integer, default=0)
+    revenue_share = Column(Numeric(10, 10), default=0.7)
+    total_earned = Column(Numeric(10, 10), default=0.0)
+    last_project_created_at = Column(DateTime, nullable=True)
+    generic_user = relationship("GenericUser", back_populates="content_creator")
+    api_credit_logs = relationship("GenericAPICreditLog", back_populates="content_creator")
+    api_keys = relationship("GenericAPIKey", back_populates="content_creator")
+    feedbacks_received = relationship("GenericFeedback", back_populates="content_creator")
+
+
+class UserSession(Base):
+    """UserSession."""
+
+    __tablename__ = "user_session"
+    id = Column(Integer, primary_key=True)
+    user_email = Column(String, ForeignKey("generic_user.email"), nullable=False)
+    session_token = Column(String, unique=True, nullable=False)
+    expires_at = Column(DateTime, nullable=False)
+    is_active = Column(Boolean, default=True)
+    created_at = Column(DateTime, default=datetime.utcnow)
+    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+    generic_user = relationship("GenericUser", back_populates="user_sessions")
+
+
+class FileStorage(Base):
+    """FileStorage."""
+
+    __tablename__ = "file_storage"
+    id = Column(Integer, primary_key=True, index=True)
+    file_data = Column(LargeBinary, nullable=False)
+    file_type = Column(String, nullable=False)
+    file_hash = Column(String, nullable=False, unique=True)
+    upload_date = Column(DateTime, default=datetime.utcnow)
+
+
+class ServiceRequest(Base):
+    """ServiceRequest."""
+
+    __tablename__ = "service_request"
+    unique_id_for_sharing = Column(String, primary_key=True, index=True)
+    status = Column(String, CheckConstraint("status IN ('pending', 'completed', 'failed')"), default="pending")
+    ip_address = Column(String)
+    request_time = Column(DateTime, default=datetime.utcnow, index=True)
+    request_last_updated_time = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+    user_input = Column(JSON)
+    input_data_string = Column(Text)
+    api_request = Column(JSON)
+    api_response = Column(JSON)
+    api_session_id = Column(String, nullable=True, unique=True)
+    total_cost = Column(Numeric(10, 10), nullable=True)
+    customer_email = Column(String, ForeignKey("customer.email"))
+    customer = relationship("Customer", back_populates="service_requests")
+
+
+# AuditLog
+class GenericAuditLog(Base):
+    """GenericAuditLog."""
+
+    __tablename__ = "generic_audit_log"
+    id = Column(Integer, primary_key=True, index=True)
+    action_type = Column(String, nullable=False, index=True)
+    outcome = Column(String, nullable=True)
+    field_affected = Column(String, nullable=True)
+    prev_value = Column(JSON, nullable=True)
+    new_value = Column(JSON, nullable=True)
+    actor_email = Column(String, ForeignKey("generic_user.email"), index=True)
+    related_request_id = Column(Integer, ForeignKey("generic_user_request.unique_id"))
+    timestamp = Column(DateTime, default=datetime.utcnow)
+    actor = relationship("GenericUser", back_populates="audit_logs")
+
+
+# Feedback
+class GenericFeedback(Base):
+    """GenericFeedback."""
+
+    __tablename__ = "generic_feedback"
+    id = Column(Integer, primary_key=True, index=True)
+    score = Column(Integer, nullable=False)
+    commentary = Column(Text, nullable=True)
+    customer_email = Column(String, ForeignKey("customer.email"), index=True)
+    content_creator_email = Column(String, ForeignKey("content_creator.email"), index=True)
+    request_id = Column(Integer, ForeignKey("generic_user_request.unique_id"))
+    last_updated = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+    is_removed = Column(Boolean, default=False)
+    removed_at = Column(DateTime, nullable=True)
+    customer = relationship("Customer", back_populates="feedbacks_provided")
+    content_creator = relationship("ContentCreator", back_populates="feedbacks_received")
+
+
+# APIKeys
+class GenericAPIKey(Base):
+    """GenericAPIKey."""
+
+    __tablename__ = "generic_api_key"
+    id = Column(Integer, primary_key=True, index=True)
+    api_key = Column(String, unique=True, nullable=False)
+    content_creator_email = Column(String, ForeignKey("content_creator.email"), index=True)
+    is_active = Column(Boolean, default=True)
+    is_revoked = Column(Boolean, default=False)
+    expires_at = Column(DateTime, nullable=True)
+    created_at = Column(DateTime, default=datetime.utcnow)
+    content_creator = relationship("ContentCreator", back_populates="api_keys")
+
+
+# Notification
+class GenericNotification(Base):
+    """GenericNotification."""
+
+    __tablename__ = "generic_notification"
+    id = Column(Integer, primary_key=True, index=True)
+    recipient_email = Column(String, ForeignKey("generic_user.email"), index=True)
+    notification_kind = Column(String, nullable=False)
+    is_read = Column(Boolean, default=False)
+    content = Column(Text, nullable=False)
+    created_at = Column(DateTime, default=datetime.utcnow)
+    read_at = Column(DateTime, nullable=True)
+    recipient = relationship("GenericUser", back_populates="notifications")
+
+
+# APICreditLog
+class GenericAPICreditLog(Base):
+    """GenericAPICreditLog."""
+
+    __tablename__ = "generic_api_credit_log"
+    id = Column(Integer, primary_key=True, index=True)
+    timestamp = Column(DateTime, default=datetime.utcnow)
+    is_paid = Column(Boolean, default=False)
+    status = Column(String, default="pending")
+    expense = Column(Numeric(10, 10), nullable=False)
+    request_id = Column(Integer, ForeignKey("generic_user_request.unique_id"))
+    token_count = Column(Integer, nullable=False)
+    content_creator_email = Column(String, ForeignKey("content_creator.email"))
+    content_creator = relationship("ContentCreator", back_populates="api_credit_logs")
+
+
+# SubscriptionType
+class GenericSubscriptionType(Base):
+    """GenericSubscriptionType."""
+
+    __tablename__ = "generic_subscription_type"
+    id = Column(Integer, primary_key=True, index=True)
+    name = Column(String, nullable=False)
+    monthly_fee = Column(Numeric(10, 10), nullable=False)
+    monthly_cap = Column(Integer, nullable=False)
+    created_at = Column(DateTime, default=datetime.utcnow)
+    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+    is_removed = Column(Boolean, default=False)
+    removed_at = Column(DateTime, nullable=True)
+    subscriptions = relationship("GenericSubscription", back_populates="subscription_type")
+
+
+# Subscription
+class GenericSubscription(Base):
+    """GenericSubscription."""
+
+    __tablename__ = "generic_subscription"
+    id = Column(Integer, primary_key=True, index=True)
+    customer_email = Column(String, ForeignKey("customer.email"), index=True)
+    start_date = Column(DateTime, default=datetime.utcnow)
+    end_date = Column(DateTime, nullable=True)
+    current_use = Column(Integer, default=0)
+    subscription_type_id = Column(Integer, ForeignKey("generic_subscription_type.id"))
+    customer = relationship("Customer", back_populates="subscriptions")
+    subscription_type = relationship("GenericSubscriptionType", back_populates="subscriptions")
+    subscription_usages = relationship("GenericSubscriptionUsage", back_populates="subscription")
+
+
+# SubscriptionUsage
+class GenericSubscriptionUsage(Base):
+    """GenericSubscriptionUsage."""
+
+    __tablename__ = "generic_subscription_usage"
+    id = Column(Integer, primary_key=True, index=True)
+    customer_email = Column(String, ForeignKey("customer.email"), index=True)
+    use_count = Column(Integer, default=0)
+    last_use = Column(DateTime, nullable=True)
+    subscription_id = Column(Integer, ForeignKey("generic_subscription.id"))
+    subscription_type_id = Column(Integer, ForeignKey("generic_subscription_type.id"))
+    customer = relationship("Customer", back_populates="subscription_usages")
+    subscription = relationship("GenericSubscription", back_populates="subscription_usages")
+    subscription_type = relationship("GenericSubscriptionType", backref="subscription_usages")
+
+
+# BillingInfo
+class GenericBillingInfo(Base):
+    """GenericBillingInfo."""
+
+    __tablename__ = "generic_billing_info"
+    id = Column(Integer, primary_key=True, index=True)
+    customer_email = Column(String, ForeignKey("customer.email"), index=True)
+    payment_type = Column(String, nullable=False)
+    payment_data = Column(JSON)
+    created_at = Column(DateTime, default=datetime.utcnow)
+    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+    is_removed = Column(Boolean, default=False)
+    removed_at = Column(DateTime, nullable=True)
+    customer = relationship("Customer", back_populates="billing_infos")
+
+
+def main() -> None:
+    """Main."""
+    models = [
+        GenericUser,
+        Customer,
+        ContentCreator,
+        UserSession,
+        FileStorage,
+        ServiceRequest,
+        GenericAuditLog,
+        GenericFeedback,
+        GenericAPIKey,
+        GenericNotification,
+        GenericAPICreditLog,
+        GenericSubscriptionType,
+        GenericSubscription,
+        GenericSubscriptionUsage,
+        GenericBillingInfo,
+    ]
+
+    # Generate the diagram and add interactivity
+    generate_data_model_diagram(models, "my_data_model_diagram", add_labels=True)
+    add_web_font_and_interactivity("my_data_model_diagram.svg", "my_interactive_data_model_diagram.svg")
+
+
+if __name__ == "__main__":
+    main()
diff --git a/requirements.txt b/requirements.txt
index bf61691..75ae1dc 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,4 @@
 sqlalchemy
 graphviz
 lxml
+ruff
\ No newline at end of file
diff --git a/ruff.toml b/ruff.toml
new file mode 100644
index 0000000..500f327
--- /dev/null
+++ b/ruff.toml
@@ -0,0 +1,17 @@
+lint.select = ["ALL"]
+target-version = "py39"
+
+line-length = 120
+
+
+lint.ignore = [
+    "G004",   # PERM: This is a preformen nit that dosnt genrely matter (https://docs.astral.sh/ruff/rules/logging-f-string/)
+    "FBT001", # PERM: This is desired behavior for this code (https://docs.astral.sh/ruff/rules/boolean-type-hint-positional-argument/)
+    "FBT002", # PERM: This is desired behavior for this code (https://docs.astral.sh/ruff/rules/boolean-default-value-positional-argument/)
+]
+
+[lint.pydocstyle]
+convention = "google"
+
+[lint.flake8-builtins]
+builtins-ignorelist = ["id"]
diff --git a/setup.py b/setup.py
index 91576c8..90b994b 100644
--- a/setup.py
+++ b/setup.py
@@ -1,37 +1,40 @@
-from setuptools import setup
+"""setup.py file for the sqlalchemy_data_model_visualizer package."""
+
 from pathlib import Path
 
+from setuptools import setup
+
 # Define the directory where this setup.py file is located
 here = Path(__file__).parent
 
 # Read the contents of README file
-long_description = (here / 'README.md').read_text(encoding='utf-8')
+long_description = (here / "README.md").read_text(encoding="utf-8")
 
 # Read the contents of requirements file
-requirements = (here / 'requirements.txt').read_text(encoding='utf-8').splitlines()
+requirements = (here / "requirements.txt").read_text(encoding="utf-8").splitlines()
 
 setup(
-    name='sqlalchemy_data_model_visualizer',
-    version='0.1.3',  # Update the version number for new releases
-    description='A tool to visualize SQLAlchemy data models with Graphviz.',
+    name="sqlalchemy_data_model_visualizer",
+    version="0.1.3",  # Update the version number for new releases
+    description="A tool to visualize SQLAlchemy data models with Graphviz.",
     long_description=long_description,
-    long_description_content_type='text/markdown',
-    author='Jeffrey Emanuel',
-    author_email='jeff@pastel.network',
-    url='https://github.com/Dicklesworthstone/sqlalchemy_data_model_visualizer',
-    py_modules=['sqlalchemy_data_model_visualizer'],
+    long_description_content_type="text/markdown",
+    author="Jeffrey Emanuel",
+    author_email="jeff@pastel.network",
+    url="https://github.com/Dicklesworthstone/sqlalchemy_data_model_visualizer",
+    py_modules=["sqlalchemy_data_model_visualizer"],
     install_requires=requirements,
     classifiers=[
-        'Development Status :: 3 - Alpha',
-        'Intended Audience :: Developers',
-        'License :: OSI Approved :: MIT License',
-        'Programming Language :: Python :: 3',
-        'Programming Language :: Python :: 3.8',
-        'Programming Language :: Python :: 3.9',
-        'Programming Language :: Python :: 3.10',
-        'Programming Language :: Python :: 3.11',
+        "Development Status :: 3 - Alpha",
+        "Intended Audience :: Developers",
+        "License :: OSI Approved :: MIT License",
+        "Programming Language :: Python :: 3",
+        "Programming Language :: Python :: 3.8",
+        "Programming Language :: Python :: 3.9",
+        "Programming Language :: Python :: 3.10",
+        "Programming Language :: Python :: 3.11",
     ],
-    license='MIT',
-    keywords='sqlalchemy visualization graphviz data-model',
+    license="MIT",
+    keywords="sqlalchemy visualization graphviz data-model",
     include_package_data=True,  # This tells setuptools to check MANIFEST.in for additional files
 )
diff --git a/sqlalchemy_data_model_visualizer.py b/sqlalchemy_data_model_visualizer.py
index 52b2929..d990988 100644
--- a/sqlalchemy_data_model_visualizer.py
+++ b/sqlalchemy_data_model_visualizer.py
@@ -1,23 +1,38 @@
-from datetime import datetime
-from typing import Optional
-from enum import Enum
-from decimal import Decimal
-from sqlalchemy.orm import sessionmaker, declarative_base, relationship
-from sqlalchemy import Column, String, DateTime, Integer, Numeric, Boolean, JSON, ForeignKey, LargeBinary, Text, UniqueConstraint, CheckConstraint, text as sql_text
-from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
-from sqlalchemy import inspect
+"""sqlalchemy_data_model_visualizer."""
+
+from pathlib import Path
+
 import graphviz
 from lxml import etree
-import os
-import re
-Base = declarative_base()
-
-def generate_data_model_diagram(models, output_file='my_data_model_diagram', add_labels=True, view_diagram=True):
+from sqlalchemy import inspect
+from sqlalchemy.orm import declarative_base
+
+
+def generate_data_model_diagram(
+    models: list[declarative_base],
+    output_file: str = "my_data_model_diagram",
+    add_labels: bool = True,
+    view_diagram: bool = True,
+) -> None:
+    """Generate a data model diagram from a list of SQLAlchemy models.
+
+    Args:
+        models (list[any]): A list of SQLAlchemy models to visualize.
+        output_file (str, optional): The name of the output file. Defaults to "my_data_model_diagram".
+        add_labels (bool, optional): Whether to add labels to the edges. Defaults to True.
+        view_diagram (bool, optional): Whether to open the diagram after rendering. Defaults to True.
+
+    Returns:
+        None: The diagram is saved to a file and optionally opened in the default viewer.
+    """
     # Initialize graph with more advanced visual settings
-    dot = graphviz.Digraph(comment='Interactive Data Models', format='svg', 
-                            graph_attr={'bgcolor': '#EEEEEE', 'rankdir': 'TB', 'splines': 'spline'},
-                            node_attr={'shape': 'none', 'fontsize': '12', 'fontname': 'Roboto'},
-                            edge_attr={'fontsize': '10', 'fontname': 'Roboto'})
+    dot = graphviz.Digraph(
+        comment="Interactive Data Models",
+        format="svg",
+        graph_attr={"bgcolor": "#EEEEEE", "rankdir": "TB", "splines": "spline"},
+        node_attr={"shape": "none", "fontsize": "12", "fontname": "Roboto"},
+        edge_attr={"fontsize": "10", "fontname": "Roboto"},
+    )
 
     # Iterate through each SQLAlchemy model
     for model in models:
@@ -25,11 +40,11 @@ def generate_data_model_diagram(models, output_file='my_data_model_diagram', add
         name = insp.class_.__name__
 
         # Create an HTML-like label for each model as a rich table
-        label = f'''<
+        label = f"""<
         <TABLE BORDER="0" CELLBORDER="1" CELLSPACING="0">
         <TR><TD COLSPAN="2" BGCOLOR="#3F51B5"><FONT COLOR="white">{name}</FONT></TD></TR>
-        '''
-                
+        """
+
         for column in insp.columns:
             constraints = []
             if column.primary_key:
@@ -38,17 +53,17 @@ def generate_data_model_diagram(models, output_file='my_data_model_diagram', add
                 constraints.append("Unique")
             if column.index:
                 constraints.append("Index")
-            
-            constraint_str = ','.join(constraints)
+
+            constraint_str = ",".join(constraints)
             color = "#BBDEFB"
-            
-            label += f'''<TR>
+
+            label += f"""<TR>
                          <TD BGCOLOR="{color}">{column.name}</TD>
                          <TD BGCOLOR="{color}">{column.type} ({constraint_str})</TD>
-                         </TR>'''
-        
-        label += '</TABLE>>'
-        
+                         </TR>"""
+
+        label += "</TABLE>>"
+
         # Create the node with added hyperlink to detailed documentation
         dot.node(name, label=label, URL=f"http://{name}_details.html")
 
@@ -56,250 +71,48 @@ def generate_data_model_diagram(models, output_file='my_data_model_diagram', add
         for rel in insp.relationships:
             target_name = rel.mapper.class_.__name__
             tooltip = f"Relation between {name} and {target_name}"
-            dot.edge(name, target_name, label=rel.key if add_labels else None, tooltip=tooltip, color="#1E88E5", style="dashed")
+            dot.edge(
+                name,
+                target_name,
+                label=rel.key if add_labels else None,
+                tooltip=tooltip,
+                color="#1E88E5",
+                style="dashed",
+            )
 
     # Render the graph to a file and open it
-    dot.render(output_file, view=view_diagram)           
+    dot.render(output_file, view=view_diagram)
+
 
+def add_web_font_and_interactivity(input_svg_file: str, output_svg_file: str) -> None:
+    """Add web font and interactivity to an SVG file.
 
-def add_web_font_and_interactivity(input_svg_file, output_svg_file):
-    if not os.path.exists(input_svg_file):
-        print(f"Error: {input_svg_file} does not exist.")
-        return
+    Args:
+        input_svg_file (str): The input SVG file.
+        output_svg_file (str): The output SVG file.
+    """
+    if not Path(input_svg_file).exists():
+        error = f"Error: {input_svg_file} does not exist."
+        raise FileNotFoundError(error)
 
     parser = etree.XMLParser(remove_blank_text=True)
-    try:
-        tree = etree.parse(input_svg_file, parser)
-    except etree.XMLSyntaxError as e:
-        print(f"Error parsing SVG: {e}")
-        return
+    # The input can be trusted
+    tree = etree.parse(input_svg_file, parser)  # noqa: S320
 
     root = tree.getroot()
 
     style_elem = etree.Element("style")
-    style_elem.text = '''
+    style_elem.text = """
     @import url("https://fonts.googleapis.com/css?family=Roboto:400,400i,700,700i");
-    '''
+    """
     root.insert(0, style_elem)
 
     for elem in root.iter():
-        if 'node' in elem.attrib.get('class', ''):
-            elem.attrib['class'] = 'table-hover'
-        if 'edge' in elem.attrib.get('class', ''):
-            source = elem.attrib.get('source')
-            target = elem.attrib.get('target')
-            elem.attrib['class'] = f'edge-hover edge-from-{source} edge-to-{target}'
-
-    tree.write(output_svg_file, pretty_print=True, xml_declaration=True, encoding='utf-8')
-
-# ________________________________________________________________
-
-
-# [Insert your sqlalchemy data model classes here below:]
-
-use_demo = 0
-
-if use_demo:
-    class GenericUser(Base):
-        __tablename__ = 'generic_user'
-        email = Column(String, primary_key=True, index=True)
-        external_id = Column(String, unique=True, nullable=False)
-        is_active = Column(Boolean, default=True)
-        is_blocked = Column(Boolean, default=False)
-        last_ip_address = Column(String, nullable=True)
-        last_user_agent = Column(String, nullable=True)
-        last_estimated_location = Column(JSON, nullable=True)
-        preferences = Column(JSON)
-        registered_at = Column(DateTime, default=datetime.utcnow, index=True)
-        last_login = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow, index=True)
-        is_deleted = Column(Boolean, default=False)
-        deleted_at = Column(DateTime, nullable=True)
-        customer = relationship("Customer", uselist=False, back_populates="generic_user")
-        content_creator = relationship("ContentCreator", uselist=False, back_populates="generic_user")
-        user_sessions = relationship("UserSession", back_populates="generic_user")
-        audit_logs = relationship("GenericAuditLog", back_populates="actor")
-        notifications = relationship("GenericNotification", back_populates="recipient")
-        
-    class Customer(Base):
-        __tablename__ = 'customer'
-        email = Column(String, ForeignKey('generic_user.email'), primary_key=True, index=True)
-        total_purchases = Column(Numeric(10, 10), default=0.0)
-        generic_user = relationship("GenericUser", back_populates="customer")
-        service_requests = relationship("ServiceRequest", back_populates="customer")
-        subscriptions = relationship("GenericSubscription", back_populates="customer")
-        subscription_usages = relationship("GenericSubscriptionUsage", back_populates="customer")
-        billing_infos = relationship("GenericBillingInfo", back_populates="customer")
-        feedbacks_provided = relationship("GenericFeedback", back_populates="customer")
-    
-    class ContentCreator(Base):
-        __tablename__ = 'content_creator'
-        email = Column(String, ForeignKey('generic_user.email'), primary_key=True, index=True)
-        projects_created = Column(Integer, default=0)
-        revenue_share = Column(Numeric(10, 10), default=0.7)
-        total_earned = Column(Numeric(10, 10), default=0.0)
-        last_project_created_at = Column(DateTime, nullable=True)
-        generic_user = relationship("GenericUser", back_populates="content_creator")
-        api_credit_logs = relationship("GenericAPICreditLog", back_populates="content_creator")
-        api_keys = relationship("GenericAPIKey", back_populates="content_creator")
-        feedbacks_received = relationship("GenericFeedback", back_populates="content_creator")
-    
-    class UserSession(Base):
-        __tablename__ = 'user_session'
-        id = Column(Integer, primary_key=True)
-        user_email = Column(String, ForeignKey('generic_user.email'), nullable=False)
-        session_token = Column(String, unique=True, nullable=False)
-        expires_at = Column(DateTime, nullable=False)
-        is_active = Column(Boolean, default=True)
-        created_at = Column(DateTime, default=datetime.utcnow)
-        updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
-        generic_user = relationship("GenericUser", back_populates="user_sessions")
-            
-    class FileStorage(Base):
-        __tablename__ = 'file_storage'
-        id = Column(Integer, primary_key=True, index=True)
-        file_data = Column(LargeBinary, nullable=False)
-        file_type = Column(String, nullable=False)
-        file_hash = Column(String, nullable=False, unique=True)
-        upload_date = Column(DateTime, default=datetime.utcnow)
-    
-    class ServiceRequest(Base):
-        __tablename__ = 'service_request'
-        unique_id_for_sharing = Column(String, primary_key=True, index=True)
-        status = Column(String, CheckConstraint("status IN ('pending', 'completed', 'failed')"), default='pending')
-        ip_address = Column(String)
-        request_time = Column(DateTime, default=datetime.utcnow, index=True)
-        request_last_updated_time = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
-        user_input = Column(JSON)
-        input_data_string = Column(Text)
-        api_request = Column(JSON)
-        api_response = Column(JSON)
-        api_session_id = Column(String, nullable=True, unique=True)
-        total_cost = Column(Numeric(10, 10), nullable=True)
-        customer_email = Column(String, ForeignKey('customer.email'))
-        customer = relationship("Customer", back_populates="service_requests")
-    
-    # AuditLog
-    class GenericAuditLog(Base):
-        __tablename__ = 'generic_audit_log'
-        id = Column(Integer, primary_key=True, index=True)
-        action_type = Column(String, nullable=False, index=True)
-        outcome = Column(String, nullable=True)
-        field_affected = Column(String, nullable=True)
-        prev_value = Column(JSON, nullable=True)
-        new_value = Column(JSON, nullable=True)
-        actor_email = Column(String, ForeignKey('generic_user.email'), index=True)
-        related_request_id = Column(Integer, ForeignKey('generic_user_request.unique_id'))
-        timestamp = Column(DateTime, default=datetime.utcnow)
-        actor = relationship("GenericUser", back_populates="audit_logs")
-    
-    # Feedback
-    class GenericFeedback(Base):
-        __tablename__ = 'generic_feedback'
-        id = Column(Integer, primary_key=True, index=True)
-        score = Column(Integer, nullable=False)
-        commentary = Column(Text, nullable=True)
-        customer_email = Column(String, ForeignKey('customer.email'), index=True)
-        content_creator_email = Column(String, ForeignKey('content_creator.email'), index=True)
-        request_id = Column(Integer, ForeignKey('generic_user_request.unique_id'))
-        last_updated = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
-        is_removed = Column(Boolean, default=False)
-        removed_at = Column(DateTime, nullable=True)
-        customer = relationship("Customer", back_populates="feedbacks_provided")
-        content_creator = relationship("ContentCreator", back_populates="feedbacks_received")
-    
-    # APIKeys
-    class GenericAPIKey(Base):
-        __tablename__ = 'generic_api_key'
-        id = Column(Integer, primary_key=True, index=True)
-        api_key = Column(String, unique=True, nullable=False)
-        content_creator_email = Column(String, ForeignKey('content_creator.email'), index=True)
-        is_active = Column(Boolean, default=True)
-        is_revoked = Column(Boolean, default=False)
-        expires_at = Column(DateTime, nullable=True)
-        created_at = Column(DateTime, default=datetime.utcnow)
-        content_creator = relationship("ContentCreator", back_populates="api_keys")
-    
-    # Notification
-    class GenericNotification(Base):
-        __tablename__ = 'generic_notification'
-        id = Column(Integer, primary_key=True, index=True)
-        recipient_email = Column(String, ForeignKey('generic_user.email'), index=True)
-        notification_kind = Column(String, nullable=False)
-        is_read = Column(Boolean, default=False)
-        content = Column(Text, nullable=False)
-        created_at = Column(DateTime, default=datetime.utcnow)
-        read_at = Column(DateTime, nullable=True)
-        recipient = relationship("GenericUser", back_populates="notifications")
-    
-    # APICreditLog
-    class GenericAPICreditLog(Base):
-        __tablename__ = 'generic_api_credit_log'
-        id = Column(Integer, primary_key=True, index=True)
-        timestamp = Column(DateTime, default=datetime.utcnow)
-        is_paid = Column(Boolean, default=False)
-        status = Column(String, default='pending')
-        expense = Column(Numeric(10, 10), nullable=False)
-        request_id = Column(Integer, ForeignKey('generic_user_request.unique_id'))
-        token_count = Column(Integer, nullable=False)
-        content_creator_email = Column(String, ForeignKey('content_creator.email'))
-        content_creator = relationship("ContentCreator", back_populates="api_credit_logs")
-    
-    # SubscriptionType
-    class GenericSubscriptionType(Base):
-        __tablename__ = 'generic_subscription_type'
-        id = Column(Integer, primary_key=True, index=True)
-        name = Column(String, nullable=False)
-        monthly_fee = Column(Numeric(10, 10), nullable=False)
-        monthly_cap = Column(Integer, nullable=False)
-        created_at = Column(DateTime, default=datetime.utcnow)
-        updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
-        is_removed = Column(Boolean, default=False)
-        removed_at = Column(DateTime, nullable=True)
-        subscriptions = relationship("GenericSubscription", back_populates="subscription_type")
-    
-    # Subscription
-    class GenericSubscription(Base):
-        __tablename__ = 'generic_subscription'
-        id = Column(Integer, primary_key=True, index=True)
-        customer_email = Column(String, ForeignKey('customer.email'), index=True)
-        start_date = Column(DateTime, default=datetime.utcnow)
-        end_date = Column(DateTime, nullable=True)
-        current_use = Column(Integer, default=0)
-        subscription_type_id = Column(Integer, ForeignKey('generic_subscription_type.id'))
-        customer = relationship("Customer", back_populates="subscriptions")
-        subscription_type = relationship("GenericSubscriptionType", back_populates="subscriptions")
-        subscription_usages = relationship("GenericSubscriptionUsage", back_populates="subscription")
-    
-    # SubscriptionUsage
-    class GenericSubscriptionUsage(Base):
-        __tablename__ = 'generic_subscription_usage'
-        id = Column(Integer, primary_key=True, index=True)
-        customer_email = Column(String, ForeignKey('customer.email'), index=True)
-        use_count = Column(Integer, default=0)
-        last_use = Column(DateTime, nullable=True)
-        subscription_id = Column(Integer, ForeignKey('generic_subscription.id'))
-        subscription_type_id = Column(Integer, ForeignKey('generic_subscription_type.id'))
-        customer = relationship("Customer", back_populates="subscription_usages")
-        subscription = relationship("GenericSubscription", back_populates="subscription_usages")
-        subscription_type = relationship("GenericSubscriptionType", backref="subscription_usages")
-    
-    # BillingInfo
-    class GenericBillingInfo(Base):
-        __tablename__ = 'generic_billing_info'
-        id = Column(Integer, primary_key=True, index=True)
-        customer_email = Column(String, ForeignKey('customer.email'), index=True)
-        payment_type = Column(String, nullable=False)
-        payment_data = Column(JSON)
-        created_at = Column(DateTime, default=datetime.utcnow)
-        updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
-        is_removed = Column(Boolean, default=False)
-        removed_at = Column(DateTime, nullable=True)
-        customer = relationship("Customer", back_populates="billing_infos")
-    
-    models = [GenericUser, Customer, ContentCreator, UserSession, FileStorage, ServiceRequest, GenericAuditLog, GenericFeedback, GenericAPIKey, GenericNotification, GenericAPICreditLog, GenericSubscriptionType, GenericSubscription, GenericSubscriptionUsage, GenericBillingInfo]
-    
-    
-    output_file_name = 'my_data_model_diagram'
-    # Generate the diagram and add interactivity
-    generate_data_model_diagram(models, output_file_name, add_labels=True)
-    add_web_font_and_interactivity('my_data_model_diagram.svg', 'my_interactive_data_model_diagram.svg')
+        if "node" in elem.attrib.get("class", ""):
+            elem.attrib["class"] = "table-hover"
+        if "edge" in elem.attrib.get("class", ""):
+            source = elem.attrib.get("source")
+            target = elem.attrib.get("target")
+            elem.attrib["class"] = f"edge-hover edge-from-{source} edge-to-{target}"
+
+    tree.write(output_svg_file, pretty_print=True, xml_declaration=True, encoding="utf-8")