@h5PdZddlmZmZmZmZmZmZddlm Z m Z ddl Z ddl Z ddlm Z ddl mZmZddlmZmZmZmZddlZddlmZddlZddlZddlZdd lmZmZmZm Z eZ!Gd d eZ"Gd d eZ#Gdde"Z$e!jKdedfdefdZ&e!jKdeddeddfde'de'fdZ(ddl)m*Z*m+Z+ddl,Z,ddl-Z-e!jKdededdfdede'fdZ.e!jKdededdfdede'fd Z/e!jKd!e#d"#edfd$e"fd%Z0e!jKd&d"'edfd$e$fd(Z1e!jed)d"'d*e'fd+Z3y),z Chat assistant API endpoints for FarmGenius This module provides endpoints for interacting with the general-purpose AI chat assistant. ) APIRouter HTTPExceptionDependsBodyFile UploadFile)StreamingResponse FileResponseN)r ) BaseModelField)OptionalDictAnyList)gTTS) AgentTypeMessage coordinatorcontext_protocolceZdZUdZedddZeed<edd Ze eed <ed d Z e eed <edd Z e eed<edd Z e e eefed<edd Ze eed<y) ChatInputzModel for chat input data.z$User's message to the chat assistant) min_length descriptionmessageNz'Session ID for continuing conversations)r session_idgeminiz'Model to use (only gemini is supported)modelenzLanguage code (en, hi, mr)languagezAdditional context for the chat context_datageneral_assistantz"Expert agent to handle the messageagent)__name__ __module__ __qualname____doc__r rstr__annotations__rr rr r!rrr#EC:\Users\aadil\OneDrive\Desktop\404\backend\app\api\endpoints\chat.pyrrs#8^_GS_ %d8a bJ b 7`aE8C=a#D6RSHhsmS-24Ef-gL(4S>*g !4BfgE8C=gr+rc4eZdZUdZeed<eed<dZeed<y) ChatResponsezModel for chat response dataresponserzgemini-1.5-pro model_usedN)r$r%r&r'r(r)r0r*r+r,r.r.s&MO&J&r+r.ceZdZdZy)StreamChatInputz3Model for streaming chat input, extending ChatInputN)r$r%r&r'r*r+r,r2r2%s=r+r2z/speech-to-text.filec^Ktj}tjdd5}|j |j d{|j tj|j5}|j|} |j|}d|icdddcdddS7q#tj$rtddtj$r}td d |d}~wwxYw#1swYnxYw dddy#1swYyxYww) zf Convert uploaded speech audio to text using SpeechRecognition. Accepts WAV or MP3 files. Fz.wav)deletesuffixNtextzCould not understand audio. status_codedetailzSTT service error: )sr RecognizertempfileNamedTemporaryFilewritereadflush AudioFilenamerecordrecognize_googleUnknownValueErrorr RequestError)r3 recognizer temp_audiosourceaudior7es r,speech_to_textrO)s J  $ $E& AZtyy{*+ \\*// *f%%f-E W!2259~ + * B A*'' [#QRSQT$>?FFwOL' ;&W 3s/(G'GBG', G  1G'=G%>C G' G'z/message)response_modelr:rc K |jxsttj}t j ||j |jd|jxsitjtjtjtjd}|j|jxsdtj}tjdtjdtjdtjdi}|j||tj}t!j"t%tj&||j|j(|dd ||j d  d {}t+|j,tr |j,n|j,jd d}t/|||j(S7^#t0$r}|d }~wt2$r,} t5d| t1ddt| d } ~ wwxYww)z Send a message to the AI chat assistant and get a response. This endpoint handles regular (non-streaming) chat interactions. r last_message)r" market_expertweather_advisor crop_doctorr"zGYou are FarmGenius, your general agriculture AI assistant across India.zrYou are MarketExpert: Provide expert analysis and advice on agricultural markets, prices, and trends across India.zuYou are WeatherAdvisor: Provide expert advice on weather patterns, forecasts, and climate impact on farming in India.zmYou are CropDoctor: Provide expert advice on crop diseases, soil, and crop management for Indian agriculture.)rr system_promptchat)rr senderreceiverra message_typecontextNr/z (No response))r/rr0z,Unexpected error in /chat/message endpoint: r<#An internal server error occurred: r9)rr(uuiduuid4r set_contextr rr!rCHAT_ASSISTANT MARKET_EXPERTWEATHER_ADVISOR CROP_DOCTORgetr#r route_messager COORDINATORr isinstancerar.rrYprint) rr agent_mapselected_agentsystem_promptsrr response_texthttp_excrNs r,rrs 8d**?c$**,.?  $$Z"++&..2 &&,"2  "+!9!9&44(88$00  #z'7'7'N;NPYPhPhi  $ $&o  # #&Z  % %(_  ! !$S   '**>>)JbJb;cd $11  ,,')11'--%2 $", * 3 3    ,6goos+KQXQ`Q`QdQdeorARB ]zV`VfVfgg# (  d FG8G6AG85H>6G88 H;H H;'H66H;;H>z/stream)r:c K |jxsttj}t j ||j |jd|jxsitjttjtj|j|jddd|id{}|rd|j vr-|r|j j#ddnd }t%d | |j j#d d fd}t'|dS7t#t$$r}|d}~wt($r,}t+d|t%ddt| d}~wwxYww)a! Stream a chat response from the AI assistant. This endpoint provides a streaming response for a more interactive experience. Note: This is a simplified implementation. In a production environment, you would implement actual streaming with the Gemini or Groq API. r)rrrrrNerrorz%Unknown error processing chat messagezNo response from chat agentir9r/z&Sorry, I couldn't generate a response.cKt}td|dz}td||D]8}t||z|}||}|t j dd{:y7w)Nrg?)lenmaxrangeminasynciosleep) total_length chunk_sizeiendchunkrs r,fake_stream_generatorz*stream_chat..fake_stream_generatorss}-LQ  23J1lJ7!j.,7%a, mmD)))8*sAA)A' A)zapplication/x-ndjsonrSz+Unexpected error in /chat/stream endpoint: r<r)rr(rrrrr rr!rrrrrrrrarrr rYr)rrrrrrrNrs @r, stream_chatrs<d**?c$**,.?  $$Z"++&..2 &&,"2  $11  ,,"11)11'--$%z2    'W__4]dGOO''1XYkHEC> > ++J8`a  * ! ! #-  K  T  d ;A3?@6YZ]^_Z`Ya4bccdsHFCD?D= A3D?<F=D?? FE  F'E==FFz/history/{session_id}rc K d|}tj|}|s|gdSg}|D]}|jd}t|tr3ddlm}m}|j||jj}|j|ddt|trtt |n||d|d |d ||dS#t$r,}td |td d t| d}~wwxYww)z Get the chat history for a specific session. This is useful for continuing conversations or displaying chat history. chat_history_)rmessages timestampr)datetimetimezone)tzrole-ra)idrrarzError retrieving chat history: r<r9N)r get_contextrrfloatrr fromtimestamputc isoformatappendintrYrrr() rchat_history_key chat_historyritemrrrrNs r,get_chat_historyrs `*:,7'334DE ","= = D-I)U+7$2292NXXZ OOfaIW\A]E),<(=cl'mnV  ?&  !)h?? ` /s346UVYZ[V\U]4^__`s3D CDB)CD D'C??DD)4r'fastapirrrrrrfastapi.responsesr r rUr?pydanticr r typingr rrrspeech_recognitionr=gttsrrrjsonapp.core.multi_agentrrrrrouterrr.r2postrOr(r\ google.cloudr]r^rosrxrrrrrr*r+r,rsNM= /%,,  RR h h'9'  i  ,0IWzW W& %)#T%:DQU]aLb Gs Gc G  G. V!%cDPT