import sqlite3

def insert_data():
    conn = sqlite3.connect("personas.db")
    cursor = conn.cursor()

    # Define personas
    personas = [
        ("Marketer", "Writes engaging social media and blog content."),
        ("Software Engineer", "Shares insights on coding and software architecture."),
        ("Investor", "Focuses on market analysis and investment strategies."),
    ]

    # Insert personas if they don't exist
    for name, description in personas:
        cursor.execute("INSERT OR IGNORE INTO personas (name, description) VALUES (?, ?)", (name, description))

    # Fetch persona IDs
    cursor.execute("SELECT persona_id, name FROM personas")
    persona_map = {name: persona_id for persona_id, name in cursor.fetchall()}

    # Insert multiple posts per persona
    posts = [
        (persona_map["Marketer"], "Top 5 social media strategies for brand growth.", "Marketing, Branding"),
        (persona_map["Marketer"], "How to write viral LinkedIn posts?", "Content Writing, Engagement"),
        (persona_map["Software Engineer"], "Understanding microservices architecture.", "Software Design, Microservices"),
        (persona_map["Software Engineer"], "How Python is shaping the AI industry?", "AI, Python"),
        (persona_map["Investor"], "Why semiconductor stocks are booming in 2024?", "Investing, Technology"),
        (persona_map["Investor"], "Risk management strategies in stock market investing.", "Risk Management, Finance"),
    ]

    cursor.executemany("INSERT INTO posts (persona_id, text_blocks, tags) VALUES (?, ?, ?)", posts)

    conn.commit()
    conn.close()
    print("Multiple posts inserted for each persona!")

insert_data()