o i& @sddlmZddlmZddlmZddlZddlZddlmZm Z ddl m Z ddl m Z zdd lmZdd lmZdd lmZWn1eyUed dZdZdZYneysZzed edZdZdZWYdZ[ndZ[wwz ddlmZedWn*eyz ddlmZedWneyedGdddZYnwYnwz ddlmZedWn*eyz ddlmZedWneyedGdddZYnwYnwddZddZddZ Gd d!d!Z!Gd"d#d#Z"ed$d%Z#d&d'Z$dS)(render) JsonResponse) csrf_exemptN)datetime timedelta) GeminiClient) RAGService)OpenAI)LLMChain)PromptTemplatezUWARNING: OpenAI dependencies could not be imported. OpenAI Agent will be unavailable.z.WARNING: Error importing OpenAI dependencies: )Empresaz*Usando modelo Empresa desde Models.Empresaz.Usando modelo Empresa desde api_empresa.modelsuANo se pudo importar ningún modelo de Empresa, usando simulaciónc@eZdZdZdS)rN__name__ __module__ __qualname__objectsrr+/var/www/html/crm_dreinet/ai_agent/views.pyr&r)OportunidadDeVentaz-Usando modelo OportunidadDeVenta desde Modelsz*Usando modelo OportunidadDeVenta desde apiu3No se pudo importar OportunidadDeVenta, simulaciónc@r)rNrrrrrr3rrc Csrtj}tj}ztjj||dd}dd|DWSty8}ztdt |gWYd}~Sd}~ww)NT)fecha_creacion__monthfecha_creacion__yeardeleted_at__isnullc Sshg|]0}|j|jdt|ddt|drt|jddndt|dr)t|jddndt|dddqS)%Y-%m-%dcuitz No disponible provincianombre localidad direccion)rfecha_creacionrrr r!)rr"strftimegetattrhasattrrr ).0crrr @s    z0get_new_companies_this_month..zError al obtener empresas: ) rnowmonthyearrrfilter Exceptionprintstr) current_month current_year new_companieserrrget_new_companies_this_month7s  r4c CszMzddlm}Wn$ty-zddlm}Wnty*tdgYYWSwYnwttdd}|jj |dd ddd }d d |DWSt yj}ztd t |gWYd}~Sd}~ww) Nr) Actividadz*No se pudo importar el modelo de Actividad)daysT) fecha__gterz-fecha c SsXg|](}t|ddt|dr|jr|jjndt|dr!|jdndt|ddd qS) tipoGeneralempresazNo especificadafecharzFecha no disponible descripcionz Sin notas)r:r<r=notas)r$r%r<rr=r#)r&arrrr(]s   z)get_recent_activities..zError al obtener actividades: ) api_actividad.modelsr5 ImportErrorModels.api_actividad_modelr.rr)rrr,order_byr-r/)r5 fecha_limite actividadesr3rrrget_recent_activitiesLs6   rGc Csz8tjjddd}tjjddd}z tjjdd}Wnd}Y|||d|d|d |d d WStyS}ztd |iWYd }~Sd }~ww)z0Retrieves real-time statistics from the databaseT)r tipo_empresar)rrzEl sistema cuenta con z clientes activos, z datos potenciales y z$ oportunidades de venta registradas.)total_clientestotal_datos_potencialestotal_oportunidadesresumenz Error stats: N)rrr,countrr-r.)rJ total_datosrLr3rrrget_system_statsgs"rPc@eZdZddZdddZdS) GeminiAgentc CsFzt|_t|_tdWdSty"}ztd|d}~ww)Nz!Inicializando GeminiAgent con RAGzError inicializando Gemini: )r clientr ragr.r-selfr3rrr__init__~szGeminiAgent.__init__Nc Cs|si}t}|r ||d<t}|r||d<t}|r||d<g}z7|j|}|D](}|d}|rR|ddkrR|ddd krRtd |d |d WSq*||d <Wntyu} ztd | g|d <WYd} ~ nd} ~ wwtj |ddd} d| d|d} |j |  S)Nempresas_nuevas_mesactividades_recientesestadisticas_sistemametadatatype answer_bankscorerg333333?zUsing Cached Answer: contentknowledge_basez RAG Error: FrH ensure_asciiindentu Eres un asistente experto para el CRM Dreinet. INSTRUCCIONES CRÍTICAS: 1. Tu objetivo es responder preguntas sobre el ESTADO ACTUAL del negocio usando los datos proporcionados. 2. Si te preguntan "cuántos", "reporte", o "estado", TIENES QUE USAR los datos de 'estadisticas_sistema' o 'empresas_nuevas_mes'. 3. NO describas capacidades. RESPONDE CON NÚMEROS. 4. Si el usuario pregunta algo que ya está en 'knowledge_base' (como un reporte anterior guardado), úsalo. 5. IMPORTANTE: SIEMPRE menciona los números exactos de clientes, datos y oportunidades si te piden un resumen. DATOS EN TIEMPO REAL: z( PREGUNTA DEL USUARIO: z% Respuesta: ) r4rGrPrTsearchgetr.r-jsondumpsrSgenerate_contentstrip) rVquery context_datar2 activitiesstats rag_resultsresultmetar3formatted_contextpromptrrr process_querys<     "   zGeminiAgent.process_queryNrrrrWrsrrrrrR}srRc@rQ) OpenAIAgentc Cshtdurtdd|_ztdtjd<tdd|_tdWdSty3}ztd|d}~ww)Nzst                    =!