Update main.py

main.py

@@ -281,9 +281,13 @@ def entities_and_types(sentence):
 def entities_and_types(sentence):
     ner_output = extract(sentence)
     print("RAW NER OUTPUT:", ner_output)
-
+         
     try:
-        entities = distill_entities(ner_output)
+        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))