fixed color issue

app.py

@@ -92,6 +92,10 @@ def openai_chat_completion_response(final_prompt):
 
     return response["choices"][0]["message"]["content"].strip(" \n")
 
+colors = {}
+for guideline in st.session_state["guidelines"]:
+    colors[guideline["entity"]] = guideline["color"]
+
 
 my_sentence = st.text_input("Your Sentence")
 if st.button("Submit"):
@@ -101,7 +105,7 @@ if st.button("Submit"):
     ners_dictionary = ast.literal_eval(ners)
     for entity_type, entity_list in ners_dictionary.items():
         entity_list = list(set(entity_list))
-        color = st.session_state["guidelines"][entity_type]["color"]
+        color = colors[entity_type]
         for ent in entity_list:
             if ent != "None":
                 my_sentence = re.sub(