>$jIBdZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZmZmZmZmZmZmZmZej,ed z j/d Zd eefd Zd ed eefdZdZdZdZ dZ!dZ"ddZ#ddZ$ddefdZ%ded efdZ&gdZ'y)zOSelecao das conversas relevantes, texto das midias e envio para o grupo Agente.N)ThreadPoolExecutor)datetime)llm) mensagens) API_WHATSAPPDADOS DIAS_PROJETOS DIR_MIDIAFUSO MAX_MIDIA_MB MODELO_MIDIASEGREDOSz whatsapp.json grupoAgentereturnc2tjtdz j}|j di}|j d|j dfDcgc]+}|s|j ddj dd-c}Scc}w)z6Numero e LID do usuario, para achar mencoes em grupos.zwhatsapp-auth/creds.jsonmeidlid:r@)jsonloadsr read_textgetsplit)credsrxs 7/home/ubuntu/projetos/agente-tarefas/agente/whatsapp.pymeus_idsr s} JJ ::EEG HE 4 B46FF4L"&&-3P V3PaTUAGGCLO ! !# &q )3P VV Vs !B)(Bdesde_tsc t}|jd|fDcgc]}|d }}|jd|fDcgc]}|d }}djdgtt dzz}|jd|d|gt Dcgc]}d |d  c}d Dcgc]}|d }}t j ||z|zDcgc]}|tk7s |c}Scc}wcc}wcc}wcc}wcc}w) zMConversas diretas + grupos onde o usuario falou + grupos onde foi mencionado.zKSELECT DISTINCT chat FROM mensagens WHERE ts > ? AND chat NOT LIKE '%@g.us'rzcSELECT chat FROM mensagens WHERE ts > ? AND chat LIKE '%@g.us' GROUP BY chat HAVING SUM(de_mim) > 0z OR z texto LIKE ?rzMSELECT DISTINCT chat FROM mensagens WHERE ts > ? AND chat LIKE '%@g.us' AND ()z%@%z%Penna%)rexecutejoinlenr dictfromkeys GRUPO_AGENTE) r!conrdiretasgruposfiltroimencoescs rchats_relevantesr3sO +C [[UX`WbdedqtdGeKKm  qadF[[.)S_q-@A BF [[ WX^W__`a ? 3 1r!AY 3?Y?ABAqtAGB}}Wv%5%?@ V@!ADUA@ VVe 4B Ws# C3 C8 C= 2 DD,DzPTranscreva fielmente este audio em portugues. Responda apenas com a transcricao.zExtraia deste documento o que importa para a agenda de quem o recebeu: resumo em ate 12 linhas, incluindo TODAS as datas, horarios, prazos, tarefas, responsaveis e valores mencionados.zDescreva esta imagem em 1 a 3 linhas. Se houver texto (print de conversa, aviso, cronograma, comprovante, convite), transcreva datas, horarios, prazos e tarefas. Se for meme, figurinha ou foto sem informacao util, responda apenas: [sem informacao]c|jj}|j}|dk(s|dk(rtj|dt gS|dk(rIddl}djd|jtj|jD}n|dvrddl }|jtj|dd }g} |jdd D]T} | jd d D]<} t!d | Ds| j#djd| D>Vdj| }n|dvr|j%dd}nyd|d|ddt gS)Nz.pdfzapplication/pdfz.docxr c34K|]}|jywN)text).0ps r z#_texto_documento..7sV*UQ!&&*Us)z.xlsxz.xlsmT) read_only data_only)max_row values_onlyc3$K|]}|du ywr7r9vs rr;z#_texto_documento..>s2cq}csz | c3:K|]}|dn t|yw)N)strrDs rr;z#_texto_documento..?s ,VRUQ19R#a&-HRUs)z.txtz.csvz.mdzutf-8replacez Documento 'z': i`)suffixlower read_bytesr parte_bytes PROMPT_DOCdocxr&DocumentioBytesIO paragraphsopenpyxl load_workbook worksheets iter_rowsanyappenddecode) caminhomimenomeextdadosrOtextorTwblinhaswsrows r_texto_documentore0sQ ..   C    E f} 11'89:FF g~ V$-- 58I*J*U*UVV " "  # #BJJu$5QU # V--#B||CT|B2c22MM%**,VRU,V"VWC$ &! ' ' Wi0$tE&5M? 3Z @@ct|dz }|jr'|jjtdzdzkDry|ddk(rFt j |j|dxsdjddtg}d }n|dd k(r4t j |j|dxsd tg}d }nCt||d|d xs |j}d|d xsdj}|d|dSt jt|ddj}d|d|S)Nmidiaitipo audioMessager\z audio/ogg;raudio imageMessagez image/jpegimagem nome_arquivoz documento rG[z: formato nao lido]i) max_saidaetapa] )r existsstatst_sizer rrMrLr PROMPT_AUDIO PROMPT_IMGrenamestripgerarr)mr[conteudorotulor`s r_processar_midiarHsD!G*$G >> w||~55 t8Kd8RRyN"OOG$6$6$81V9;S :Z:Z[^:_`a:bceqr 6n $OOG$6$6$8!F):S|TV`a#GQvY.8I8YW\\Za/5267==?  vh12 2 IIlHG L R R TE vhb  rfcnt}djdt|z}|jd|d|g|Dcgc] }t |}}|jdDchc] }|d|df}}|D cgc]} | d| df|vs| }} d} d } t |5} | j | |D]A\} } | s |jd | d| d| f| d z } | d zd k(s2|jC d d d |j| Scc}wcc}wcc} w#1swY*xYw)zCTranscreve/le as midias ainda sem texto. Devolve quantas processou.,?z|SELECT chat, id, tipo, mime, midia, nome_arquivo FROM mensagens WHERE midia IS NOT NULL AND ts > ? AND chat IN (z[) AND tipo IN ('audioMessage', 'imageMessage', 'documentMessage') ORDER BY tsz SELECT chat, id FROM midia_textochatrc |t|fS#t$r+}|d|ddt|jdfcYd}~Sd}~wwxYw)Nrpriz: erro ao ler (z)])r Exceptiontype__name__)r|es rtarefaz processar_midias..tarefagsT I&q)) ) I!F)ODG4D4D3ERHH H Is  A >AArz3INSERT OR REPLACE INTO midia_texto VALUES (?, ?, ?)rN)rr&r'r%r(rmapcommit) con_agentechatsr!paralelor+marcasr, pendentesfeitosr|rnexr`s rprocessar_midiasr[sq +C XXcCJ& 'F"%++ ==CHEZ ] E ##Qa#I -7,>,>?a,b c,bqqy!D'",bF c%LIq!F)QtW)=V)KIILI A H %vy1HAu""#X[\]c[dfghlfmotZuvQr6Q;%%' 2 & H/ dL & %s0D+D!D&D&*D+(D+/D++D4c \t}d|rdndzdz}g}|j||r|||fn||fD]}|d}|ddvrZ|jd||d fj} | r| dnd |djd dd } d | vr|s[| |rd|ndz}n|ddvrq|stt j |dt jd} |dxs|drdn|dxsd} |drd|dddd|}|j|d |dd | d| d||d|dd|S)zQMensagens de um chat como dicts com o texto final (incluindo midias transcritas).z3SELECT * FROM v_mensagens WHERE chat = ? AND ts > ?z AND ts <= ?rGz ORDER BY tsr`ri)rjrmdocumentMessagez7SELECT texto FROM midia_texto WHERE chat = ? AND id = ?rrpMessagez nao lido]z[sem informacao]z | legenda: )stickerMessagetsz%a %d/%m %H:%M autor_nomeem_grupoalguem chat_nomecontatocitadaz(respondendo a: "Nxz") rsz: )rrr`rr) rr%fetchonerIr fromtimestampr strftimerY) rrr!ate_tsr+qrbr,r`mtrhquandoautors rlinhas_conversarzs +C=SY_abessA F [[T8V4T8DT U'  V9K K##$]`dfghlfm_noxxzB#%BwKQqy/@/@B/O.PPZ+[E!U*5gEL"8rJE vY- -  ''$6??@PQ,a* HAkND_V_ X;(8Tc):(;4wGE QtWAdG&E7RTUZT[>\$%kN* O P!V$ Mrfr`c |jdd}tjjtdt j d|ijddid}td D]8} t jtjj|d }n|5jdr$ddlm}m}||d||dg|dgzddS#t$r|d k(rtjd YwxYw)z^Envia ao grupo Agente. Com `con`, registra o id para os comandos nao lerem a propria mensagem.z***z/enviarr`z Content-Typezapplication/jsonPOST)dataheadersmethod)timeout Nrr) progressosalvar_progressoz wa:enviadasi)rIurllibrequestRequestrrdumpsencoderangeloadurlopenrtimesleeprbancorr)r`r+req tentativar,rrs renviarrs MM$ $E .. L>!9 GUZK[@\@c@c@e*8:L)MV\ ! ^C1X   &..00b0ABA  155;6mi]B.OSTUYSZR[.[]a]b-cd H A~ JJrN s14C!!$DDdiascHttj|dzz S)NiQ)intr)rs rdesders tyy{ dUl **rf)r3rrrrr )r7)(__doc__rQrrurllib.requestrconcurrent.futuresrrrGrrrconfigrr r r r r rrrrr*listrHr rr3rwrNrxrerrrrr__all__rCrfrrsU 1tzz8o5@@BCMR W$s)W Ws WtCy W b i L A0!& >6 # &+++ jrf