setup_db.py

import pandas as pd
import sqlite3

# CSV file name
FILE = "U.S._Chronic_Disease_Indicators.csv"

# SQLite database name
DB = "health.db"

print("Loading dataset into pandas...")

# Read CSV
df = pd.read_csv(FILE, low_memory=False)

# Clean column names
df.columns = [
    c.strip().lower().replace(" ", "_")
    for c in df.columns
]

print("Creating SQLite database...")

# Connect to SQLite
conn = sqlite3.connect(DB)

# Save dataframe into SQL table
df.to_sql(
    "health_indicators",
    conn,
    if_exists="replace",
    index=False
)

# Close connection
conn.close()

print(f"Loaded {len(df):,} rows into health.db")
print("Table name: health_indicators")
print("Columns:")
print(list(df.columns))