q(iTddlZddlmZmZddlmZeZGddZdadZy)N) GPT2TokenizerGPT2LMHeadModel) get_configc,eZdZdZdZddZddZdZy)RecipeGeneratorz#GPT-2 based recipe generation modelctj|_tj|_ t j |j|_tj |j|_ |jj|j|jjtd|jy#t$r&}td|d|_ d|_Yd}~yd}~wwxYw)z"Initialize the model and tokenizeru✓ Recipe model loaded on u✗ Error loading model: N)configDEVICEdevice MODEL_NAME model_namerfrom_pretrained tokenizerrmodeltoevalprint Exception)selfes E/Users/apple/Desktop/TU/AI/AI-Recipe-Generator/models/recipe_model.py__init__zRecipeGenerator.__init__ smm  ++ "*::4??KDN(88IDJ JJMM$++ & JJOO  / }= > " -aS1 2DJ!DNN "sBC C;C66C;c|j |jy dj|}d|}|r|d|z }|dz }|jj|dj |j }t j5|jj||jd |zd tjtjd d |jjt j| }ddd|jjd d }|S#1swY+xYw#t $r&} t#d| dt%| cYd} ~ Sd} ~ wwxYw)a Generate a recipe based on ingredients and meal type Args: ingredients (list): List of ingredients (e.g., ["chicken", "garlic", "tomato"]) meal_type (str): Type of meal (e.g., "breakfast", "lunch", "dinner") num_words (int): Approximate number of words to generate Returns: str: Generated recipe text Nz0Model not available. Please check configuration.z, z Recipe with z for z: pt)return_tensorsT) max_lengthnum_return_sequences temperaturetop_p do_sampleno_repeat_ngram_size pad_token_idattention_maskr)skip_special_tokenszError generating recipe: z"Error: Could not generate recipe. )rrjoinencoderr torchno_gradgenerateshaper TEMPERATURETOP_P eos_token_id ones_likedecoderrstr) r ingredients meal_type num_wordsingredients_strprompt input_ids output_idsrecipers rgenerate_recipezRecipeGenerator.generate_recipesY :: !7E" A"ii 4O#O#45FE)-- eOF--fT-JMMdkkZI !ZZ00(q1I=)* & 2 2 ,,")*!%!|jduxr|jduS)z&Check if model is loaded and availableN)rr)rs r is_availablezRecipeGenerator.is_available`szz%D$..*DDrAN))__name__ __module__ __qualname____doc__rr;r@rCrArrrs-" 1Af&ErArc.t tatS)z6Get or create recipe generator instance (lazy loading))recipe_generatorrrJrArget_recipe_generatorrMhs*, rA) r) transformersrrr rrrLrMrJrArrOs1 7 [E[E~rA