Update app102.py

app102.py

@@ -704,28 +704,7 @@ if "task_choice" in st.session_state:
                 labels_valid = not errors  # True if no validation errors
 
         elif classification_type == "Named Entity Recognition (NER)":
-            # # NER entity options
-            # ner_entities = [
-            #     "PERSON - Names of people, fictional characters, historical figures",
-            #     "ORG - Companies, institutions, agencies, teams",
-            #     "LOC - Physical locations (mountains, oceans, etc.)",
-            #     "GPE - Countries, cities, states, political regions",
-            #     "DATE - Calendar dates, years, centuries",
-            #     "TIME - Times, durations",
-            #     "MONEY - Monetary values with currency"
-            # ]
-            # selected_entities = st.multiselect(
-            #      "Select entities to recognize", 
-            #     ner_entities,
-            #     default=["PERSON - Names of people, fictional characters, historical figures",
-            #              "ORG - Companies, institutions, agencies, teams",
-            #             "LOC - Physical locations (mountains, oceans, etc.)",
-            #             "GPE - Countries, cities, states, political regions",
-            #             "DATE - Calendar dates, years, centuries",
-            #             "TIME - Times, durations",
-            #             "MONEY - Monetary values with currency"],
-            #             key="ner_entity_selection"
-            # )
+            
             #new 22/4/2025
             #if classification_type == "Named Entity Recognition (NER)":
             use_few_shot = True
@@ -1228,15 +1207,36 @@ if "task_choice" in st.session_state:
                         #                         'Use few-shot example?': 'Yes' if use_few_shot else 'No',
                         #                     })
 
-                                            #new 22/4/2025
+                        #                     #new 22/4/2025
+                        # if classification_type == "Named Entity Recognition (NER)":
+                        #     labeled_examples = [{
+                        #         'ner_output': response.strip(),
+                        #         'system_prompt': st.session_state.system_prompt,
+                        #         'system_role': st.session_state.system_role,
+                        #         'task_type': 'Named Entity Recognition (NER)',
+                        #         'Use few-shot example?': 'Yes' if use_few_shot else 'No',
+                        #     }]
+
+                        #new 24/4/2025
+                        labeled_examples = []
+
                         if classification_type == "Named Entity Recognition (NER)":
-                            labeled_examples = [{
-                                'ner_output': response.strip(),
-                                'system_prompt': st.session_state.system_prompt,
-                                'system_role': st.session_state.system_role,
-                                'task_type': 'Named Entity Recognition (NER)',
-                                'Use few-shot example?': 'Yes' if use_few_shot else 'No',
-                            }]
+                            # Split response into lines and try to extract the text and entities
+                            for line in response.strip().split('\n'):
+                                if line.strip():
+                                    parts = line.rsplit('Entities:', 1)
+                                    if len(parts) == 2:
+                                        text = parts[0].strip()
+                                        entities = parts[1].strip()
+                                        if text and entities:
+                                            labeled_examples.append({
+                                                'text': text,
+                                                'entities': entities,
+                                                'system_prompt': st.session_state.system_prompt,
+                                                'system_role': st.session_state.system_role,
+                                                'task_type': 'Named Entity Recognition (NER)',
+                                                'Use few-shot example?': 'Yes' if use_few_shot else 'No',
+                                            })
 
                                             ######