"""
==================================================================================
    Action Recognition Dataset - Logo Generator
==================================================================================
    Project: Action Recognition Dataset
    Website: RSK World (https://rskworld.in)
    Founded by: Molla Samser
    Designer & Tester: Rima Khatun
    Contact: help@rskworld.in | +91 93305 39277
    
    © 2026 RSK World. All Rights Reserved.
==================================================================================

This script generates logo images for the Action Recognition Dataset.
"""

import cv2
import numpy as np


def create_favicon():
    """Create a simple favicon for the website"""
    size = 64
    img = np.zeros((size, size, 4), dtype=np.uint8)
    
    # Background circle
    center = size // 2
    cv2.circle(img, (center, center), center - 2, (255, 212, 0, 255), -1)
    
    # Video play icon
    pts = np.array([
        [size // 3, size // 4],
        [size // 3, size * 3 // 4],
        [size * 2 // 3, size // 2]
    ], np.int32)
    cv2.fillPoly(img, [pts], (23, 20, 17, 255))
    
    cv2.imwrite('favicon.png', img)
    print("[OK] Created: favicon.png")


def create_logo():
    """Create the main logo image"""
    width = 400
    height = 100
    img = np.zeros((height, width, 4), dtype=np.uint8)
    
    # Draw video icon
    icon_size = 40
    x, y = 20, 30
    
    # Two overlapping rectangles for video icon
    cv2.rectangle(img, (x, y), (x + icon_size, y + int(icon_size * 0.7)), (255, 212, 0, 255), -1)
    cv2.rectangle(img, (x + 15, y + 8), (x + icon_size + 15, y + int(icon_size * 0.7) + 8), (255, 212, 0, 255), -1)
    
    # Text "RSKWorld"
    font = cv2.FONT_HERSHEY_SIMPLEX
    cv2.putText(img, "RSK", (x + icon_size + 30, y + 35), font, 1.5, (255, 255, 255, 255), 2, cv2.LINE_AA)
    cv2.putText(img, "World", (x + icon_size + 115, y + 35), font, 1.5, (255, 212, 0, 255), 2, cv2.LINE_AA)
    
    cv2.imwrite('logo.png', img)
    print("[OK] Created: logo.png")


def create_social_preview():
    """Create social media preview image (1200x630)"""
    width = 1200
    height = 630
    
    # Dark background
    img = np.full((height, width, 3), (23, 20, 17), dtype=np.uint8)
    
    # Add gradient effect
    for y in range(height):
        alpha = 0.3 * (1 - y / height)
        overlay_color = (int(255 * alpha), int(212 * alpha), 0)
        cv2.line(img, (0, y), (width, y), 
                 tuple(c1 + c2 for c1, c2 in zip((23, 20, 17), overlay_color)), 1)
    
    # Grid pattern
    grid_color = (35, 32, 28)
    for x in range(0, width, 50):
        cv2.line(img, (x, 0), (x, height), grid_color, 1)
    for y in range(0, height, 50):
        cv2.line(img, (0, y), (width, y), grid_color, 1)
    
    font = cv2.FONT_HERSHEY_SIMPLEX
    
    # Main title
    title = "Action Recognition"
    title_size = cv2.getTextSize(title, font, 2.5, 3)[0]
    cv2.putText(img, title, ((width - title_size[0]) // 2, 200), font, 2.5, (255, 255, 255), 3, cv2.LINE_AA)
    
    # Subtitle with gradient color
    subtitle = "Video Dataset"
    sub_size = cv2.getTextSize(subtitle, font, 2.0, 3)[0]
    cv2.putText(img, subtitle, ((width - sub_size[0]) // 2, 280), font, 2.0, (255, 212, 0), 3, cv2.LINE_AA)
    
    # Description
    desc = "For AI/ML Training | 3D CNNs | Video Classification"
    desc_size = cv2.getTextSize(desc, font, 0.8, 2)[0]
    cv2.putText(img, desc, ((width - desc_size[0]) // 2, 350), font, 0.8, (150, 150, 150), 2, cv2.LINE_AA)
    
    # Stats boxes
    stats = [("1,500+", "Videos"), ("12", "Classes"), ("50+", "Hours")]
    box_width = 200
    box_height = 100
    start_x = (width - (3 * box_width + 2 * 50)) // 2
    
    for i, (num, label) in enumerate(stats):
        x = start_x + i * (box_width + 50)
        y = 420
        
        # Box background
        cv2.rectangle(img, (x, y), (x + box_width, y + box_height), (40, 38, 35), -1)
        cv2.rectangle(img, (x, y), (x + box_width, y + box_height), (255, 212, 0), 2)
        
        # Number
        num_size = cv2.getTextSize(num, font, 1.2, 2)[0]
        cv2.putText(img, num, (x + (box_width - num_size[0]) // 2, y + 45), font, 1.2, (255, 212, 0), 2, cv2.LINE_AA)
        
        # Label
        label_size = cv2.getTextSize(label, font, 0.6, 1)[0]
        cv2.putText(img, label, (x + (box_width - label_size[0]) // 2, y + 80), font, 0.6, (200, 200, 200), 1, cv2.LINE_AA)
    
    # Footer
    footer = "rskworld.in | © 2026 RSK World"
    cv2.putText(img, footer, (20, height - 20), font, 0.5, (100, 100, 100), 1, cv2.LINE_AA)
    
    # RSK World branding
    cv2.putText(img, "RSK", (width - 150, height - 25), font, 0.8, (255, 255, 255), 2, cv2.LINE_AA)
    cv2.putText(img, "World", (width - 95, height - 25), font, 0.8, (255, 212, 0), 2, cv2.LINE_AA)
    
    cv2.imwrite('action-recognition.png', img)
    print("[OK] Created: action-recognition.png")


if __name__ == '__main__':
    print("=" * 50)
    print("RSK World Logo Generator")
    print("Website: https://rskworld.in")
    print("Contact: help@rskworld.in")
    print("=" * 50)
    print()
    
    try:
        create_favicon()
        create_logo()
        create_social_preview()
        print()
        print("All images generated successfully!")
    except Exception as e:
        print(f"Error: {e}")
        print("Make sure you have OpenCV installed:")
        print("  pip install opencv-python numpy")