create_sample_data.py

import pandas as pd
import numpy as np
from datetime import datetime, timedelta
import random

def generate_sample_data():
    """Generate sample sales data for testing"""
    print("🔄 Generating sample sales data...")
    
    # Set random seed for reproducibility
    np.random.seed(42)
    
    # Define categories and products
    categories = {
        'Electronics': ['Laptop', 'Phone', 'Tablet', 'Headphones', 'Charger'],
        'Clothing': ['T-shirt', 'Jeans', 'Jacket', 'Shoes', 'Hat'],
        'Furniture': ['Chair', 'Table', 'Desk', 'Sofa', 'Bookshelf'],
        'Food': ['Coffee', 'Snack', 'Drink', 'Candy', 'Biscuit']
    }
    
    regions = ['North', 'South', 'East', 'West', 'Central']
    
    # Generate 1000 records
    data = []
    end_date = datetime.now()
    start_date = end_date - timedelta(days=365)
    
    for i in range(1000):
        category = random.choice(list(categories.keys()))
        product = random.choice(categories[category])
        date = start_date + timedelta(days=random.randint(0, 365))
        
        # Some missing values (5% chance)
        quantity = random.randint(1, 50) if random.random() > 0.05 else None
        price = round(random.uniform(10, 1000), 2) if random.random() > 0.05 else None
        
        data.append({
            'date': date.strftime('%Y-%m-%d'),
            'product_name': product,
            'category': category,
            'region': random.choice(regions),
            'quantity': quantity,
            'price': price,
            'cost': round(random.uniform(5, 800), 2) if price else None
        })
    
    df = pd.DataFrame(data)
    
    # Create data directory if not exists
    import os
    os.makedirs('data', exist_ok=True)
    
    # Save to CSV
    output_file = 'data/raw_sales_data.csv'
    df.to_csv(output_file, index=False)
    
    print(f"✅ Sample data created: {output_file}")
    print(f"📊 Total records: {len(df)}")
    print("\nFirst 5 rows:")
    print(df.head())
    
    return df

if __name__ == "__main__":
    generate_sample_data()