o i@sHddlmZmZmZddlZddlmZmZddlZddl m Z ddl m Z m Z mZddlmZmZmZmZmZddlmZddlZdd lmZGd d d e ZeeZed efd dZeedZe dddZ!e"dddZ#e$ddefddZ%e"dddZ&e"dde'fdd Z(e"d!de'fd"d#Z)dS)$)time localtimestrftimeN)FastAPI HTTPException) BaseModel)ListDictAny) load_modelpredictupdate_model_dataget_player_data model_retarin)asynccontextmanager) setup_loggingc@seZdZUeed<dS)PredictionRequestgwN)__name__ __module__ __qualname__int__annotations__rr//var/www/html/fantasy/fantasy_model/app/main.pyrs rappc Csxttdzt}|stdtdWnty1}z td|tdd}~wwdVtddS)zG Lifespan event to load the model when the application starts. z1Starting FastAPI application and loading model...zModel loading failed.z$Failed to load the model on startup.zError loading model: z6Application startup failed due to model loading error.Nz'FastAPI application is shutting down...)rloggerinfor error RuntimeError Exception)rmodelerrrlifespans   r$)r$httpcstt}t}tjd||jt|jt|jdt dt |dd||IdH}t|}tjd||j |ddd|S) NzIncoming Request)methodurlclientz%Y-%m-%d %H:%M:%S) request_idrequest timestamp)extrazRequest Finished) status_code process_time)r)response) struuidZuuid4rrrr&r'r(rrr-)r*Z call_nextr)Z start_timer/r.rrr log_requests)s(     r2/cCsddiS)Nmessagez"Welcome to the FPL Prediction API!rrrrr read_rootDsr5z/predictr*c Cstd|j|jstdtddd|j}td|dzt|}td|d |jd d WStyS}ztd |tdt|dd }~ww)z Endpoint to make predictions using the pre-trained model. :param request: PredictionRequest containing the features for prediction. :return: Dictionary with predictions. z#Get Player XP Predictions for GW : z$No gameweek provided for prediction.iz$No features provided for prediction.)r-Zdetailz#Prediction request received for GW .zPrediction for GW z completed successfully.records)ZorientzPrediction error: N) rrrrrr Zto_dict ValueErrorr0)r*rZ predictionsr#rrrpredict_endpointGs  r9z/updatecCstdttddS)NzStarting Data Updatez#Data Update completed successfully.)rrrrrrr update_datags r:z /gw_data/{gw}rcCs0td|dt|}td|d|S)Nz Fetching GW z Players Dataz% Players Data completed successfully.)rrr)rZ players_datarrrgw_datamsr;z /retrain/{gw}cCs tdt|tddS)NzStarting Model Retrainz%Model Retrain completed successfully.)rrr)rrrrretraints r<)*rrrr1ZfastapirrZpandaspdZpydanticrtypingrr r Z core.modelr r rrr contextlibrloggingZapp.logging_configrr getLoggerrrr$r middlewarer2getr5Zpostr9r:rr;r<rrrrs6