+i,dZddlZddlZddlmZddlmZddlmZee jjdz Z dZ dgd gd d d d dggdd dgdgdd dgdgdd dgdgdd dgdgdd dgdgdd dgd gd!d d"gd#gdd d$gd#gdd d%gd&gd'd d( Z d)d*d+iiZgd,gd-d.Zeej#ZGd/d0Zd1Zd2Zd3Zde fd4Ze fd5Zy)6z@ UI/UX Pro Max Core - BM25 search engine for UI/UX style guides N)Path)log) defaultdictdataz styles.csv)Style CategoryKeywordsBest ForTypeAI Prompt Keywords)rr r zPrimary ColorszEffects & Animationr Performance AccessibilityzFramework Compatibility Complexityr zCSS/Technical KeywordszImplementation ChecklistzDesign System Variables)file search_cols output_colsz colors.csv Product TypeNotes)rPrimaryz On Primary Secondaryz On SecondaryAccentz On Accent Background ForegroundCardzCard ForegroundMutedzMuted ForegroundBorder DestructivezOn DestructiveRingrz charts.csv) Data Typer Best Chart Type When to UseWhen NOT to UseAccessibility Notes) rr r zSecondary Optionsr!r"zData Volume ThresholdzColor GuidancezAccessibility Grader#z A11y FallbackzLibrary RecommendationzInteractive Levelz landing.csv) Pattern Namer Conversion Optimization Section Order)r$r r&zPrimary CTA PlacementzColor Strategyr%z products.csv)rr Primary Style RecommendationzKey Considerations)rr r'zSecondary StyleszLanding Page PatternzDashboard Style (if applicable)zColor Palette Focuszux-guidelines.csv)CategoryIssue DescriptionPlatform) r(r)r+r*DoDon'tzCode Example GoodzCode Example BadSeverityztypography.csv)Font Pairing Namer(Mood/Style Keywordsr Heading Font Body Font) r/r(r1r2r0r Google Fonts URLz CSS ImportzTailwind Configrz icons.csv)r( Icon Namer r )r(r4r Libraryz Import CodeUsager Stylezreact-performance.csv)r(r)r r*zapp-interface.csvzgoogle-fonts.csv)Familyr(StrokeClassificationsr Subsets Designers) r8r(r9r:Stylesz Variable Axesr;r<zPopularity Rankr3) stylecolorchartlandingproductux typographyiconsreactweb google-fontsz react-nativerzstacks/react-native.csv)r( Guideliner*r,r-) r(rIr*r,r-z Code GoodzCode Badr.zDocs URL)rrc*eZdZdZddZdZdZdZy)BM25z&BM25 ranking algorithm for text searchc||_||_g|_g|_d|_i|_t t|_d|_ y)Nr) k1bcorpus doc_lengthsavgdlidfrint doc_freqsN)selfrMrNs 8/home/jay/.agents/skills/frontend-design/scripts/core.py__init__z BM25.__init__\s@  $S)ctjddt|j}|j Dcgc]}t |dkDs|c}Scc}w)z8Lowercase, split, remove punctuation, filter short wordsz[^\w\s] )resubstrlowersplitlen)rVtextws rWtokenizez BM25.tokenizefsCvvj#s4y'89::<6a3q6A:666s AAc|Dcgc]}|j|c}|_t|j|_|jdk(ry|jDcgc] }t|c}|_t |j|jz |_|jD]@}t}|D]/}||vs|j|xxdz cc<|j|1B|jjD]6\}}t|j|z dz|dzz dz|j|<8ycc}wcc}w)zBuild BM25 index from documentsrNg?) rerOrbrUrPsumrQsetrTadditemsrrR)rV documentsdocseenwordfreqs rWfitzBM25.fitks5>?ct}}S)? T[[! 66Q; 04 <CH<))*TVV3 ;; #C5D #t#NN4(A-(HHTN # #....0 KJD$ $&&4-#"5$*!E!IJDHHTN K@=s EEc|j|}g}t|jD]\}}d}|j|}t t }|D]} || xxdz cc<|D]~} | |j vs|| } |j | } | |jdzz} | |jd|jz |j|z|jz zzz}|| | z|z z }|j||ft|ddS)z!Score all documents against queryrrgc |dS)Nrg)xs rWzBM25.score..s AaDrYT)keyreverse) re enumeraterOrPrrSrRrMrNrQappendsorted)rVquery query_tokensscoresidxrmscoredoc_len term_freqsrotokentfrR numerator denominators rWrz BM25.score~s }}U+ !$++. (HCE&&s+G$S)J &4 A%  && ;DHH$#E*B((5/C "dggk 2I"$tww!dff*tvv?ORVR\R\?\2\']"]KS9_{::E  ; MM3, ' ("f.$??rYN)g?g?)__name__ __module__ __qualname____doc__rXrerqrrtrYrWrKrKYs07 K&@rYrKct|dd5}ttj|cdddS#1swYyxYw)z!Load CSV and return list of dictsrzutf-8)encodingN)openlistcsv DictReader)filepathfs rW _load_csvrs6 hg .'!CNN1%&'''s 7Ac |jsgSt|}|Dcgc]djfd|D}}t}|j ||j |} g} | d|D]E\} } | dkDs || | j |D cic]} | vs| j| dc} G| Scc}wcc} w)zCore search function using BM25r[c3TK|]}tj|d!yw)N)r_get).0colrows rW z_search_csv..s!GC#cggc2./Gs%(Nrr)existsrjoinrKrqrrzr)rrrr| max_resultsrrrlbm25rankedresultsrrrs ` rW _search_csvrs ??  X DTXXCG;GGXIX 6DHHY ZZ FG\k*Y U 19s)C NN[WcCSVJCb!11W XY NYXs!C C $C c (|jgdgdgdgdgdgdgdgdgd gd gd d }|jDcic]\}}|tfd |D}}}t||j}||dkDr|SdScc}}w)z/Auto-detect the most relevant domain from query) r?palettehex#rgbrsemanticaccent destructivemuted foreground) r@graph visualizationtrendbarpiescatterheatmapfunnel)rApagecta conversionhero testimonialpricingsection)Rsaas ecommercez e-commercefintech healthcaregaming portfoliocrypto dashboardfitness restauranthoteltravelmusic educationlearninglegal insurancemedicalbeautypharmacydentalpetdatingweddingrecipedeliveryridebookingcalendartimertrackerdiarynotechat messengercrminvoiceparkingtransitvpnalarmweathersleep meditationfastinghabitgrocerymemewardrobez plant carereading flashcardpuzzletriviaarcade photography streamingpodcast newsletter marketplace freelancer coworkingairlinemuseumtheaterchurchz non-profitcharity kindergartendaycarez senior care veterinaryfloristbakerybrewery construction automotivez real estate logistics agriculturezcoding bootcamp)r>designui minimalism glassmorphism neumorphism brutalismz dark modeflataurorapromptcssimplementationvariable checklisttailwind) rC usability accessibilitywcagtouchscroll animationkeyboard navigationmobile)z font pairingztypography pairingz heading fontz body font)z google fontz font familyz font weightz font stylez variable fontnotozfont forz find fontz font subsetz font languagezmonospace fontz serif fontzsans serif fontz display fontzhandwriting fontfontrDserifsans)iconrElucide heroiconssymbolglyph pictogramzsvg icon)rFznext.jsnextjssuspensememo usecallback useeffectrerenderbundle waterfallbarrelzdynamic importrsczserver component) ariafocusoutliner virtualize autocompleteformz input type preconnect) r?r@rArBr>rCrDrHrErFrGc3K|]6}tjdtj|zdzs3d8yw)z\brgN)r]searchescape)rkw query_lowers rWrz detect_domain..s2h"))EBIIbM)r`rkrhmaxr)r|domain_keywordsdomainkeywordsr~bestr?s @rW detect_domainrEs++-K~khM~Ydezy OBQBWBWBYZm}msu}fchhhhhZFZ v6:: &D$rzFile not found: )errorrBrr)rBr|rcountr)rE CSV_CONFIGrDATA_DIRrrrb)r|rBrconfigrrs rWr<r<s ~u% ^^FJw$7 8F&.(H ?? +H:6&II(F=$96-;PRWYdeGvW  rYc|tvrdd|ddjtiStt|dz }|j sd||dSt |t dt d ||}d ||t|dt||d S) z Search stack-specific guidelinesrGzUnknown stack: z . Available: z, rzStack file not found: )rGstackrrrM)rBrMr|rrHr) STACK_CONFIGrAVAILABLE_STACKSrJrr _STACK_COLSrb)r|rMrrrs rW search_stackrQs L ?5'tyyIY?Z>[\]],u-f55H ?? 1( M@Z\acnoGU#F+W  rY)rrr]pathlibrmathr collectionsr__file__parentrJ MAX_RESULTSrIrNrPrkeysrOrKrrrEr<rQrtrYrWrYs # > ' '& 0 _c &0E |T _K iz $GL  !xw Ht (GL $GL  #neg8 vV67 K{  ))+,;@;@~' 21.;,,7rY