Fix empty initial schema errors

esgen/functions.py

@@ -21,15 +21,12 @@ def entity_schema_generation(name, class_id, threshold, property_types):
     try:
         model = ESGenerationModel(name, class_id, threshold, property_types)
         model.generate_initial_schema()
-        if not model.pending["pending"]:
-            error_message = "No qualified properties found. Please retry with a lower cutoff."
-            return error_message, dict(), [[None, error_message]]
         prop = model.pending["pending"][0]
         prop["examples"] = collect_examples(model.class_id, get_id(prop["uri"]))
         bot_message = chatbot_verbaliser(prop["label"], get_id(prop["uri"]), prop["frequency"], prop["examples"])
         return str(model), model.pending, [[None, bot_message]]
     except (IndexError, KeyError):
-        error_message = "Failed to generate the initial EntitySchema. Please check your input class or raise an issue."
+        error_message = "Failed to generate the initial EntitySchema. Please check your input class, retry with a lower cutoff, or raise an issue."
         return error_message, dict(), [[None, error_message]]
 
 

esgen/queries.py

@@ -94,6 +94,7 @@ def filter_properties(class_id: str, num_instances: int, threshold: float, prope
     """
     # instance_count = count_instances(class_id)
     instances = collect_instances(class_id, num_instances)
+    num_instances = len(instances)  # update num_instance in case less than 200
     properties = collect_properties(instances, property_types)
     for key, value in properties.items():
         # prop_count = count_properties(instances, prop["property"])