Update ✨Entity Linking Application✨.py

✨Entity Linking Application✨.py

@@ -36,22 +36,6 @@ else:
 
 model = SentenceTransformer("Lajavaness/bilingual-embedding-large", trust_remote_code=True)
 
-
-api_token = st.text_input("Enter your API key from [GitHub](https://github.com/marketplace/models/azure-openai/gpt-4o):", "", type="password")
-
-
-if api_token:
-    endpoint = "https://models.inference.ai.azure.com"
-    model_name = "gpt-4o"
-    client = OpenAI(
-    base_url=endpoint,
-    api_key=api_token,
-)
-    st.success("API Token is set for this session.")
-else:
-    st.warning("Please enter an API token to proceed.")
-
-
 async def fetch_json(url, session):
     async with session.get(url) as response:
         return await response.json()
@@ -289,13 +273,26 @@ def main_cli():
     if "disabled" not in st.session_state:
         st.session_state.disabled = 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)   
+    
+    if api_token:
+        endpoint = "https://models.inference.ai.azure.com"
+        model_name = "gpt-4o"
+        client = OpenAI(
+        base_url=endpoint,
+        api_key=api_token,
+    )
+        st.success("API Token is set for this session.")
+    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)
     
-    if st.button("Run Entity Linking", key="disabled"):
+    if st.button("Run Entity Linking", key="disabled", disabled=st.session_state.disabled):
         if input_sentence_user and input_mention_user:
             # check if the mention is in the sentence
             if input_mention_user in input_sentence_user: