Versão 0.2.0

pages/9_Tela9_Laudo.py

@@ -1,230 +1,230 @@
-
-import streamlit as st
-from template import render_header, require_uploaded_pdf, CACHE, GERARCACHE
-from agents.criadornotatecnica import criadornotatecnica5, modelo, temperature, especializacao, top_p, viewpoint
-from bd.embeddings import get_embeddings
-from bd.bdv import buscar_por_embeddings
-import json 
-
-st.set_page_config(page_title="Tela 9 — Laudo", layout="wide", initial_sidebar_state="collapsed")
-render_header()
-
-st.header("Tela 9 — Laudo (Markdown + Preview)")
-
-if not require_uploaded_pdf():
-    st.stop()
-
-result = st.session_state["result_classificador"]
-
-def normas_para_strings(normas):
-    resultado = []
-
-    for norma in normas:
-        partes = []
-
-        # Campos básicos da norma
-        partes.append(f"id_norma: {norma.get('id_norma')}")
-        partes.append(f"espécie_normativa: {norma.get('especie_normativa')}")
-        partes.append(f"sigla: {norma.get('sigla')}")
-        partes.append(f"número: {norma.get('numero')}")
-        partes.append(f"ano: {norma.get('ano')}")
-        partes.append(f"órgão_emissor: {norma.get('orgao_emissor')}")
-        partes.append(f"referência_completa: {norma.get('referencia_completa')}")
-
-        # Artigos mencionados
-        artigos = norma.get("artigos_mencionados", [])
-        artigos_str_list = []
-
-        for art in artigos:
-            artigo_str = (
-                f"artigo {art.get('artigo')}, "
-                f"parágrafos: {', '.join(art.get('paragrafos', [])) or 'nenhum'}, "
-                f"incisos: {', '.join(art.get('incisos', [])) or 'nenhum'}, "
-                f"alíneas: {', '.join(art.get('alineas', [])) or 'nenhum'}, "
-                f"fontes: {', '.join(art.get('ids_fontes_no_processo', []))}, "
-                f"observação: {art.get('observacao_curta')}"
-            )
-            artigos_str_list.append(artigo_str)
-
-        if artigos_str_list:
-            partes.append("artigos_mencionados: " + " | ".join(artigos_str_list))
-        else:
-            partes.append("artigos_mencionados: nenhum")
-
-        # Junta tudo em uma única string
-        norma_str = " ; ".join(partes)
-        resultado.append(norma_str)
-
-    return resultado
-
-
-def pegarLegislacao(legislacoes_marcadas):
-
-    textoleg = ""
-    #legislacoes_marcadas = st.session_state.get("legislacao", [])
-
-    if legislacoes_marcadas:
-        for idx, legislacao in enumerate(legislacoes_marcadas, start=1):
-            #print(f"{idx, legislacao}")
-            embedding = get_embeddings([legislacao]) 
-            resultados = buscar_por_embeddings(embedding, top_k=2)  
-
-            pontos = resultados.get("result", []) if resultados else []
-            textoleg = f"{textoleg}\n\n**{[legislacao]}**\n\n"
-            if pontos:
-                print(f"Legislações similares encontradas para a legislação {idx}:")
-                for i, ponto in enumerate(pontos, start=1):
-                    chave = ponto.get("id", "Sem ID")
-                    score = ponto.get("score", 0)
-                    payload = ponto.get("payload", {})
-
-                    conteudo = payload.get("conteudo", "")
-
-                    textoleg = f"{textoleg}\n\n- {conteudo}\n"
-            else:
-                print(f"Nenhum resultado encontrado para a legislação {idx}.")
-
-    else:
-        print("Marque pelo menos uma legislação para buscar similares.")
-
-    return textoleg
-
-
-from typing import Any, Dict, List
-
-def _md_escape(text: str) -> str:
-    # simples: evita quebrar markdown com quebras e mantém legível
-    return (text or "").replace("\r\n", "\n").strip()
-
-def format_grupos_markdown(grupos: List[Dict[str, Any]]) -> str:
-    linhas: List[str] = []
-
-    for g in grupos:
-        nome_grupo = _md_escape(g.get("nome_grupo", "Grupo sem nome"))
-        status_grupo = _md_escape(g.get("status_grupo", ""))
-
-        # Título do grupo (em negrito)
-        #if status_grupo:
-        #    linhas.append(f"**{nome_grupo}** _(status do grupo: {status_grupo})_")
-        #else:
-        #    linhas.append(f"**{nome_grupo}**")
-        
-        linhas.append(f"**{nome_grupo}**")
-
-        itens = g.get("itens", []) or []
-        if not itens:
-            linhas.append("- (sem itens)")
-            linhas.append("")  # linha em branco
-            continue
-
-        for it in itens:
-            desc = _md_escape(it.get("descricao_item", "Item sem descrição"))
-            status = _md_escape(it.get("status", ""))
-            evidencias = it.get("evidencias", []) or []
-            pendencias = it.get("pendencias", []) or []
-
-            # Cabeçalho do item
-            if status:
-                linhas.append(f"- **Item:** {desc}")
-                linhas.append(f"  - **Status:** {status}")
-            else:
-                linhas.append(f"- **Item:** {desc}")
-
-            # Evidências
-            if evidencias:
-                linhas.append("  - **Evidências:**")
-                for ev in evidencias:
-                    doc = _md_escape(ev.get("id_documento", ""))
-                    tipo = _md_escape(ev.get("tipo_documento", ""))
-                    loc = _md_escape(ev.get("localizacao", ""))
-                    resumo = _md_escape(ev.get("resumo_evidencia", ""))
-
-                    partes = []
-                    if doc: partes.append(f"doc: {doc}")
-                    if tipo: partes.append(f"tipo: {tipo}")
-                    if loc: partes.append(f"local: {loc}")
-                    if resumo: partes.append(f"resumo: {resumo}")
-
-                    linhas.append(f"    - " + (" | ".join(partes) if partes else "(evidência sem detalhes)"))
-            else:
-                linhas.append("  - **Evidências:** (nenhuma)")
-
-            # Pendências
-            if pendencias:
-                linhas.append("  - **Pendências:**")
-                for p in pendencias:
-                    linhas.append(f"    - {_md_escape(str(p))}")
-            else:
-                linhas.append("  - **Pendências:** (nenhuma)")
-
-        linhas.append("")  # linha em branco entre grupos
-
-    return "\n".join(linhas).rstrip()
-
-
-#default_md = st.session_state.get("resumo_md") or (
-#    "## Resumo Inicial\n\n"
-#    "Escreva aqui um breve resumo da Nota Técnica.\n\n"
-#    f"**Arquivo original**: {st.session_state.get('uploaded_pdf_name')}\n"
-#)
-
-entrar = True
-
-#default_md = ""
-if "notatecnica_md" not in st.session_state or st.session_state["notatecnica_md"] == "" or entrar == True:
-    print("processando IA - gerando laudo")
-    with st.spinner("🤖 Gerando Laudo com IA..."):
-        nome_arquivo = "notatecnica.txt"
-        if CACHE == False:
-            #default_md = sumarizador5(st.session_state["uploaded_pdf_text"], model=modelo, temperature=temperature, especializacao=especializacao, top_p=top_p, viewpoint=viewpoint)
-            #default_md = st.session_state["resumo_md"]
-            p1 = st.session_state["resumo_md"]
-            p2 = json.dumps(result["nota_tecnica"]["checklists"]["checklists"], ensure_ascii=False, indent=2)
-            leg = normas_para_strings(result["nota_tecnica"]["normas_processo"]["normas"])
-            p3 = pegarLegislacao(leg)
-
-            p2_text = format_grupos_markdown(result["nota_tecnica"]["checklists"]["checklists"])
-
-            default_md = f"""
-# LAUDO
-
-## Cheklist
-
-{p2_text}
-
-## Normas/Legislação
-
-{p3}
-            
-## Nota Técnica Básica
-
-{p1}
-
-            """
-            #default_md = criadornotatecnica5(st.session_state["uploaded_pdf_text"], model=modelo, temperature=temperature, especializacao=especializacao, top_p=top_p, viewpoint=viewpoint,
-            #                                 resultado_passo_1=p1, resultado_passo_2=p2, resultado_passo_3=p3)
-            if GERARCACHE == True:   
-                with open(nome_arquivo, "w", encoding="utf-8") as arquivo:
-                    arquivo.write(default_md)
-        else:
-            with open(nome_arquivo, 'r', encoding="utf-8") as f:
-                default_md = f.read()
-                #print(default_md)
-else:
-    default_md = st.session_state["notatecnica_md"]
-
-
-col1, col2 = st.columns([0.55, 0.45])
-with col1:
-    st.session_state["notatecnica_md"] = st.text_area("Editar Laudo em Markdown", value=default_md, height=420)
-
-with col2:
-    st.markdown("**Pré-visualização do Laudo**")
-    st.markdown(st.session_state["notatecnica_md"], unsafe_allow_html=False)
-
-st.divider()
-col1x, col2x, col3x = st.columns([0.1, 0.1, 0.8])
-with col1x:
-    st.page_link("pages/8_Tela8_Grupo5.py", label="⬅️ Voltar")
-with col2x:
-    st.page_link("pages/10_Tela10_Conclusao.py", label="Próximo passo ➡️")
+
+import streamlit as st
+from template import render_header, require_uploaded_pdf, CACHE, GERARCACHE
+#from agents.criadornotatecnica import criadornotatecnica5, modelo, temperature, especializacao, top_p, viewpoint
+from bd.embeddings import get_embeddings
+from bd.bdv import buscar_por_embeddings
+import json 
+
+st.set_page_config(page_title="Tela 9 — Laudo", layout="wide", initial_sidebar_state="collapsed")
+render_header()
+
+st.header("Tela 9 — Laudo (Markdown + Preview)")
+
+if not require_uploaded_pdf():
+    st.stop()
+
+result = st.session_state["result_classificador"]
+
+def normas_para_strings(normas):
+    resultado = []
+
+    for norma in normas:
+        partes = []
+
+        # Campos básicos da norma
+        partes.append(f"id_norma: {norma.get('id_norma')}")
+        partes.append(f"espécie_normativa: {norma.get('especie_normativa')}")
+        partes.append(f"sigla: {norma.get('sigla')}")
+        partes.append(f"número: {norma.get('numero')}")
+        partes.append(f"ano: {norma.get('ano')}")
+        partes.append(f"órgão_emissor: {norma.get('orgao_emissor')}")
+        partes.append(f"referência_completa: {norma.get('referencia_completa')}")
+
+        # Artigos mencionados
+        artigos = norma.get("artigos_mencionados", [])
+        artigos_str_list = []
+
+        for art in artigos:
+            artigo_str = (
+                f"artigo {art.get('artigo')}, "
+                f"parágrafos: {', '.join(art.get('paragrafos', [])) or 'nenhum'}, "
+                f"incisos: {', '.join(art.get('incisos', [])) or 'nenhum'}, "
+                f"alíneas: {', '.join(art.get('alineas', [])) or 'nenhum'}, "
+                f"fontes: {', '.join(art.get('ids_fontes_no_processo', []))}, "
+                f"observação: {art.get('observacao_curta')}"
+            )
+            artigos_str_list.append(artigo_str)
+
+        if artigos_str_list:
+            partes.append("artigos_mencionados: " + " | ".join(artigos_str_list))
+        else:
+            partes.append("artigos_mencionados: nenhum")
+
+        # Junta tudo em uma única string
+        norma_str = " ; ".join(partes)
+        resultado.append(norma_str)
+
+    return resultado
+
+
+def pegarLegislacao(legislacoes_marcadas):
+
+    textoleg = ""
+    #legislacoes_marcadas = st.session_state.get("legislacao", [])
+
+    if legislacoes_marcadas:
+        for idx, legislacao in enumerate(legislacoes_marcadas, start=1):
+            #print(f"{idx, legislacao}")
+            embedding = get_embeddings([legislacao]) 
+            resultados = buscar_por_embeddings(embedding, top_k=2)  
+
+            pontos = resultados.get("result", []) if resultados else []
+            textoleg = f"{textoleg}\n\n**{[legislacao]}**\n\n"
+            if pontos:
+                print(f"Legislações similares encontradas para a legislação {idx}:")
+                for i, ponto in enumerate(pontos, start=1):
+                    chave = ponto.get("id", "Sem ID")
+                    score = ponto.get("score", 0)
+                    payload = ponto.get("payload", {})
+
+                    conteudo = payload.get("conteudo", "")
+
+                    textoleg = f"{textoleg}\n\n- {conteudo}\n"
+            else:
+                print(f"Nenhum resultado encontrado para a legislação {idx}.")
+
+    else:
+        print("Marque pelo menos uma legislação para buscar similares.")
+
+    return textoleg
+
+
+from typing import Any, Dict, List
+
+def _md_escape(text: str) -> str:
+    # simples: evita quebrar markdown com quebras e mantém legível
+    return (text or "").replace("\r\n", "\n").strip()
+
+def format_grupos_markdown(grupos: List[Dict[str, Any]]) -> str:
+    linhas: List[str] = []
+
+    for g in grupos:
+        nome_grupo = _md_escape(g.get("nome_grupo", "Grupo sem nome"))
+        status_grupo = _md_escape(g.get("status_grupo", ""))
+
+        # Título do grupo (em negrito)
+        #if status_grupo:
+        #    linhas.append(f"**{nome_grupo}** _(status do grupo: {status_grupo})_")
+        #else:
+        #    linhas.append(f"**{nome_grupo}**")
+        
+        linhas.append(f"**{nome_grupo}**")
+
+        itens = g.get("itens", []) or []
+        if not itens:
+            linhas.append("- (sem itens)")
+            linhas.append("")  # linha em branco
+            continue
+
+        for it in itens:
+            desc = _md_escape(it.get("descricao_item", "Item sem descrição"))
+            status = _md_escape(it.get("status", ""))
+            evidencias = it.get("evidencias", []) or []
+            pendencias = it.get("pendencias", []) or []
+
+            # Cabeçalho do item
+            if status:
+                linhas.append(f"- **Item:** {desc}")
+                linhas.append(f"  - **Status:** {status}")
+            else:
+                linhas.append(f"- **Item:** {desc}")
+
+            # Evidências
+            if evidencias:
+                linhas.append("  - **Evidências:**")
+                for ev in evidencias:
+                    doc = _md_escape(ev.get("id_documento", ""))
+                    tipo = _md_escape(ev.get("tipo_documento", ""))
+                    loc = _md_escape(ev.get("localizacao", ""))
+                    resumo = _md_escape(ev.get("resumo_evidencia", ""))
+
+                    partes = []
+                    if doc: partes.append(f"doc: {doc}")
+                    if tipo: partes.append(f"tipo: {tipo}")
+                    if loc: partes.append(f"local: {loc}")
+                    if resumo: partes.append(f"resumo: {resumo}")
+
+                    linhas.append(f"    - " + (" | ".join(partes) if partes else "(evidência sem detalhes)"))
+            else:
+                linhas.append("  - **Evidências:** (nenhuma)")
+
+            # Pendências
+            if pendencias:
+                linhas.append("  - **Pendências:**")
+                for p in pendencias:
+                    linhas.append(f"    - {_md_escape(str(p))}")
+            else:
+                linhas.append("  - **Pendências:** (nenhuma)")
+
+        linhas.append("")  # linha em branco entre grupos
+
+    return "\n".join(linhas).rstrip()
+
+
+#default_md = st.session_state.get("resumo_md") or (
+#    "## Resumo Inicial\n\n"
+#    "Escreva aqui um breve resumo da Nota Técnica.\n\n"
+#    f"**Arquivo original**: {st.session_state.get('uploaded_pdf_name')}\n"
+#)
+
+entrar = True
+
+#default_md = ""
+if "notatecnica_md" not in st.session_state or st.session_state["notatecnica_md"] == "" or entrar == True:
+    print("processando IA - gerando laudo")
+    with st.spinner("🤖 Gerando Laudo com IA..."):
+        nome_arquivo = "notatecnica.txt"
+        if CACHE == False:
+            #default_md = sumarizador5(st.session_state["uploaded_pdf_text"], model=modelo, temperature=temperature, especializacao=especializacao, top_p=top_p, viewpoint=viewpoint)
+            #default_md = st.session_state["resumo_md"]
+            p1 = st.session_state["resumo_md"]
+            p2 = json.dumps(result["nota_tecnica"]["checklists"]["checklists"], ensure_ascii=False, indent=2)
+            leg = normas_para_strings(result["nota_tecnica"]["normas_processo"]["normas"])
+            p3 = pegarLegislacao(leg)
+
+            p2_text = format_grupos_markdown(result["nota_tecnica"]["checklists"]["checklists"])
+
+            default_md = f"""
+# LAUDO
+
+## Cheklist
+
+{p2_text}
+
+## Normas/Legislação
+
+{p3}
+            
+## Nota Técnica Básica
+
+{p1}
+
+            """
+            #default_md = criadornotatecnica5(st.session_state["uploaded_pdf_text"], model=modelo, temperature=temperature, especializacao=especializacao, top_p=top_p, viewpoint=viewpoint,
+            #                                 resultado_passo_1=p1, resultado_passo_2=p2, resultado_passo_3=p3)
+            if GERARCACHE == True:   
+                with open(nome_arquivo, "w", encoding="utf-8") as arquivo:
+                    arquivo.write(default_md)
+        else:
+            with open(nome_arquivo, 'r', encoding="utf-8") as f:
+                default_md = f.read()
+                #print(default_md)
+else:
+    default_md = st.session_state["notatecnica_md"]
+
+
+col1, col2 = st.columns([0.55, 0.45])
+with col1:
+    st.session_state["notatecnica_md"] = st.text_area("Editar Laudo em Markdown", value=default_md, height=420)
+
+with col2:
+    st.markdown("**Pré-visualização do Laudo**")
+    st.markdown(st.session_state["notatecnica_md"], unsafe_allow_html=False)
+
+st.divider()
+col1x, col2x, col3x = st.columns([0.1, 0.1, 0.8])
+with col1x:
+    st.page_link("pages/8_Tela8_Grupo5.py", label="⬅️ Voltar")
+with col2x:
+    st.page_link("pages/10_Tela10_Conclusao.py", label="Próximo passo ➡️")