o h+@sddlmZmZddlmZddlmZmZmZm Z ddl Z ddl Z ddlmZddlmZddlmZddlmZmZed d d Zed d ZdaGdddeZGdddeZGdddeZGdddeZddZedddZ e!dddZ"e!ddd Z#ej$d!ed"d#efd$d%Z%ej$d&ed"d#efd'd(Z&Gd)d*d*eZ'ej$d+ee(e fd"d#e'fd,d-Z)e!d.d/d0Z*dS)1)FastAPI HTTPException) BaseModel)ListOptionalDictAnyN)RecommenderLogger MABSimulation)UnifiedRecommendationEnv)DataController DataManagerMAB Recommendation API1.0.0)titleversionapi)namec@sNeZdZUeed<dZeeed<dZeeed<iZ ee ee fed<dS)RecommendationRequestuser_idn_recommendationsepsilon_greedyalgorithm_typealgorithm_paramsN) __name__ __module__ __qualname__int__annotations__rrrstrrrrr"r"H/Users/divyeshpatel/Desktop/sahana/Recommender/recommender_rl/app/api.pyrs rc@s6eZdZUeed<eed<eed<dZeeed<dS)FeedbackRequestritem_id event_typeN timestamp)rrrrr r!r'rr"r"r"r#r$s r$c@s>eZdZUeed<eeed<eed<dZeee ed<dS)RecommendationResponserrecommended_itemsrNconfidence_scores) rrrrr rr!r*rfloatr"r"r"r#r(s  r(c@s6eZdZUeed<eed<dZeeeefed<dS)FeedbackResponsestatusmessageNupdated_rewards) rrrr!r r/rrrr"r"r"r#r,%s r,c Csz\z$ddl}|j|j|j|jtd}td|iddddd }Wn&tyK}zt d t |Gd d d }|}WYd}~nd}~wwt |d ddida t dWdStyy}ztdt |WYd}~dSd}~ww)z*Initialize the MAB simulation environment.rNzenhanced_recommender.dbdb_path2r*T)Z data_sourcemax_arms n_suggestionsseedZuse_user_contextz!Failed to create DB environment: c@$eZdZddZddZddZdS)z.initialize_simulation..MockEnvironmentcSd|_d|_d|_dSNr1rr3r4Z n_productsselfr"r"r#__init__= z7initialize_simulation..MockEnvironment.__init__cS ddiifSNrrr"r:r"r"r#resetA z4initialize_simulation..MockEnvironment.resetcSiddddgifSNgFZ event_typesr"r;actionr"r"r#stepCz3initialize_simulation..MockEnvironment.stepNrrrr<r@rFr"r"r"r#MockEnvironment<s rIrepsilon皙?rrz'MAB Simulation initialized successfullyz!Failed to initialize simulation: F)ospathjoindirnameabspath__file__r Exceptionloggerwarningr!r simulationinfoerror)rMr0envZ env_errorrIer"r"r#initialize_simulation*s2&   r[startupc stdtsZtdtdz'ddlm}ddlm}Gddd}|}||d d d id atd WdSt yY}ztdt |daWYd}~dSd}~wwdS)z!Initialize simulation on startup.zStarting API initialization...z*Failed to initialize simulation on startupz%Attempting fallback initialization...r)EpsilonGreedyMABr c@r6)z&startup_event..MockEnvironmentcSr7r8r9r:r"r"r#r<`r=z/startup_event..MockEnvironment.__init__cSr>r?r"r:r"r"r#r@erAz,startup_event..MockEnvironment.resetcSrBrCr"rDr"r"r#rFhrGz+startup_event..MockEnvironment.stepNrHr"r"r"r#rI_s rIrrJrKrLz,Fallback simulation initialized successfullyz%Fallback initialization also failed: N) rTrWr[rXZrl_recommender.mabalgorithmsr]rl_recommender.simulationr rVrSr!)r]r rIrYZfallback_errorr"r"r# startup_eventPs$      r_/cCsddddddddS) z#Root endpoint with API information.rrzGet recommendations for a userz#Submit feedback for recommendationszGet API and simulation statuszConfigure algorithm parameters)zGET /recommend_itemzPOST /feedbackz GET /statuszPOST /configure)r.rZ endpointsr"r"r"r"r#rootrsraz/statusc Cstdur ddddSz%tjtjjjtjjtjjd}ddd |ttjd d ttjd d d dWSt yL}zddt |ddWYd}~Sd}~ww)z1Get the current status of the API and simulation.NrXSimulation not initializedF)r-r.simulation_readytypeclassn_armsr4readyzSimulation readyTr3zN/Ar4)r3r4)r-r.rc algorithm environmentzError getting status: ) rVrcurrent_algorithm __class__rrgr4getattrrjrSr!)algorithm_inforZr"r"r# get_statuss4    roz/recommend_item)Zresponse_modelrequestc Cs2tdur tdddzi|jtjks|jrt|j|jtjtj|j|jd}g}t tjdrT|D]!}|t tjj krMt dt dtjj |}||q1|d q1nd gt |}td t |d |jt|j|tj|d WSty}ztd|jdt|tddt|dd}~ww)z+ Get recommended items for a user. Nrb status_codeZdetail)rirrarm_avg_rewardsrKg?g?gffffff?z Generated z recommendations for user )rr)rr*z*Error generating recommendations for user : z"Error generating recommendations: )rVrrr set_algorithmZ invoke_agentrkrrhasattrlenrtmaxminappendrTrWr(rSrXr!)rpZrecommendationsr*r%Z confidencerZr"r"r#recommend_items>    r|z /feedbackc CsLtdur tdddzv|jg|jg|jg|jdurtjnt|jgd}t |}t |d}t tj drht |j}d|krLttj jkrhnn|jttj j|t tj j|ttj j|d}td |jd |jd |jtd d |dWSty}ztd|jd t|tddt|dd}~ww)z1 Submit feedback for a recommended item. Nrqrbrr)rr%r&r'rtr)r%Zcurrent_avg_rewardZ total_pullsZ total_rewardzProcessed feedback from user z for item rusuccesszFeedback processed successfully)r-r.r/z$Error processing feedback from user zError processing feedback: )rVrrr%r&r'pd TimestampnowZ DataFrameZupdate_algorithmrwrkrrxrtr+Z arm_countsZ arm_rewardsrTrWr,rSrXr!)rpZ event_dataZevent_dfr/Zitem_idxrZr"r"r#feedbacks<       "rc@s.eZdZUeed<iZeeeefed<dS)ConfigureRequestrrN) rrrr!r rrrrr"r"r"r#rs rz /configurec Cstdur tdddz(t|j|jpi}td|jdd|j|j|jj|j |j ddWSt yS}zt d t |tdd t |dd}~ww) z& Configure the MAB algorithm. NrqrbrrzAlgorithm configured to r}rd)r-r.rnzError configuring algorithm: )rVrrvrrrTrWrlrrgr4rSrXr!)rpZ new_algorithmrZr"r"r#configure_algorithms$   rz/healthcCsdtjdS)zHealth check endpoint.Zhealthy)r-r')r~rr isoformatr"r"r"r# health_check'sr)+ZfastapirrZpydanticrtypingrrrrZpandasr~Z gymnasiumZgymZrl_recommender.loggerr r^r Z'rl_recommender.UnifiedRecommendationEnvr Zdbr rapprTrVrr$r(r,r[Zon_eventr_getraropostr|rrr!rrr"r"r"r#s@      & !   $ ./