utils.py

import sqlite3
import pandas as pd

def execute_query_and_return_df(query):
  """
  Executes a SQL query and returns the results as a Pandas DataFrame.

  Args:
    query: The SQL query to execute.

  Returns:
    A Pandas DataFrame containing the query results.
  """
  conn = sqlite3.connect("/content/Chinook_Sqlite.sqlite")
  try:
    df = pd.read_sql_query(query, conn)
    return df
  except Exception as e:
    print(f"Error executing query: {e}")
    return None
  finally:
    conn.close()