o zj7@sdZddlmZmZmZmZmZmZddlm Z m Z ddl Z ddl Z ddl Z ddlmZmZddlmZmZmZmZddlZddlZddlZddlmZmZmZmZeZGdd d eZGd d d eZ Gd d d eZ!e"dedfdefddZ#e"dedddedddfde$de$fddZ%e"dededddfdede$fddZ&e"dededddfdede$fd d!Z'ej"d"e d#d$edfd%efd&d'Z(ej"d(d#d)edfd%e!fd*d+Z)ej*d,d#d)d-e$fd.d/Z+dS)0z Chat assistant API endpoints for CropPilot This module provides endpoints for interacting with the general-purpose AI chat assistant. ) APIRouter HTTPExceptionDependsBodyFile UploadFile)StreamingResponse FileResponseN) BaseModelField)OptionalDictAnyList) AgentTypeMessage coordinatorcontext_protocolc@seZdZUdZeddddZeed<eddd Ze eed <ed d d Z e eed <eddd Z e eed<eddd Z e e eefed<eddd Ze eed<dS) 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 rrrr rr r'r'9/var/www/html/CropPilot/backend/app/api/endpoints/chat.pyrs  rc@s.eZdZUdZeed<eed<dZeed<dS) ChatResponsezModel for chat response dataresponserzgemini-2.5-flash model_usedN)r!r"r#r$r%r&r+r'r'r'r(r)s r)c@seZdZdZdS)StreamChatInputz3Model for streaming chat input, extending ChatInputN)r!r"r#r$r'r'r'r(r,$sr,z/speech-to-text.filec s.zddl}Wnty}ztddd|d}~ww|}tjdddg}||IdH|| |j B}| |}z| |}d |iWWdWdS|j ygtd d d|jy|}z td d |dd}~ww1swYWddS1swYdS)zf Convert uploaded speech audio to text using SpeechRecognition. Accepts WAV or MP3 files. rNzFSpeechRecognition is not installed. Run: pip install SpeechRecognition status_codedetailFz.wav)deletesuffixtextzCould not understand audio.zSTT service error: )speech_recognition ImportErrorr RecognizertempfileNamedTemporaryFilewritereadflush AudioFilenamerecordrecognize_googleUnknownValueError RequestError)r-sre recognizer temp_audiosourceaudior4r'r'r(speech_to_text(s>    "rKz/text-to-speechT)embedrr4rc szddlm}Wnty}ztddd|d}~wwz|||d}t}|||dt|dd WSt yM}z td d |dd}~ww) zN Convert text to speech audio using gTTS and return as an MP3 stream. r)gTTSr.z,gTTS is not installed. Run: pip install gTTSr/N)r4langz audio/mpeg media_typer6z TTS error: ) gttsrMr8rioBytesIO write_to_fpseekr Exception)r4rrMrFttsmp3_fpr'r'r(text_to_speechCs,   rYz/sttzen-USc szddlm}Wnty}ztddd|d}~ww|}|IdH}|j|d}|j|jjj |dd d }|j ||d }d d d|j D} | sTtdddd| iS)z] Accepts an audio file and returns the transcript using Google Cloud Speech-to-Text. rspeechr.zYGoogle Cloud Speech-to-Text client is not available. Run: pip install google-cloud-speechr/NcontentrTencoding language_codeaudio_channel_countenable_automatic_punctuationconfigrJ cSg|]}|jdjqSr alternatives transcript.0resultr'r'r( nzstt..r5Could not transcribe audio.rj) google.cloudr[r8r SpeechClientr=RecognitionAudioRecognitionConfig AudioEncodingLINEAR16 recognizejoinresults) r-rr[rFclient audio_contentrJrdr*rjr'r'r(sttXs4  r|z /speech-chatc s8zddlm}ddlm}Wnty"}ztddd|d}~ww|}|IdH}|j|d}|j|jj j |d d d }|j ||d } d dd| j D} | sZtdddt| |d} t| IdH} | j} |}|j| d}|j||jjd}|j|j jd}|j|||d}t|jd}| | |dS)z Accepts an audio file, transcribes it with Google Cloud STT, sends to chat AI, and returns the response as audio and text using Google Cloud TTS. rrZ) texttospeechr.zhGoogle Cloud Speech/TTS is not available. Run: pip install google-cloud-speech google-cloud-texttospeechr/Nr\rTr^rcrecSrfrgrhrkr'r'r(rnrozspeech_chat..r5rp)rr)r4)r` ssml_gender)audio_encoding)inputvoice audio_configzutf-8)user_transcriptai_textai_audio_base64)rqr[r}r8rrrr=rsrtrurvrwrxryr chat_messager*TextToSpeechClientSynthesisInputVoiceSelectionParamsSsmlVoiceGenderNEUTRAL AudioConfigMP3synthesize_speechbase64 b64encoder{decode)r-rr[r}rF client_sttr{rJrdr*r4 chat_input chat_responser tts_clientsynthesis_inputrr tts_response audio_base64r'r'r( speech_chatssX     rz/message)response_modelr0rc sJzx|jp tt}t||j|jd|jpit j t j t j t j d}||jp,dt j }t j dt j dt j dt j di}|||t j }ttt j||j|j|dd ||jd d Id H}t|jtrj|jn|jd d}t|||jdWSty}z|d }~wty} ztd| tddt| dd } ~ ww)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_doctorrzFYou are CropPilot, 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)rrsenderreceiverr] message_typecontextNr*z (No response))r*rr+z,Unexpected error in /chat/message endpoint: r6#An internal server error occurred: r/)rr%uuiduuid4r set_contextrrrrCHAT_ASSISTANT MARKET_EXPERTWEATHER_ADVISOR CROP_DOCTORgetr r route_messager COORDINATORr isinstancer]r)rrVprint) rr agent_mapselected_agentsystem_promptsrr response_texthttp_excrFr'r'r(rsZ rz/stream)r0c sz^|jp tt}t||j|jd|jpit t t j t j|j|jddd|idIdH}|r:d|jvrK|rC|jddnd }td |d |jd d fdd}t|ddWStym}z|d}~wty}ztd|tddt|d d}~ww)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 agentr.r/r*z&Sorry, I couldn't generate a response.cs^t}td|d}td||D]}t|||}||}|VtdIdHqdS)Nrg?)lenmaxrangeminasynciosleep) total_length chunk_sizeiendchunkrr'r(fake_stream_generators z*stream_chat..fake_stream_generatorzapplication/x-ndjsonrOz+Unexpected error in /chat/stream endpoint: r6r)rr%rrrrrrrrrrrrrrr]rrrrVr)rrrrrrrFr'rr( stream_chatsJ     rz/history/{session_id}rc sz[d|}t|}|s|gdWSg}|D]>}|d}t|tr6ddlm}m}|j||jd }| |ddt|trHt t|n||d|d |d q||dWSt yz}zt d |td d t|d d}~ww)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-r])idrr]rzError retrieving chat history: r6r/N)r get_contextrrfloatrr fromtimestamputc isoformatappendintrVrrr%) rchat_history_key chat_historyritemrrrrFr'r'r(get_chat_history1s0     & r),r$fastapirrrrrrfastapi.responsesrr rRrr:pydanticr r typingr r rrrrjsonapp.core.multi_agentrrrrrouterrr)r,postrKr%rYr|rrrrrr'r'r'r(s<  *&&6 ? F