Error handling

app.py

@@ -58,8 +58,8 @@ def text2sql(dataset_name, query_input):
     print(f"start text2sql for {dataset_name}")
     try:
         first_parquet = get_first_parquet(dataset_name)
-    except Exception as e:
-        return f"❌ Dataset does not exist or is not supported {e}"
+    except Exception as error:
+        return f"❌ Dataset does not exist or is not supported {error=}"
     first_parquet_url = first_parquet["url"]
     print(first_parquet_url)
     con = duckdb.connect()
@@ -81,9 +81,11 @@ def text2sql(dataset_name, query_input):
 
     ### Response (use duckdb shorthand if possible) replace table name with {first_parquet_url} in the generated sql query:
     """
-    
-    sql_output =  query_remote_model(text)
-
+    try:
+        sql_output =  query_remote_model(text)
+    except Exception as error:
+        return f"❌ Unable to get the SQL query based on the text. {error=}"
+   
     try:
         query_result = con.sql(sql_output).df()
     except Exception as error: