in src/ui/ui.py [0:0]
def _button_helper_execute_button(self, b):
"""Helper method defining the button's action to triggers the ML model prediction
and execute the SQL query.
Args:
b (-): Required argument on action functions for buttons.
Returns:
None
"""
if self.tool.credentials_exist():
with self.processing_flag:
print("š Processing ....")
nlq_w_placeholders = self.tool.replace_name_for_placeholder(
self.nlq, self.proc_entities
)
sql_query = self.tool.ml_call(nlq_w_placeholders)
try:
rendered_sql_query = self.tool.render_template_query(
sql_query, self.proc_entities
)
except:
rendered_sql_query = None
try:
output = self.tool.execute_sql_query(rendered_sql_query)
except:
output = "\n•An error ocurred. We apologise for the inconvenience. Please try to re-formulate your query."
else:
output = "\nāļø Please set your data credentials to execute the query."
self._display_main(output, sql_query, rendered_sql_query)
self.processing_flag.clear_output()