Reason Message

app.py

@@ -221,7 +221,7 @@ def main(table, text_query):
 
     try:
         sql_query_result_raw = conn.sql(generated_sql_query)
-        sql_query_result,  = sql_query_result_raw.fetchall()
+        sql_query_result  = sql_query_result_raw.fetchall()
         sql_query_df = sql_query_result_raw.df()
         
         print(f"SQL Query Result: {sql_query_result}")
@@ -238,13 +238,18 @@ def main(table, text_query):
     except Exception as e:
         return generate_output(schema, prompt, generated_sql_query,
                                fig, sql_query_df)
-    try:
+    
+    if visualization_type != 'none':
+        try:
 
-        plot = visualize_result(text_query=text_query, sql_query=generated_sql_query,
-                      sql_result=sql_query_result, visualization_type=visualization_type)
+            plot = visualize_result(text_query=text_query, sql_query=generated_sql_query,
+                        sql_result=sql_query_result, visualization_type=visualization_type)
 
-    except Exception as e:
-        return generate_output(schema, prompt, generated_sql_query, fig, sql_query_df)
+        except Exception as e:
+            return generate_output(schema, prompt, generated_sql_query, fig, sql_query_df)
+    else:
+      gr.Warning(f"⚠️ {reason}")
+      return generate_output(schema, prompt, generated_sql_query, fig, sql_query_df)
 
     return generate_output(schema, prompt, generated_sql_query, plot, sql_query_df)