Update app.py

app.py

@@ -86,6 +86,23 @@ def detect_plot_intent(nl_query):
     plot_keywords = ['plot', 'graph', 'chart', 'distribution', 'visualize', 'trend', 'histogram', 'bar', 'line', 'scatter', 'pie']
     return any(keyword in nl_query.lower() for keyword in plot_keywords)
 
+
+def execute_query(sql_query):
+    """
+    Executes the SQL query and returns the results.
+    """
+    if sql_query.startswith("Error"):
+        return None, sql_query
+
+    try:
+        con = duckdb.connect()
+        con.execute(f"CREATE OR REPLACE VIEW contract_data AS SELECT * FROM '{dataset_path}'")
+        result_df = con.execute(sql_query).fetchdf()
+        con.close()
+        return result_df, ""
+    except Exception as e:
+        return None, f"Error executing query: {e}"
+
 def generate_plot(nl_query, result_df):
     if not detect_plot_intent(nl_query):
         return None, ""