From eff7be430a03fe44d1330867a27d31cc71a92fc9 Mon Sep 17 00:00:00 2001
From: spookybear0 <50897638+spookybear0@users.noreply.github.com>
Date: Sun, 15 Sep 2024 21:42:39 -0700
Subject: [PATCH] fix edge case for color name

---
 helpers/tdfhelper.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/helpers/tdfhelper.py b/helpers/tdfhelper.py
index ec0c2d3..d7af165 100644
--- a/helpers/tdfhelper.py
+++ b/helpers/tdfhelper.py
@@ -31,7 +31,15 @@ def element_to_color(element: str) -> str:
         "Fire": "Red",
         "Ice": "Blue",
         "Earth": "Green",
-        "None": "None"
+        "None": "None",
+
+        # extras for edge cases
+        "Green": "Green",
+        "Red": "Red",
+        "Blue": "Blue",
+        "Solid Red": "Red",
+        "Solid Blue": "Blue",
+        "Solid Green": "Green",
     }
 
     return conversion[element]