o zj @s,ddlmZmZddlmZmZmZddlm Z m Z ddl m Z m Z mZmZddlZddlmZeZeZejrAejdkrAejndZdZeroz eedeejZWneynZzed eeWYdZ[ndZ[wwiZGd d d e ZGd d d e ZejdeddedfdefddZ dS)) configureGenerativeModel) APIRouter HTTPExceptionBody) BaseModelField)OptionalDictAnyListN) get_settingsYOUR_GEMINI_API_KEY_NOT_SET)api_keyzFailed to initialize Gemini: c@sTeZdZUeddddZeed<edddZeeed <ed d dZ eeed <dS) ChatInput.z$User's message to the chat assistant) min_length descriptionmessageNz'Session ID for continuing conversations)r session_idenz Language code (en, hi, mr, etc.)language) __name__ __module__ __qualname__rrstr__annotations__rr rrrF/var/www/html/CropPilot/backend/app/api/endpoints/multilingual_chat.pyrs rc@s2eZdZUeed<eed<dZeed<eed<dS) ChatResponseresponserzgemini-2.5-flash model_usedhistoryN)rrrrrr!listrrrrr#s   rz/message)response_model status_code. chat_inputc sts tdddzV|jptt}t|g}|s+|ddd|j digd|dd|j igdtj |d }| |j }|j }|d d|igd|t|<t||tj|d WStyv}z tdd t|dd }~ww)z Multilingual Gemini chat endpoint using google-generativeai SDK. Stores and uses conversation history per session. izGemini model not initialized)r&detailusertextaeYou are CropPilot, a professional multilingual AI assistant for Indian agriculture. You ONLY answer questions related to farming, crops, weather, agri-markets, government schemes, or rural livelihoods. If a question is not about agriculture, politely refuse and ask the user to ask a farming-related question. Always answer in the user's selected language: z. Never answer in English unless the user selected English. Be friendly, clear, and provide detailed, actionable advice for Indian farmers.)roleparts)r"model)r rr!r"zGemini error: N) genai_modelrrruuiduuid4conversation_historiesgetappendrr start_chat send_messager*rr GEMINI_MODEL Exception)r'rr" chat_sessionr ai_responseerrr chat_message)s>    r;)!google.generativeairgenai_configurerfastapirrrpydanticrrtypingr r r r r/app.core.configr router _settingsGEMINI_API_KEYr.r6r7r:printrr1rrpostr;rrrrs8