KCjdZddlmZddlmZddlZddlmZddlm Z ddl m Z eZ e e Zdd d d d d dZGddZeZy)u ML model loader. Production rule: if a model fails to load at startup, the application MUST refuse to start, not start in a half-working state and fail on the first real request. A health check that says "I'm up" while secretly missing the yield model is worse than a crash — it's a silent SLA violation the client discovers from a support ticket, not from monitoring. All 6 models are loaded once, here, and held in memory for the lifetime of the worker process. They are never reloaded per-request. )Path)AnyN) get_settings)ModelNotLoadedError) get_loggerzsarimax_climate.pklzxgb_feasibility.pklz xgb_yield.pklz dt_trend.pklzrf_recommendation.pklzsarimax_market.pkl)sarimax_climatexgb_feasibility xgb_yielddt_trendrf_recommendationsarimax_marketcreZdZdZd dZd dZd dZd dZdede fd Z e de fd Z e deefd Zy) ModelRegistryz?Holds every loaded model in memory and exposes typed accessors.returnNc\i|_d|_ttj|_y)NF)_models_loadedrsettings MODEL_DIR _model_dirselfs GC:\Crop_Prediction\Backend\crop-ai-system\app\models_loader\registry.py__init__zModelRegistry.__init__'s"')  x112ctjd|jdg}g}tj D]}\}}|j|z }|j s|j t|@ tj||j|<tjd|d||rtdd||rtdd ||jd |_tjd t|jd y#t$r0}|j t|t|fYd}~d}~wwxYw) NzLoading models from z ...zLoaded model 'z' from z;Required model files are missing. The service cannot start.zMissing files: messagedetailzAOne or more model files failed to load. The service cannot start.zFailed to load: TzAll z models loaded successfully)loggerinforREQUIRED_MODEL_FILESitemsexistsappendstrjoblibloadr Exceptionr_validate_loaded_modelsrlen)rmissingfailedkeyfilenamepathexcs rload_allzModelRegistry.load_all,sC *4??*;4@A1779 5MC??X-D;;=s4y) 5$*KK$5 S! nSE?@ 5 %U( 2  %[)&2  $$&  d3t||,--HIJ! 5 s4y#c(344 5s==D$$ E-%EEc Lgd}|D]M}|jj|}t|trd|vs3t dd|dt |dD]I}|jj|}t|tr/t dd|dt |y ) a3 Sanity-check the structure of each loaded artifact. Catches the case where a .pkl file loads without raising an exception but doesn't contain what the inference code expects (e.g. wrong dict shape from a bad training run accidentally copied to the model directory). )r r r r modelz2A model file is present but has an invalid format.'z&' expected dict with 'model' key, got r)rr z,' expected dict of {key: fitted_model}, got N)rget isinstancedictrtype)r bundle_modelsr.objs rr*z%ModelRegistry._validate_loaded_modelsKsZ   C,,""3'Cc4(G3,>)Pse#I$s)U 9 C,,""3'Cc4()Pse#QRVWZR[Q\] rc6|js tddy)Nz2Models are not yet loaded. Service is starting up.z2ModelRegistry accessed before load_all() completedr)rrrs r_ensure_loadedzModelRegistry._ensure_loadedcs!||%LK rr.c||j||jvrtd|d|j|S)NzModel 'z' was not loaded)r)r=rr)rr.s rr6zModelRegistry.getjs>  dll "%u0!s!s! $)49))rr)rIpathlibrtypingrr'app.core.configrapp.core.exceptionsrapp.core.logging_configrrrFr r"rmodel_registryrLrrrSsa  (3. > H -, 0* R)R)nr