fahmiaziz/text2sql-rag

1import sqlite32import pandas as pd3 4def execute_query_and_return_df(query):5  """6  Executes a SQL query and returns the results as a Pandas DataFrame.7 8  Args:9    query: The SQL query to execute.10 11  Returns:12    A Pandas DataFrame containing the query results.13  """14  conn = sqlite3.connect("Chinook_Sqlite.sqlite")15  try:16    df = pd.read_sql_query(query, conn)17    return df18  except Exception as e:19    print(f"Error executing query: {e}")20    return None21  finally:22    conn.close()