database.py

import sqlite3

DB_NAME = "clothing_reviews.db"

def init_db():
    conn = sqlite3.connect(DB_NAME)
    cursor = conn.cursor()
    cursor.execute('''CREATE TABLE IF NOT EXISTS products 
                      (id INTEGER PRIMARY KEY, category TEXT, name TEXT)''')
    cursor.execute('''CREATE TABLE IF NOT EXISTS reviews 
                  (id INTEGER PRIMARY KEY, product_id INTEGER, 
                   rating INTEGER, review_text TEXT, 
                   sentiment TEXT, date TEXT)''')
    
    cursor.execute("SELECT COUNT(*) FROM products")
    if cursor.fetchone()[0] == 0:
        sample_products = [
            ('Kurti', 'Embroidered Cotton Chikankari Kurti'),
            ('Jeans', 'High-Waisted Distressed Skinny Jeans'),
            ('Tops', 'Floral Print Peplum Blouse'),
            ('Leggings', 'High-Performance Seamless Yoga Leggings'),
            ('Kurti', 'Anarkali Style Floor-Length Kurti'),
            ('Tops', 'Oversized Linen Button-Down Shirt'),
            ('Jeans', 'Classic Straight-Leg Denim'),
            ('Leggings', 'Fleece-Lined Winter Thermal Leggings'),
            ('Kurti', 'Straight-Cut Rayon Daily Wear Kurti'),
            ('Tops', 'Silk Camisole with Lace Trim')
        ]
        cursor.executemany("INSERT INTO products (category, name) VALUES (?, ?)", sample_products)
    
    conn.commit()
    conn.close()

def query_db(query, params=(), fetch=False):
    conn = sqlite3.connect(DB_NAME)
    cursor = conn.cursor()
    cursor.execute(query, params)
    result = cursor.fetchall() if fetch else None
    conn.commit()
    conn.close()
    return result