import sqlite3
import os
# PERHATIKAN: Ini harus setup_database, BUKAN init_db
from database import setup_database

def isi_data_awal():
    # 1. Pastikan database dan tabel sudah dibuat
    setup_database()
    
    db_name = "cafe.db"

    # 2. Cek folder aset
    # Pastikan kamu punya folder bernama 'aset' dan di dalamnya ada gambar-gambarnya
    if not os.path.exists("aset"):
        print("❌ WADUH! Folder 'aset' gak ketemu. Pastikan nama folder di kiri adalah 'aset'.")
        return

    conn = sqlite3.connect(db_name)
    cur = conn.cursor()

    print("♻️  Membersihkan data lama...")
    cur.execute("DELETE FROM menu")
    cur.execute("DELETE FROM sqlite_sequence WHERE name='menu'") # Reset ID jadi 1

    # 3. Data Menu (Path gambar pakai 'aset/')
    menus = [
        ("Ayam Goreng", "Makanan", 15000, 20, "aset/ayam_goreng.jpg"),
        ("Bakso Urat", "Makanan", 12000, 15, "aset/bakso.jpg"),
        ("Mie Ayam", "Makanan", 10000, 25, "aset/mie_ayam.jpg"),
        ("Es Teh Manis", "Minuman", 3000, 50, "aset/es_teh.jpg"),
        ("Jus Jeruk", "Minuman", 5000, 30, "aset/jus_jeruk.jpg"),
    ]

    print("📥 Sedang mengisi data menu...")

    try:
        cur.executemany("""
            INSERT INTO menu (nama, kategori, harga, stok, gambar) 
            VALUES (?, ?, ?, ?, ?)
        """, menus)
        conn.commit()
        print("✅ MANTAP! Data menu berhasil masuk. Sekarang jalankan main.py!")
    except Exception as e:
        print(f"❌ Error: {e}")
    finally:
        conn.close()

if __name__ == "__main__":
    isi_data_awal()