Spaces:
Sleeping
Sleeping
File size: 2,454 Bytes
a5a6a2e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 | from landmarks import get_landmarks
from geometry import extract_features
from classifier import classify_face_shape
import logging
logger = logging.getLogger(__name__)
RECOMMENDATIONS = {
# Keys must match frame-metadata.ts:
# Wayfarer, Aviator, Round, Square, Rectangle, Cat-eye, Clubmaster, Geometric, Oversized
"Oval": {
"glasses": ["Square", "Rectangle", "Wayfarer", "Aviator"],
"hair": ["Quiff", "Side Part", "Buzz Fade", "Pompadour"],
"beard": ["Light Stubble", "Short Boxed", "Verdi"],
"description": "Balanced proportions. Most frame styles work well."
},
"Round": {
"glasses": ["Rectangle", "Square", "Wayfarer", "Geometric"],
"hair": ["Pompadour", "High Fade", "Faux Hawk", "Side Part"],
"beard": ["Goatee", "Van Dyke", "Extended Chin Curtain"],
"description": "Soft angles with similar width and height. Angular frames add definition."
},
"Square": {
"glasses": ["Round", "Aviator", "Cat-eye", "Clubmaster"],
"hair": ["Textured Crop", "Side Fade", "Buzz Cut", "Crew Cut"],
"beard": ["Light Stubble", "Circle Beard", "Royale"],
"description": "Strong jawline and broad forehead. Round frames soften features."
},
"Oblong": {
"glasses": ["Oversized", "Aviator", "Square", "Wayfarer"],
"hair": ["Side Part", "Buzz Cut", "Fringe", "Caesar Cut"],
"beard": ["Mutton Chops", "Chin Strap", "Short Stubble"],
"description": "Face is longer than it is wide. Oversized frames balance length."
},
"Heart": {
"glasses": ["Round", "Wayfarer", "Cat-eye", "Geometric"],
"hair": ["Long Fringe", "Side Part", "Textured Quiff", "Swept Back"],
"beard": ["Full Beard", "Garibaldi", "Extended Goatee"],
"description": "Wider forehead and narrower chin. Bottom-heavy frames help balance."
},
"Diamond": {
"glasses": ["Round", "Cat-eye", "Clubmaster", "Oval"], # Oval frame type not in metadata, but maybe acceptable as fallback or mapped later. Kept for logic.
"hair": ["Faux Hawk", "Textured Crop", "Quiff", "Fringe"],
"beard": ["Full Beard", "Goatee", "Balbo"],
"description": "Cheekbones are the widest part. Curves soften the cheekbones."
}
}
def get_recommendations(face_shape):
"""Returns style recommendations based on face shape."""
return RECOMMENDATIONS.get(face_shape, RECOMMENDATIONS["Oval"])
|