| import random |
| from models import WardrobeItem |
|
|
| class RecommendationEngine: |
| def __init__(self, items: list[WardrobeItem]): |
| self.items = items |
| self.tops = [i for i in items if i.category.lower() == "top"] |
| self.bottoms = [i for i in items if i.category.lower() == "bottom"] |
| self.shoes = [i for i in items if i.category.lower() == "shoes"] |
|
|
| def generate(self, style: str = "any"): |
| """ |
| Generates a recommended outfit structure. |
| Returns: { "items": [List of Objects], "note": "Style Note" } |
| """ |
| if not self.tops or not self.bottoms: |
| return None |
| |
| |
| |
| |
| top = random.choice(self.tops) |
| bottom = random.choice(self.bottoms) |
| |
| outfit_items = [top, bottom] |
| |
| |
| shoes = None |
| if self.shoes: |
| shoes = random.choice(self.shoes) |
| outfit_items.append(shoes) |
| |
| |
| note = self._generate_note(top, bottom, shoes, style) |
| |
| return { |
| "items": outfit_items, |
| "note": note |
| } |
| |
| def _generate_note(self, top, bottom, shoes, style): |
| |
| is_denim = "jean" in (bottom.tags or "").lower() |
| is_tee = "tee" in (top.tags or "").lower() or "t-shirt" in (top.tags or "").lower() |
| |
| if is_denim and is_tee: |
| return "Classic Casual: Timeless denim paired with a comfortable tee." |
| elif is_denim: |
| return "Smart Casual: Upgraded denim look." |
| elif style == "formal": |
| return "Sharp & Professional: Ready for business." |
| elif style == "party": |
| return "Evening Ready: A look that stands out." |
| |
| if shoes: |
| return "Head-to-Toe Look: Complete outfit with matching footwear." |
| |
| return "Daily Essential: A versatile combination from your wardrobe." |
|
|
