Update main.py

main.py

@@ -278,20 +278,24 @@ def entities_and_types(sentence):
 
     return entity_dict
 '''
+def convert_ner_format(ner_output):
+    return [[item["token"], item["tags"]] for item in ner_output]
+    
 def entities_and_types(sentence):
+    print("\n=== NER DEBUG ===")
+    print("INPUT:", sentence)
+
     ner_output = extract(sentence)
     print("RAW NER OUTPUT:", ner_output)
-         
-    try:
-        if ner_output != []:
-            entities = distill_entities(ner_output)
-        else:
-            output_list = jsons_to_list_of_lists(extract(sentence))
-        entities = distill_entities(output_list)   
-        print("DISTILLED ENTITIES:", entities)
-    except Exception as e:
-        print("❌ DISTILL ERROR:", str(e))
-        return {}
+
+    # ✅ FIX HERE
+    converted = convert_ner_format(ner_output)
+
+    print("CONVERTED FORMAT:", converted)
+
+    entities = distill_entities(converted)
+
+    print("DISTILLED ENTITIES:", entities)
 
     entity_dict = {}
     for name, entity_type, _, _ in entities: