o jEi9"@stddlZddlZddlZddlZddlmZddlmZee Z eZeZgdZ ddZ d dd Z d d ZdS) N)XaiTokenManager)XaiCache)Textbook (Summer)Textbook (Winter)High School AP Textbooks Law SchoolNursing SchoolMedical SchoolCommunity College Gardening Grilling/BBQ ChristmasNew Year/FitnessTax PrepTravel Halloween ThanksgivingRomance/Valentine's Day Year-roundcCs&|s tddSd|d|d|}t|}|r)td|d|d|Sts8td|d dSd |d |d |d |rJ|dkrJ|ndd|rU|dkrU|ndddtd }td|dtd|d|dddd|dgdddd} d |d!d"} z`t j d#d$O} | j d%| | d&} | | } | d'd(d)d*}|tvrtd+|d,|t|||Wd-WStd.|d  Wd-WdS1swYWd-St jt jt jttfy}ztd/|d0|WYd-}~dSd-}~ww)1z] Queries the XAI API to classify seasonality, now with caching and token management. z9XAI API key is not provided. Skipping LLM classification.rz seasonality:|z5XAI Cache HIT for seasonality. Found classification 'z ' for title 'z'.z*XAI daily limit reached. Cannot classify 'z'. Defaulting to Year-round.z Based on the following book details and historical sales data, choose the single most likely sales season from the provided list. Respond with ONLY the name of the season from the list. **Book Details:** - **Title:** "z" - **Category:** "z" - **Publisher:** "zF" **Sales Data Insights:** - **Inferred Peak Price Month:** "-zN/Az*" - **Inferred Trough Price Month:** "z" **Season List:** z, z z"XAI Seasonality Request for ASIN 'z' (Cache MISS):z - Peak: z , Trough: systemzYou are a book classification expert. Your task is to select the most appropriate seasonal category for a book from a given list.)rolecontentuserzgrok-4-fast-reasoningg?2)messagesmodel temperature max_tokenszBearer zapplication/json) Authorizationz Content-Typeg>@)timeoutz$https://api.x.ai/v1/chat/completions)headersjsonchoicesrmessagerzLLM classified 'z' as: Nz)LLM returned an invalid classification: 'z"XAI API request failed for title 'z': )loggerwarning xai_cachegetinfoxai_token_managerrequest_permissionjoinSEASON_CLASSIFICATIONShttpxClientpostraise_for_statusr$stripsetHTTPStatusError RequestErrorJSONDecodeErrorKeyError IndexErrorerror)titlecategories_sub manufacturerpeak_season_strtrough_season_strapi_key cache_key cached_resultpromptpayloadr#clientresponsedata llm_choiceerK=/var/www/agentarbitrage/keepa_deals/seasonality_classifier.py_query_xai_for_seasonalitysn       ( rMc s|sd}|sd}|s d}|}|}|gd}tfdd|D} d|vr3d|vs1| r3dSd|vs=d |vs=| rKd |vrCd Sd |vrId Sd Sd|vsSd|vrUdSd|vs]d|vr_dSd|vsgd|vridSd|vrodSd|vswd|vrydSd|vsd|vsd|vrdSd|vsd |vrd!|vrd"Sd#|vsd$|vsd%|vsd&|vrd'|vrd(Sd)|vrd*|vsd+|vrd,Sd-|vsd-|vrd.Sd/|vrd0Sd1|vrd2Sd3|vsd4|vrd5Std6|d7t||||||} | S)8a< Classifies a book's seasonality based on heuristics and an XAI model, now enriched with inferred sales data. Args: title (str): The title of the book. categories_sub (str): The sub-category string. manufacturer (str): The manufacturer/publisher. peak_season_str (str): The inferred month of peak prices (e.g., "Jan"). trough_season_str (str): The inferred month of trough prices (e.g., "Jul"). xai_api_key (str, optional): The API key for the XAI service. Returns: str: The classified season. )cengagez mcgraw-hillpearsonwiley macmillansagec3s|]}|vVqdSNrK).0pub mfr_lowerrKrL sz'classify_seasonality..apz high schoolrcollege universitysummerrwinterrz law schoolzbar examrnursingnclexr zmedical schoolmcatr zcommunity colleger gardeningr grillingbbqbarbecuer christmasholidaygiftrznew yearfitnessdietz self-help resolutionrtaxprepguidertravelr halloweenr thanksgivingr valentineromancerz)Heuristics resulted in 'Year-round' for 'z/'. Querying XAI with sales data for refinement.)loweranyr'r+rM) r<r=r>r?r@ xai_api_key title_lower cat_lowertextbook_publishersis_textbook_publisher llm_resultrKrWrLclassify_seasonalitygs^( r|cCsvidddddddddd d d d d dd dddddddddddddddddd }||d S)!a{ Maps a detailed season classification from the 'Detailed_Seasonality' column to a specific selling period string for the 'Sells' column. Args: detailed_season (str): The detailed season name. Returns: str: A string representing the peak selling period (e.g., "Nov-Dec"). Returns "All Year" if no specific period is applicable. rz Jul - Seprz Jan - FebrrOctr z Mar - Aprr z Aug - Sepr zSep - Oct & Jan - Febr r z May - Junrz Nov - DecrJanrz Jan - Aprrz May - Augrz Sep - OctrNovrFebrzAll Year)r*)detailed_season period_maprKrKrLget_sells_periodsH       rrT)rer0r$loggingr,rr)r getLogger__name__r'r/rMr|rrKrKrKrLs    O P