minor ui changes

app.py

@@ -3,11 +3,16 @@ import openai
 import os
 import re
 import ast
+import pandas as pd
 
-st.title("Named Entity Recognition (NER) with GPT-3")
+st.title(":rocket: Named Entity Recognition (NER) with GPT-3")
 
-if "guidelines" not in st.session_state:
-    st.session_state["guidelines"] = [
+# st.header("Guidelines")
+st.markdown(
+    "You can edit the guidelines here. Press  `Delete` to remove a row after selecting it."
+)
+df = pd.DataFrame(
+    [
         {
             "entity": "PERSON",
             "definition": "Short name or full name of a person from any geographic regions.",
@@ -24,24 +29,8 @@ if "guidelines" not in st.session_state:
             "color": "blue",
         },
     ]
-
-
-st.header("Guidelines")
-
-# display guidelines in a table
-st.table(st.session_state["guidelines"])
-
-st.write("You can add new guidelines here.")
-new_entity = st.text_input("Entity")
-new_definition = st.text_input("Definition")
-color = st.text_input("Color")
-if st.button("Add Guideline"):
-    st.session_state["guidelines"].append(
-        {"entity": new_entity, "definition": new_definition, "color": color}
-    )
-    new_entity = ""
-    new_definition = ""
-
+)
+edited_df = st.experimental_data_editor(df, num_rows="dynamic")
 
 examples = [
     {
@@ -57,7 +46,7 @@ examples = [
 
 def generate_guidelines_prompt(guidelines):
     guidelines_prompt = "Entity Definition:\n"
-    for guideline in guidelines:
+    for guideline in guidelines.values():
         guidelines_prompt += f"{guideline['entity']}: {guideline['definition']}\n"
     guidelines_prompt += "\nOutput Format:\n"
     guidelines_prompt += "{{'PERSON': [list of entities present], 'DATE': [list of entities present], 'LOC': [list of entities present]}}\n"
@@ -76,7 +65,7 @@ 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(st.session_state["guidelines"])
+GUIDELINES_PROMPT = generate_guidelines_prompt(edited_df.to_dict(orient="index"))
 
 
 def openai_chat_completion_response(final_prompt):

requirements.txt

@@ -1,2 +1,3 @@
 streamlit
-openai
+openai
+pandas