From be096b3b86bf0b38306f21e079cc93fe188de395 Mon Sep 17 00:00:00 2001
From: Nik Clark <24965799+zzubnik@users.noreply.github.com>
Date: Tue, 1 Apr 2025 01:13:21 +0100
Subject: [PATCH] Add files via upload

---
 TT_CountryOfOrigin.py |  6 +++---
 __init__.py           | 14 +++++++++++++-
 2 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/TT_CountryOfOrigin.py b/TT_CountryOfOrigin.py
index c617cb1..0a9120c 100644
--- a/TT_CountryOfOrigin.py
+++ b/TT_CountryOfOrigin.py
@@ -5,7 +5,7 @@ from typing import List
 import comfy.model_management
 from nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
 
-class CountryOfOrigin:
+class TT_CountryOfOrigin:
     @classmethod
     def INPUT_TYPES(cls):
         options = cls.load_options()
@@ -35,5 +35,5 @@ class CountryOfOrigin:
         return (selection,)
 
 # Register the node
-NODE_CLASS_MAPPINGS["CountryOfOrigin"] = CountryOfOrigin
-NODE_DISPLAY_NAME_MAPPINGS["CountryOfOrigin"] = "CountryOfOrigin"
+NODE_CLASS_MAPPINGS["TT_CountryOfOrigin"] = TT_CountryOfOrigin
+NODE_DISPLAY_NAME_MAPPINGS["TT_CountryOfOrigin"] = "TT_CountryOfOrigin"
\ No newline at end of file
diff --git a/__init__.py b/__init__.py
index d2635a4..d84d64c 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,3 +1,15 @@
 from .TT_FemaleClothes import TT_FemaleClothes
 from .TT_RandomPlacesOnEarth import TT_RandomPlacesOnEarth
-from .TT_CountryOfOrigin import TT_CountryOfOrigin
\ No newline at end of file
+from .TT_CountryOfOrigin import TT_CountryOfOrigin
+
+NODE_CLASS_MAPPINGS = {
+    "TT_FemaleClothes": TT_FemaleClothes,
+    "TT_RandomPlacesOnEarth": TT_RandomPlacesOnEarth,
+    "TT_CountryOfOrigin": TT_CountryOfOrigin,
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "TT_FemaleClothes": "TT_FemaleClothes",
+    "TT_RandomPlacesOnEarth": "TT_RandomPlacesOnEarth",
+    "TT_CountryOfOrigin": "TT_CountryOfOrigin",
+}
\ No newline at end of file