import sqlite3

def init_db():
    conn = sqlite3.connect("ai_data.db")
    cursor = conn.cursor()
    cursor.execute("CREATE TABLE IF NOT EXISTS goals (id INTEGER PRIMARY KEY AUTOINCREMENT, text TEXT)")
    cursor.execute("CREATE TABLE IF NOT EXISTS feedback (id INTEGER PRIMARY KEY AUTOINCREMENT, score INTEGER, note TEXT)")
    conn.commit()
    conn.close()

def save_goal(goal: str):
    conn = sqlite3.connect("ai_data.db")
    cursor = conn.cursor()
    cursor.execute("INSERT INTO goals (text) VALUES (?)", (goal,))
    conn.commit()
    conn.close()

def list_goals():
    conn = sqlite3.connect("ai_data.db")
    cursor = conn.cursor()
    cursor.execute("SELECT text FROM goals")
    rows = cursor.fetchall()
    conn.close()
    return "\n".join(row[0] for row in rows)

def save_feedback(score: int, note: str = ""):
    conn = sqlite3.connect("ai_data.db")
    cursor = conn.cursor()
    cursor.execute("INSERT INTO feedback (score, note) VALUES (?, ?)", (score, note))
    conn.commit()
    conn.close()

def list_feedback():
    conn = sqlite3.connect("ai_data.db")
    cursor = conn.cursor()
    cursor.execute("SELECT score, note FROM feedback")
    rows = cursor.fetchall()
    conn.close()
    return "\n".join(f"{score} - {note}" for score, note in rows)