o zj]@s6dZddlZddlmZmZmZmZmZmZddl Z ddl Z ddl Z ddl m Z ddlmZmZddlZddlmZmZmZmZmZddlmZmZmZmZmZddlmZmZdd l m!Z!e!Z"e j#e j$d e %e&Z'Gd d d eZ(Gd ddeZ)GdddeZ*GdddeZ+GdddeZ,ddZ-dS)zx Specialized agents for the CropPilot multi-agent system. Each agent is responsible for a specific domain of expertise. N)DictListAnyOptionalCallableUnion)Image)datetime timedelta)Agent AgentTypeMessage coordinatorcontext_protocol)get_disease_predictionget_yield_estimateprocess_voice_command_aiget_market_summary_airun_gemini_text_prompt)get_all_prices get_scraper) get_settings)levelcLeZdZdZfddZdedeefddZdedeefdd ZZ S) DiseaseDetectorAgentz8Agent specializing in crop disease detection from imagescs.ttj|d|j|d|jdS)N analyze_image get_treatment)super__init__r DISEASE_DETECTORregister_handlerhandle_analyze_imagehandle_get_treatmentself __class__2/var/www/html/CropPilot/backend/app/core/agents.pyr#szDiseaseDetectorAgent.__init__messagereturnc szE|jd}|st|j|jddiddWSt|IdH}|ds8|jr,|jdnd}|r8t |d|it|j|jd |id |jd WSt yq}zt d |t|j|jdd t |iddWYd}~Sd}~ww)z%Analyze image to detect crop diseases image_byteserrorzNo image data providedsenderreceivercontent message_typeNzError: session_idlast_disease_detectionresultanalysis_resultr.r/r0r1contextzError in disease detection: zError analyzing image: )r0getr agent_typer.r startswithr7rupdate_context Exceptionloggerr,str)r$r)r+r4 context_ider'r'r(r!(sD  z)DiseaseDetectorAgent.handle_analyze_imagec s|jd}|jdd}|jr|jdnd}d}|r*t|}|r*|dd}z!d|d |d }t|IdH}|} t|j|jd | id |jd WSt yv} zt d| t|j|jddt | iddWYd} ~ Sd} ~ ww)z2Get treatment recommendations for detected disease disease_name crop_type unknown cropr2Nenlanguagez7Provide organic and conventional treatment options for z in zk. Include locally available options common in Maharashtra, India. Keep it brief and practical. treatmenttreatment_recommendationr6z(Error getting treatment recommendation: r,r-) r0r8r7r get_contextrr r9r.r<r=r,r>) r$r)rArBr?rEr7promptr4rFr@r'r'r(r"Vs<   z)DiseaseDetectorAgent.handle_get_treatment) __name__ __module__ __qualname____doc__rr rr!r" __classcell__r'r'r%r(r s  .rc6eZdZdZfddZdedeefddZZS)YieldPredictorAgentz,Agent specializing in crop yield predictionscs ttj|d|jdS)N predict_yield)rrr YIELD_PREDICTORr handle_predict_yieldr#r%r'r(rszYieldPredictorAgent.__init__r)r*c szF|jd}|st|j|jddiddWSt|IdH}|jr'|jdnd}|r9t|d|j |j |dit|j|jd |id |jd WSt yr}zt d |t|j|jdd t|iddWYd}~Sd}~ww)z+Generate yield prediction from farmer input yield_inputr,zNo yield input data providedr-Nr2last_yield_prediction)croparea predictionr4prediction_resultr6zError in yield prediction: zError making yield prediction: )r0r8r r9r.rr7rr;rB area_sizer<r=r,r>)r$r)rTr4r?r@r'r'r(rSsJ z(YieldPredictorAgent.handle_predict_yield) rJrKrLrMrr rrSrNr'r'r%r(rP~s rPcsbeZdZdZfddZdedeefddZdedeefdd Zdedeefd d Z Z S) MarketAnalyzerAgentz=Agent specializing in market data analysis and price trackingcsHttj|d|j|d|j|d|jd|_g|_ dS)N get_pricesanalyze_trends get_summary) rrr MARKET_ANALYZERr handle_get_priceshandle_analyze_trendshandle_get_summary last_update cached_pricesr#r%r'r(rs  zMarketAnalyzerAgent.__init__r)r*c s zc|jd}|jdd}|jdd}|r'td}|j||dIdH}n|jr,|r:tIdH|_t|_ |j}|j rF|j d nd}|rVt |d |||d it |j|jd |id |j dWSty}ztd|t |j|jddt|iddWYd}~Sd}~ww)z#Get current market prices for cropsrVlocationBaramati force_refreshF agmarknet)districtNr2last_market_query)rVrepricesrk price_datar6zError getting market prices: r,r-)r0r8rr\rdrasyncioget_event_looptimercr7rr;r r9r.r<r=r,r>) r$r)rVrergscraperrkr?r@r'r'r(r`sL  z%MarketAnalyzerAgent.handle_get_pricesc sz|jdd}td|t}g}tdd}tdddD]-}|t |d  d }t d d }|d |dd|} t | d} | || d| }q#|dd} |dd} | | | d} | dkrtd|d| dd} n| dkrd|d| dd} n d|d| dd} |jr|jdnd}|rt|d|| |d it|j|j| |d!d"|jd#WSty}z%td$|d%|t|j|jd&d'|d%t|id&d(WYd}~Sd}~ww))z4Analyze market trends for specific crops (Simulated)rVrCz)Generating simulated trend analysis for: iir)daysz%Y-%m-%dggQ?gMbP?)datepricerwdz Prices for zC have shown a noticeable upward trend over the past month (approx. z.1fz % increase).zE have shown a noticeable downward trend over the past month (approx. z % decrease).z@ have remained relatively stable over the past month (change of z%).r2Ntrend_analysis_)summaryhistorical_data)r)r}trend_analysis_resultr6z&Error generating simulated trends for z: r,zError analyzing trends for r-)r0r8r=infor nowrandomrandintranger strftimeuniformroundappendr7rr;r r9r.r<r,r>)r$r)rVtodayhistorical_prices base_priceirvprice_fluctuationrw start_price end_pricetrend_percentage trend_summaryr?r@r'r'r(rasb      z)MarketAnalyzerAgent.handle_analyze_trendsc sz(|jstIdH|_t|_t|jIdH}t|j|j d|id|j dWSt yT}zt d|t|j|j ddt|iddWYd}~Sd}~ww)z*Get a summary of current market conditionsNr|market_summaryr6zError getting market summary: r,r-)rdrrmrnrorcrr r9r.r7r<r=r,r>)r$r)r|r@r'r'r(rb+s.z&MarketAnalyzerAgent.handle_get_summary) rJrKrLrMrr rr`rarbrNr'r'r%r(r[s  1>r[cr) VoiceAssistantAgentz?Agent specializing in voice interaction and language processingcs<ttj|d|j|d|jdddd|_dS)Nprocess_commandtext_to_speechEnglishHindiMarathirDhimr)rrr VOICE_ASSISTANTr handle_process_commandhandle_text_to_speech languagesr#r%r'r(rJs zVoiceAssistantAgent.__init__r)r*c s:z|jdst|j|jddiddWS|jr |jdnd}d}|r3t|}|r3|dd}t|IdH}d }d}t fd d d DrPd }t j }nt fdd dDrad }t j }nzt fdd dDrd }t j }gd}tfdd |Dd} | r|t j kr|jt j d| id|jdIdH} | r| jdg} | rg} | D])} d| vr| | dd| ddqd| vr| | dd| ddq| rdd| }t|j|j|||r|jnddd |jd!WSty}ztd"|t|j|jdd"t|iddWYd}~Sd}~ww)#z!Process voice command transcripts transcriptr,zNo transcript providedr-r2NrDrEFc3|] }|vVqdSNlower.0wordrr'r( uz=VoiceAssistantAgent.handle_process_command..)disease infectionspotsleafanalyzeTc3rrrrrr'r(r{r)yieldharvest productionzhow muchc3rrrrrr'r(rr)rwmarketsellcostmandi)wheatonionsoybean sugarcanericecottonc3s |] }|vr|VqdSrr)rrVrr'r(rsrVr\)r/r0r1r7rkprice_per_quintalz is z rupees per quintalprice_per_tonnez rupees per tonnezHere are the current prices: z, )response should_route target_agentvoice_responser6z Error processing voice command: )r0r8r r9r.r7rrHranyr rrRr_next send_messagerjoinvaluer<r=r,r>)r$r)r?rEr7r4rr common_cropsrVmarket_responserk price_textsrwr@r'rr(rVs       z*VoiceAssistantAgent.handle_process_commandc s|jd}|st|j|jddiddS|jr|jdnd}d}|r1t|}|r1|dd}t|j|jd d |j|d id |jd S)z3Generate text-to-speech audio (mock implementation)textr,z%No text provided for speech synthesisr-r2NrDrEr)z.Text-to-speech synthesis would happen here in r tts_responser6) r0r8r r9r.r7rrHr)r$r)rr?rEr7r'r'r(rs.   z)VoiceAssistantAgent.handle_text_to_speech) rJrKrLrMrr rrrrNr'r'r%r(rGs   `rcrO)TranslatorAgentzCAgent specializing in language translation for multilingual supportcs.ttj|d|jdddd|_dS)N translaterrrr)rrr TRANSLATORr handle_translaterr#r%r'r(rs  zTranslatorAgent.__init__r)r*c sz|jd}|jdd}|jd}|s$t|j|jddiddWS||jvr9t|j|jdd|iddWSz2|j|d }|j|d }d |d |d |}t|IdH}|} t|j|jd| id|jdWWSty} z t d| t|j|jddt | iddWYd} ~ WSd} ~ wwty} zt d| t|j|jddt | iddWYd} ~ Sd} ~ ww)z Translate text between languagesrtarget_languagerDsource_languager,z No text provided for translationr-zUnsupported target language: zthe source languagerz"Translate the following text from z to zu. Only return the translated text, nothing else. Text to translate: Ntranslated_texttranslation_resultr6zError in translation: ) r0r8r r9r.rrr7r<r=r,r>) r$r)rrrsource_lang_nametarget_lang_namerIr4rr@r'r'r(rsl      z TranslatorAgent.handle_translate) rJrKrLrMrr rrrNr'r'r%r(rs  rcCsTt}t|t}t|t}t|t}t|t}t|tS)z.Initialize and register all specialized agents)rrregister_agentrPr[rr) disease_agent yield_agent market_agent voice_agenttranslator_agentr'r'r( init_agents%s     r).rMrmtypingrrrrrrloggingjsonioPILrr r rapp.core.multi_agentr r r rrapp.core.ai_servicesrrrrrapp.services.market_scraperrrapp.core.configrsettings basicConfigINFO getLoggerrJr=rrPr[rrrr'r'r'r(s0    ^3 O