Update ✨Entity Linking Application✨.py

✨Entity Linking Application✨.py

@@ -283,7 +283,6 @@ def main_cli():
     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)
     deep_search = st.selectbox("Perform deep search? (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="run_button", disabled=st.session_state.running):
         if input_sentence_user and input_mention_user:
@@ -296,82 +295,77 @@ def main_cli():
                     list_with_full_names = []
                     list_with_names_to_show = []
                      
-                    if disambi == "Yes":
-                        response = client.chat.completions.create(
-                            messages=[
-                                {
-                                    "role": "system",
-                                    "content": """
-                                                I will give you one or more labels within a sentence. Your task is as follows:
-    
-                                                Identify each label in the sentence, and check if it is an acronym.
-    
-                                                If the label is an acronym, respond with the full name of the acronym.
-                                                If the label is not an acronym, respond with the label exactly as it was given to you.
-                                                If a label contains multiple terms (e.g., 'phase and DIC microscopy'), treat each term within the label as a separate label.
-    
-                                                This means you should identify and explain each part of the label individually.
-                                                Each part should be on its own line in the response.
-                                                Context-Specific Terms: If the sentence context suggests a relevant term that applies to each label (such as "study" in 'morphological, sedimentological, and stratigraphical study'), add that term to each label’s explanation.
-    
-                                                Use context clues to determine the appropriate term to add (e.g., 'study' or 'microscopy').
-                                                Output Format: Your response should contain only the explanations, formatted as follows:
-    
-                                                Each label or part of a label should be on a new line.
-                                                Do not include any additional text, and do not repeat the original sentence.
-                                                Example 1:
-    
-                                                Input:
-    
-                                                label: phase and DIC microscopy
-                                                context: Tardigrades have been extracted from samples using centrifugation with Ludox AM™ and mounted on individual microscope slides in Hoyer's medium for identification under phase and DIC microscopy.
-                                                Expected response:
-    
-                                                phase: phase microscopy
-                                                DIC microscopy: Differential interference contrast microscopy
-                                                Example 2:
-    
-                                                Input:
-    
-                                                label: morphological, sedimentological, and stratigraphical study
-                                                context: This paper presents results of a morphological, sedimentological, and stratigraphical study of relict beach ridges formed on a prograded coastal barrier in Bream Bay, North Island New Zealand.
-                                                Expected response:
-    
-                                                morphological: morphological study
-                                                sedimentological: sedimentological study
-                                                stratigraphical: stratigraphical study
-                                                IMPORTANT:
-    
-                                                Each label, even if nested within another, should be treated as an individual item.
-                                                Each individual label or acronym should be output on a separate line.                               
-                                                """
-                                },
-                                {
-                                    "role": "user",
-                                    "content": f"label:{input_mention_user}, context:{input_sentence_user}"
-                                }
-                            ],
-                            temperature=1.0,
-                            top_p=1.0,
-                            max_tokens=1000,
-                            model=model_name
-                        )
-                       
+                    response = client.chat.completions.create(
+                        messages=[
+                            {
+                                "role": "system",
+                                "content": """
+                                            I will give you one or more labels within a sentence. Your task is as follows:
+
+                                            Identify each label in the sentence, and check if it is an acronym.
+
+                                            If the label is an acronym, respond with the full name of the acronym.
+                                            If the label is not an acronym, respond with the label exactly as it was given to you.
+                                            If a label contains multiple terms (e.g., 'phase and DIC microscopy'), treat each term within the label as a separate label.
+
+                                            This means you should identify and explain each part of the label individually.
+                                            Each part should be on its own line in the response.
+                                            Context-Specific Terms: If the sentence context suggests a relevant term that applies to each label (such as "study" in 'morphological, sedimentological, and stratigraphical study'), add that term to each label’s explanation.
+
+                                            Use context clues to determine the appropriate term to add (e.g., 'study' or 'microscopy').
+                                            Output Format: Your response should contain only the explanations, formatted as follows:
+
+                                            Each label or part of a label should be on a new line.
+                                            Do not include any additional text, and do not repeat the original sentence.
+                                            Example 1:
+
+                                            Input:
+
+                                            label: phase and DIC microscopy
+                                            context: Tardigrades have been extracted from samples using centrifugation with Ludox AM™ and mounted on individual microscope slides in Hoyer's medium for identification under phase and DIC microscopy.
+                                            Expected response:
+
+                                            phase: phase microscopy
+                                            DIC microscopy: Differential interference contrast microscopy
+                                            Example 2:
+
+                                            Input:
+
+                                            label: morphological, sedimentological, and stratigraphical study
+                                            context: This paper presents results of a morphological, sedimentological, and stratigraphical study of relict beach ridges formed on a prograded coastal barrier in Bream Bay, North Island New Zealand.
+                                            Expected response:
+
+                                            morphological: morphological study
+                                            sedimentological: sedimentological study
+                                            stratigraphical: stratigraphical study
+                                            IMPORTANT:
+
+                                            Each label, even if nested within another, should be treated as an individual item.
+                                            Each individual label or acronym should be output on a separate line.                               
+                                            """
+                            },
+                            {
+                                "role": "user",
+                                "content": f"label:{input_mention_user}, context:{input_sentence_user}"
+                            }
+                        ],
+                        temperature=1.0,
+                        top_p=1.0,
+                        max_tokens=1000,
+                        model=model_name
+                    )
+                   
+                    
+                    kati = response.choices[0].message.content.splitlines()
+                    print(response.choices[0].message.content)
+                    for i in kati:
+                        context = i.split(":")[-1].strip()
+                        original_name = i.split(":")[0].strip()
+                        list_with_full_names.append(context)
+                        list_with_names_to_show.append(original_name)
                         
-                        kati = response.choices[0].message.content.splitlines()
-                        print(response.choices[0].message.content)
-                        for i in kati:
-                            context = i.split(":")[-1].strip()
-                            original_name = i.split(":")[0].strip()
-                            list_with_full_names.append(context)
-                            list_with_names_to_show.append(original_name)
-                            
-                        name = ",".join(list_with_full_names)
+                    name = ",".join(list_with_full_names)
     
-                    else:
-                        name = input_mention_user
-                        list_with_full_names.append(name)
-                        list_with_names_to_show.append(name)
                     
                     input_sentence_user = input_sentence_user.replace(input_mention_user, name)  # Changing the mention to the correct one