Warning 2

app.py

@@ -246,6 +246,7 @@ def main(table, text_query):
                         sql_result=sql_query_result, visualization_type=visualization_type)
 
         except Exception as e:
+            gr.Warning(f"⚠️ {e}")
             return generate_output(schema, prompt, generated_sql_query, fig, sql_query_df)
     else:
       gr.Warning(f"⚠️ {reason}")
@@ -253,8 +254,8 @@ def main(table, text_query):
 
     return generate_output(schema, prompt, generated_sql_query, plot, sql_query_df)
 
-def generate_output(schema, prompt, generated_sql_query, fig, sql_query_df):
-    return [schema, prompt, generated_sql_query, fig, sql_query_df]
+def generate_output(schema, prompt, generated_sql_query, result_plot, sql_query_df):
+    return [schema, prompt, generated_sql_query, result_plot, sql_query_df]
 
 custom_css = """
 .gradio-container {