Spaces:
Sleeping
Sleeping
| """ | |
| Shared utilities for seeding scripts. | |
| """ | |
| from PIL import Image | |
| from io import BytesIO | |
| import re | |
| import random | |
| def image_to_bytes(pil_image: Image.Image, fmt: str = "WEBP", quality: int = 85) -> bytes: | |
| """Convert PIL image to bytes in the specified format.""" | |
| buf = BytesIO() | |
| pil_image.convert("RGB").save(buf, format=fmt, quality=quality) | |
| return buf.getvalue() | |
| def clean_text(text: str) -> str: | |
| """Remove extra whitespace and non-printable chars.""" | |
| return re.sub(r"\s+", " ", text).strip() | |
| def estimate_price(article_type: str) -> float: | |
| """Assign a realistic price in ৳ based on article type.""" | |
| price_ranges = { | |
| "Shirts": (399, 1499), | |
| "Tshirts": (299, 999), | |
| "Sweaters": (699, 2499), | |
| "Jeans": (999, 3499), | |
| "Casual Shoes": (799, 2999), | |
| "Sports Shoes": (1499, 4999), | |
| "Formal Shoes": (1499, 4999), | |
| "Sneakers": (1499, 4999), | |
| "Bags": (499, 2999), | |
| "Handbags": (799, 3999), | |
| "Backpacks": (599, 2499), | |
| "Dresses": (799, 3499), | |
| "Jackets": (999, 4999), | |
| "Watches": (799, 5999), | |
| "Sunglasses": (399, 1999), | |
| "Kurtas": (399, 1499), | |
| "Tops": (399, 1299), | |
| "Shorts": (399, 999), | |
| "Trousers": (699, 2499), | |
| "Track Pants": (499, 1499), | |
| "Sandals": (399, 1499), | |
| "Flip Flops": (199, 699), | |
| "Heels": (999, 3499), | |
| "Belts": (299, 1499), | |
| "Wallets": (399, 1999), | |
| "Sarees": (999, 4999), | |
| "Dupatta": (299, 999), | |
| "Caps": (199, 799), | |
| "Socks": (99, 399), | |
| "Jewellery": (199, 2999), | |
| "Scarves": (299, 999), | |
| "Ties": (299, 999), | |
| } | |
| low, high = price_ranges.get(article_type, (499, 1999)) | |
| # Round to nearest 99 for realistic pricing | |
| raw = random.uniform(low, high) | |
| return round(raw / 100) * 100 - 1 | |