Update app.py

app.py

@@ -59,7 +59,11 @@ def text2sql(dataset_name, query_input):
     try:
         first_parquet = get_first_parquet(dataset_name)
     except Exception as error:
-        return f"❌ Dataset does not exist or is not supported {error=}"
+        return {
+            query_output: "",
+            df:pd.DataFrame([{"error": f"❌ Could not get dataset schema. {error=}"}])
+        }
+
     first_parquet_url = first_parquet["url"]
     print(first_parquet_url)
     con = duckdb.connect()
@@ -84,7 +88,10 @@ def text2sql(dataset_name, query_input):
     try:
         sql_output =  query_remote_model(text)
     except Exception as error:
-        return f"❌ Unable to get the SQL query based on the text. {error=}"
+        return {
+            query_output: "",
+            df:pd.DataFrame([{"error": f"❌ Unable to get the SQL query based on the text. {error=}"}])
+        }
    
     try:
         query_result = con.sql(sql_output).df()