fixed color issue

app.py

@@ -92,6 +92,7 @@ if st.button("Submit"):
 
     ners = openai_chat_completion_response(GUIDELINES_PROMPT)
     ners_dictionary = ast.literal_eval(ners)
+    st.json(ners_dictionary)
     for entity_type, entity_list in ners_dictionary.items():
         entity_list = list(set(entity_list))
         color = colors[entity_type]