q(ibddlZddlZddlmZmZddlmZeZGddZdadefdZ y)N)ListDict) get_configceZdZdZdZdeefdZddeede deefdZ dd ede deefd Z dd e de deefd Z dd ede deefdZ ddeed ede d ede deef dZde defdZy)RecipeSearchServicez$Search service for RecipeNLG datasetc.|j|_y)z-Initialize search service with sample recipesN)_load_sample_recipesrecipesselfs I/Users/apple/Desktop/TU/AI/AI-Recipe-Generator/services/search_service.py__init__zRecipeSearchService.__init__ s002 returnc ~ddgddddddd d d gd d ddd dd ddgddddddd ddgddddd dd ddgddddddd gS)z Load sample recipes (in production, load from RecipeNLG dataset) For now, using hardcoded sample data zClassic Tomato Pasta)pastatomatogarlic olive oilbasiluYBoil pasta. Sauté garlic in olive oil. Add tomatoes. Combine with pasta. Top with basil.luncheasy)idtitle ingredients instructions meal_type cooking_timeservings difficultyzGrilled Chicken Salad)chickenlettucercucumberrlemonzDGrill chicken. Chop vegetables. Mix with olive oil and lemon. Serve.zMushroom Risotto)ricemushroomonionrbrothparmesanuWSauté onion and garlic. Add rice. Gradually add broth. Stir in mushrooms and parmesan.dinner(mediumzVegetable Stir Fry)broccolicarrotz bell pepperz soy saucergingerzCHeat oil. Stir fry vegetables. Add soy sauce and ginger. Serve hot.Pancakes)floureggmilkz baking powderbutterz maple syrupzLMix dry ingredients. Add wet ingredients. Cook on griddle. Serve with syrup. breakfastr s r r z(RecipeSearchService._load_sample_recipess/R {$ "$ 0a f$ "$ +[ y% "& -e e% "$ #a n( "$ S3 3 rNrlimitc |tj}|Dcgc]}|j}}g}|jD]Q}|dDcgc]}|jc} t fd|D}|dkDs>|j ||dS|j dd|d|Dcgc]}|d  c}Scc}wcc}wcc}w) a Search recipes by ingredients Args: ingredients (list): List of ingredients to search limit (int): Maximum number of results Returns: list: Matching recipes sorted by relevance Nrc3NK|]tfdDsdyw)c3&K|]}|v ywNr>.0r_ingings r zFRecipeSearchService.search_by_ingredients...^s@nRW@nsrNany)rErGrecipe_ingredientss @r rHz.^soS@n[m@n=n!os%%r)recipescorec |dS)NrMr>)xs r z;RecipeSearchService.search_by_ingredients..gs !G*rT)keyreverserL)configMAX_SEARCH_RESULTSlowerr sumappendsort) r rr?rGsearch_ingredientsscored_recipesrLmatchesitemrKs @r search_by_ingredientsz)RecipeSearchService.search_by_ingredientsIs =--E6AAcciikAAll F9? 9N!O##))+!O o'9ooG{%%$$'    4dC+9&5+AB4XBB!B "PCsB= C. Cr c|tj}|jDcgc])}|dj|jk(s(|+}}|d|Scc}w)a Search recipes by meal type Args: meal_type (str): Type of meal (breakfast, lunch, dinner) limit (int): Maximum number of results Returns: list: Recipes matching the meal type Nr rSrTr rU)r r r?rresultss r search_by_meal_typez'RecipeSearchService.search_by_meal_typejsX =--E"llZa n.B.B.D HY.Y1ZZv[ )A Amax_timec~|tj}|jDcgc] }|d|ks |}}|d|Scc}w)a Search recipes by maximum cooking time Args: max_time (int): Maximum cooking time in minutes limit (int): Maximum number of results Returns: list: Recipes within time limit Nr!)rSrTr )r rdr?r`ras r search_by_cooking_timez*RecipeSearchService.search_by_cooking_time{sI =--E"llLa.?8.K1LLvMs ::r#c|tj}|jDcgc])}|dj|jk(s(|+}}|d|Scc}w)a  Search recipes by difficulty level Args: difficulty (str): Difficulty level (easy, medium, hard) limit (int): Maximum number of results Returns: list: Recipes matching difficulty Nr#r_)r r#r?r`ras r search_by_difficultyz(RecipeSearchService.search_by_difficultysZ =--E"ll\a o.C.C.EIYIYI[.[1\\v]rcmax_cooking_timec, |tj}|jj}|rB|Dcgc]}|j }}|D cgc] t fd|Ds }} |r4|D cgc])} | dj |j k(s(| +}} |r|D cgc] } | d|ks | }} |r4|D cgc])} | dj |j k(s(| +}} |d|Scc}wcc} wcc} wcc} wcc} w)a{ Advanced search with multiple filters Args: ingredients (list): Required ingredients meal_type (str): Type of meal max_cooking_time (int): Maximum cooking time difficulty (str): Difficulty level limit (int): Maximum results Returns: list: Filtered recipes Nc3NK|]tfddDyw)c3BK|]}|jvywrC)rUrDs r rHz@RecipeSearchService.advanced_search...sGUC5;;=(GsrNrI)rErGr`s @r rHz6RecipeSearchService.advanced_search..s)1Ga 6FGG1s!%r r!r#)rSrTr copyrUrJ) r rr rir#r?rarGrYr`s `r advanced_searchz#RecipeSearchService.advanced_searchs) =--E,,##% 9D!E##))+!E !E")QS1-1.qG ")YQQ{^-A-A-CyGX-XqYGY ")SQQ~->BR-RqSGS ")[QQ|_-B-B-D HXHXHZ-Zq[G[v%"FZT\s5C=D,D8)D"D. D <D )D2D recipe_idc>|jD]}|d|k(s |cSy)zGet a specific recipe by IDrN)r )r rorLs r get_recipe_by_idz$RecipeSearchService.get_recipe_by_ids+ll Fd|y(  rrC)NNNNN)__name__ __module__ __qualname____doc__rrrr strintr]rbrfrhrnrqr>rr rrs.38 d4j8 tCcC3CRVW[R\CBST "s3$t*"s3$t*"OSGK$()49))),)AD)!)-1$Z)V#$rrrc.t tatS)z%Get or create search service instance)search_servicerr>rr get_search_servicerzs,. r) jsonostypingrrrSrrryrzr>rr r~s9  EER/r