utils/handle_sql_commands.py

# handle_sql_commands.py
import duckdb
import pandas as pd
import streamlit as st


def execute_sql_duckdb(sql_command: str, dataframes: dict) -> pd.DataFrame or None:
    try:
        con = duckdb.connect(database=":memory:", read_only=False)
        for df_name, df in dataframes.items():
            con.register(df_name, df)
        result_df = con.execute(sql_command).fetchdf()
        con.close()
        return result_df
    except duckdb.Error as e:
        st.error(f"DuckDB Error: {e}")
        return None