---

import sqlite3

def init_db():
    conn = sqlite3.connect("dances.db")
    cursor = conn.cursor()
    
    # Удаляем старую таблицу, чтобы структура всегда была актуальной
    cursor.execute("DROP TABLE IF EXISTS elements")
    
    # Создаем таблицу заново со всеми нужными колонками
    cursor.execute("""
        CREATE TABLE elements (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            title TEXT,
            category TEXT,
            school TEXT,
            description TEXT,
            description_2 TEXT,
            video_url TEXT,
            is_favorite INTEGER
        )
    """)
    
    conn.commit()
    conn.close()

def add_element(title, category, school, description, description_2, video_url, is_favorite=0):
    conn = sqlite3.connect("dances.db")
    cursor = conn.cursor()
    cursor.execute("""
        INSERT INTO elements (title, category, school, description, description_2, video_url, is_favorite)
        VALUES (?, ?, ?, ?, ?, ?, ?)
    """, (title, category, school, description, description_2, video_url, is_favorite))
    conn.commit()
    conn.close()

def get_elements():
    conn = sqlite3.connect("dances.db")
    cursor = conn.cursor()
    cursor.execute("""
        SELECT title, category, school, description, description_2, video_url, is_favorite 
        FROM elements
    """)
    elements = cursor.fetchall()
    conn.close()
    return elements