Update ✨Entity Linking Application✨.py

✨Entity Linking Application✨.py

@@ -270,13 +270,12 @@ def main_cli():
     st.title("✨ Entity Linking Application ✨")
     st.caption("This web application is part of my master’s dissertation.")
 
-    if "disabled" not in st.session_state:
-        st.session_state.disabled = False
-
-    if "button_disabled" not in st.session_state:
-        st.session_state.button_disabled = False 
+    if 'disabled' in st.session_state and st.session_state.run_button == True:
+        st.session_state.running = True
+    else:
+        st.session_state.running = False
         
-    api_token = st.text_input("Enter your API key from [GitHub](https://github.com/marketplace/models/azure-openai/gpt-4o):", "", type="password", disabled=st.session_state.disabled)   
+    api_token = st.text_input("Enter your API key from [GitHub](https://github.com/marketplace/models/azure-openai/gpt-4o):", "", type="password", disabled=st.session_state.running)   
     
     if api_token:
         endpoint = "https://models.inference.ai.azure.com"
@@ -289,13 +288,13 @@ def main_cli():
     else:
         st.warning("Please enter an API token to proceed.")
         
-    input_sentence_user = st.text_input("Enter a sentence:", "", disabled=st.session_state.disabled)
-    input_mention_user = st.text_input("Enter a textural reference (mention) that is inside the sentence:", "", disabled=st.session_state.disabled)
-    single = st.selectbox("Search each word individually? (Useful for difficult mentions)", ['Yes', 'No'], index=1, disabled=st.session_state.disabled)
-    combi = st.selectbox("Make combinations of each word? (Useful for difficult mentions)", ['Yes', 'No'], index=1, disabled=st.session_state.disabled)
-    disambi = st.selectbox("Run acronym disambiguation? (Enable it if the mention include an acronym or if it is nested)", ['Yes', 'No'], index=0, disabled=st.session_state.disabled)
+    input_sentence_user = st.text_input("Enter a sentence:", "", disabled=st.session_state.running)
+    input_mention_user = st.text_input("Enter a textural reference (mention) that is inside the sentence:", "", disabled=st.session_state.running)
+    single = st.selectbox("Search each word individually? (Useful for difficult mentions)", ['Yes', 'No'], index=1, disabled=st.session_state.running)
+    combi = st.selectbox("Make combinations of each word? (Useful for difficult mentions)", ['Yes', 'No'], index=1, disabled=st.session_state.running)
+    disambi = st.selectbox("Run acronym disambiguation? (Enable it if the mention include an acronym or if it is nested)", ['Yes', 'No'], index=0, disabled=st.session_state.running)
     
-    if st.button("Run Entity Linking", key="button_disabled", disabled=st.session_state.button_disabled):
+    if st.button("Run Entity Linking", key="disabled", disabled=st.session_state.running):
         if input_sentence_user and input_mention_user:
             # check if the mention is in the sentence
             if input_mention_user in input_sentence_user: