feat: enhanced error management in the workflow and returned messages

climateqa/engine/talk_to_data/main.py

@@ -50,7 +50,7 @@ async def ask_drias(query: str, index_state: int = 0, user_id: str | None = None
 
     if "error" in final_state and final_state["error"] != "":
         # No Sql query, no dataframe, no figure, no plot information, empty sql queries list, empty result dataframes list, empty figures list, empty plot information list, index state = 0, empty table list, error message
-        return None, None, None, None, [], [], [], 0, [], final_state["error"]
+        return None, None, None, None, [], [], [], [], 0, [], final_state["error"]
 
     sql_query = sql_queries[index_state]
     dataframe = result_dataframes[index_state]
@@ -112,7 +112,7 @@ async def ask_ipcc(query: str, index_state: int = 0, user_id: str | None = None)
 
     if "error" in final_state and final_state["error"] != "":
         # No Sql query, no dataframe, no figure, no plot information, empty sql queries list, empty result dataframes list, empty figures list, empty plot information list, index state = 0, empty table list, error message
-        return None, None, None, None, [], [], [], 0, [], final_state["error"]
+        return None, None, None, None, [], [], [], [], 0, [], final_state["error"]
 
     sql_query = sql_queries[index_state]
     dataframe = result_dataframes[index_state]

climateqa/engine/talk_to_data/workflow/ipcc.py

@@ -152,10 +152,18 @@ async def ipcc_workflow(user_input: str) -> State:
 
     # Set error messages if needed
     if not errors['have_relevant_table']:
-        state['error'] = "There is no relevant table in our database to answer your question"
+        state['error'] = (
+            "Sorry, I couldn't find any relevant table in our database to answer your question.\n"
+            "Try asking about a different climate indicator like temperature or precipitation."
+        )
     elif not errors['have_sql_query']:
-        state['error'] = "There is no relevant sql query on our database that can help to answer your question"
+        state['error'] = (
+            "Sorry, I couldn't generate a relevant SQL query to answer your question.\n"
+            "Try rephrasing your question to focus on a specific location, a year, or a month."
+        )
     elif not errors['have_dataframe']:
-        state['error'] = "There is no data in our table that can answer to your question"
-
+        state['error'] = (
+            "Sorry, there is no data in our tables that can answer your question.\n"
+            "Try asking about a more common location, or a different year."
+        )
     return state

front/tabs/tab_ipcc.py

@@ -68,6 +68,8 @@ def show_filter_by_scenario(table_names, index_state, dataframes):
         return gr.update(visible=False)
         
 def filter_by_scenario(dataframes, figures, table_names, index_state, scenario):
+    if len(dataframes) == 0:
+        return None, None 
     df = dataframes[index_state]
     if not table_names[index_state].startswith("Map"):
         return df, figures[index_state](df)