o zj@sddlmZmZmZddlmZddlmZmZddl m Z m Z m Z m Z ddlZddlmZddlmZmZmZmZeZGdd d eZejd d d dd eedeefddZejdd d ddZejddd defddZejdd d d efddZdS)) APIRouter HTTPExceptionDepends)get_market_summary_ai)get_all_prices get_scraper) AgentTypeMessage coordinatorcontext_protocolN) BaseModel)OptionalListDictAnyc@sNeZdZUeed<eed<eed<eed<dZeeed<dZeeed<dS) MarketListingcropquantityprice_expectedcontactBaramatilocationNnotes) __name__ __module__ __qualname__str__annotations__intrr rrr;/var/www/html/CropPilot/backend/app/api/endpoints/market.pyr s rz/prices) status_coderrc szMtt}t|d||pddtttj tj ||pddddd|idId H}|r4d |j vrE|r=|j d d nd }t d |dd|j dgiWStyi}ztd|t d t|dd }~ww)zLReturns market prices from scraped sources for crops in specified locations. market_pricesr) request_typerrT)rr force_refresh get_prices session_idsenderreceivercontent message_typecontextNerrorzUnknown error fetching pricesNo response from market agentr"detail market_datapriceszError fetching market prices: ruuiduuid4r set_contextr route_messager r COORDINATORMARKET_ANALYZERr+getr Exceptionprint)rrr'messager.errr get_market_pricess:   rAz/summaryc szCtt}t|dditttj tj idd|idIdH}|r*d|j vr;|r3|j ddnd }t d |d d |j d d iWStyb}ztd|t d dt|d d}~ww)zMGenerates a brief AI summary of current market data using multi-agent system.r$market_summary get_summaryr'r(Nr.z Unknown error generating summaryr/r0r1summaryzNo market summary availablezError getting market summary: z#Could not generate market summary: r5)r'r?r.r@rrr get_market_summary;s,    rEz /listingslistingc sz3ttdd}tt}|}||d<d|d<td||td|d||d WStyO}ztd |td t|d d}~ww) z9Allows farmers to create market listings for their crops.Nidz User Listingtypelisting_zReceived listing: zListing created successfully)r? listing_iddetailszError creating market listing: r0r1) rr6r7dictr r8r>r=r)rGnew_idr' listing_dictr@rrr add_market_listingXs$ rQz/trends/{crop}c szLtt}t|d|dtttj tj d|idd|idIdH}|r-d|j vr>|r6|j dd nd }t d |d |j d d|j dgdWStyh}ztd|t d t|d d}~ww)z*Analyze market trends for a specific crop. market_trends)r$rranalyze_trendsr'r(Nr.zUnknown error analyzing trendsr/r0r1r?zNo trend data availablehistorical_data)r?rTzError analyzing market trends: r5)rr'r?r.r@rrr get_market_trendsxs0      rU)NN)fastapirrrapp.core.ai_servicesrapp.services.market_scraperrrapp.core.multi_agentrr r r r6pydanticr typingr rrrrouterrr<rrArEpostrQrUrrrr s"    &