fixed color issue

app.py

@@ -65,7 +65,6 @@ openai.api_key = os.getenv("OPENAI_API_KEY")
 SYSTEM_PROMPT = "You are a smart and intelligent Named Entity Recognition (NER) system. I will provide you the definition of the entities you need to extract, the sentence from where your extract the entities and the output format with examples."
 USER_PROMPT_1 = "Are you clear about your role?"
 ASSISTANT_PROMPT_1 = "Sure, I'm ready to help you with your NER task. Please provide me with the necessary information to get started."
-GUIDELINES_PROMPT = generate_guidelines_prompt(edited_df.to_dict(orient="index"))
 
 
 def openai_chat_completion_response(final_prompt):
@@ -82,15 +81,15 @@ def openai_chat_completion_response(final_prompt):
     return response["choices"][0]["message"]["content"].strip(" \n")
 
 
-colors = {}
-for guideline in edited_df.to_dict(orient="index").values():
-    colors[guideline["entity"]] = guideline["color"]
-
-
 my_sentence = st.text_input("Your Sentence")
 if st.button("Submit"):
+    colors = {}
+    for guideline in edited_df.to_dict(orient="index").values():
+        colors[guideline["entity"]] = guideline["color"]
+    
+    GUIDELINES_PROMPT = generate_guidelines_prompt(edited_df.to_dict(orient="index"))
     GUIDELINES_PROMPT = GUIDELINES_PROMPT.format(my_sentence)
-    print(GUIDELINES_PROMPT)
+
     ners = openai_chat_completion_response(GUIDELINES_PROMPT)
     ners_dictionary = ast.literal_eval(ners)
     for entity_type, entity_list in ners_dictionary.items():