Create app.py

app.py

@@ -0,0 +1,248 @@
+"""
+Agente CBMGO - Interface Principal Gradio
+Sistema Inteligente de Prevencao de Incendio - NT-01/2025
+"""
+import gradio as gr
+import math
+from datetime import datetime
+
+
+def calcular_extintores(area_m2, grupo):
+    tabela = {
+        "A": (500, "2-A:10-B:C"), "B": (250, "2-A:20-B:C"),
+        "C": (750, "1-A:5-B:C"), "D": (400, "2-A:20-B:C"),
+        "E": (500, "2-A:10-B:C"), "F": (300, "4-A:20-B:C"),
+        "G": (600, "1-A:5-B:C"), "H": (200, "2-A:20-B:C"),
+    }
+    ua, cap = tabela.get(grupo.upper(), (500, "2-A:10-B:C"))
+    return max(1, math.ceil(area_m2 / ua)), cap
+def calcular_hidrantes(altura_m):
+    if altura_m < 12:
+        return "Mangueira / Coluna Seca", 300
+    elif altura_m < 30:
+        return "Hidrante com Reservatorio", 450
+    return "Chuveiros Automaticos + Hidrante", 600
+
+
+def aplicar_regras(area_m2, altura_m, ocupacao, lotacao):
+    ex = ["extintores_portateis", "sinalizacao_emergencia"]
+    ocup = ocupacao.lower()
+    if area_m2 > 750 or ocup in ["comercial", "servicos"]:
+        ex.append("hidrantes")
+    if altura_m > 12:
+        ex += ["iluminacao_emergencia", "saidas_emergencia"]
+    if area_m2 > 1000:
+        ex.append("spda")
+    if ocup == "industrial" and area_m2 > 2000:
+        ex += ["chuveiros_automaticos", "reservatorio_incendio"]
+    if lotacao > 100:
+        ex += ["alarme_incendio", "deteccao_automatica"]
+    return list(set(ex))
+
+def aba_calculadora(nome, area_m2, altura_m, ocupacao, lotacao):
+    if area_m2 <= 0:
+        return "Erro: Informe a area total valida.", "", ""
+    grupos = {"residencial":"A","comercial":"D","servicos":"D",
+              "educacional":"G","saude":"H","industrial":"F",
+              "deposito":"J","reuniao":"C","esporte":"C"}
+    grupo = grupos.get(ocupacao.lower(), "D")
+    qtd, cap = calcular_extintores(area_m2, grupo)
+    sist, vazao = calcular_hidrantes(altura_m)
+    reserva = round(vazao * 60 / 1000, 1)
+    ex = aplicar_regras(area_m2, altura_m, ocupacao, lotacao)
+    data = datetime.now().strftime("%d/%m/%Y")
+
+    calc = (
+        "=== CALCULOS NT-01/2025 ===\n"
+        + f"Edificacao: {nome}\n"
+        + f"Ocupacao: {ocupacao} | Grupo: {grupo}\n"
+        + f"Area: {area_m2:.1f} m2 | Altura: {altura_m:.1f} m | Lot: {int(lotacao)}\n\n"
+        + f"EXTINTORES: {qtd} un. | Cap: {cap} | Dist.max: 15m\n"
+        + f"HIDRANTES: {sist} | Vazao: {vazao} L/min | Reserva: {reserva} m3\n"
+        + ("SPDA: OBRIGATORIO\n" if area_m2 > 1000 else "SPDA: Verificar\n")
+        + ("ILUM.EMERG: OBRIGATORIA\n" if altura_m > 12 else "ILUM.EMERG: Verificar\n")
+    )
+    exig_txt = "\n".join([f"  - {e.replace('_',' ').title()}" for e in ex])
+    exig = f"=== EXIGENCIAS ({len(ex)} sistemas) ===\n{exig_txt}\n\nRef: NT-01/2025 CBMGO"
+    exig_desc = ", ".join([e.replace("_"," ") for e in ex])
+    memorial = (
+        "MEMORIAL DESCRITIVO - PSCIP\n"
+        + f"Data: {data} | Edificacao: {nome} | Ref: NT-01/2025 CBMGO\n\n"
+        + "1. IDENTIFICACAO\n"
+        + f"   Uso: {ocupacao.title()} | Grupo NT-01: {grupo}\n"
+        + f"   Area: {area_m2:.2f} m2 | Altura: {altura_m:.2f} m\n\n"
+        + f"2. SISTEMAS EXIGIDOS (Grupo {grupo})\n"
+        + f"   {exig_desc}\n\n"
+        + "3. EXTINTORES PORTATEIS\n"
+        + f"   Qtd: {qtd} un. | Cap: {cap} | Dist.max: 15m\n"
+        + "   Distribuicao conforme Anexo B NT-01/2025\n\n"
+        + "4. SISTEMA DE HIDRANTES\n"
+        + f"   Sistema: {sist} | Vazao: {vazao} L/min | Reserva: {reserva} m3\n"
+        + "   Conforme ABNT NBR 13714 e NT especifica CBMGO.\n\n"
+        + "5. SINALIZACAO DE EMERGENCIA\n"
+        + "   Conforme ABNT NBR 13434.\n\n"
+        + "6. ILUMINACAO DE EMERGENCIA\n"
+        + ("   Obrigatoria. ABNT NBR 10898.\n\n" if altura_m > 12 else "   Verificar necessidade.\n\n")
+        + "7. SPDA\n"
+        + ("   Obrigatorio. ABNT NBR 5419.\n\n" if area_m2 > 1000 else "   Verificar necessidade.\n\n")
+        + "Responsavel Tecnico: _______________ CREA/CAU: ___________\n"
+    )
+    return calc, exig, memorial
+
+def aba_auditoria(memorial_txt, area_m2, altura_m, ocupacao, lotacao):
+    if not memorial_txt or len(memorial_txt) < 30:
+        return "Insira o texto do memorial para auditar."
+    ex = aplicar_regras(area_m2, altura_m, ocupacao, lotacao)
+    termos = {
+        "extintores_portateis": ["extintor"],
+        "hidrantes": ["hidrante","mangotinho"],
+        "sinalizacao_emergencia": ["sinalizacao","sinaliza"],
+        "iluminacao_emergencia": ["iluminacao de emergencia"],
+        "spda": ["spda","para-raios"],
+        "alarme_incendio": ["alarme","deteccao"],
+        "chuveiros_automaticos": ["chuveiro","sprinkler"],
+        "saidas_emergencia": ["saida de emergencia","rota de fuga"],
+    }
+    ml = memorial_txt.lower()
+    ok, erros = [], []
+    for e in ex:
+        ts = termos.get(e, [e.replace("_"," ")])
+        (ok if any(t in ml for t in ts) else erros).append(e)
+    status = "APTO PARA PROTOCOLO" if not erros else "PENDENCIAS ENCONTRADAS"
+    rel = (
+        "=== AUDITORIA NT-01/2025 ===\n"
+        + f"Data: {datetime.now().strftime('%d/%m/%Y %H:%M')}\n"
+        + f"STATUS: {status}\n\n"
+        + f"CONFORMES ({len(ok)}):\n"
+        + "\n".join([f"  [OK] {i.replace('_',' ').title()}" for i in ok])
+    )
+    if erros:
+        rel += f"\n\nERROS CRITICOS ({len(erros)}):\n"
+        rel += "\n".join([f"  [!!] {e.replace('_',' ').title()} - NAO ENCONTRADO" for e in erros])
+    return rel + "\n\nRef: NT-01/2025 CBMGO"
+
+def aba_siapi(nome, processo, rt_nome, rt_crea, tem_mem, tem_pran, tem_form):
+    faltam = []
+    if not tem_mem: faltam.append("Memorial Descritivo")
+    if not tem_pran: faltam.append("Pranchas/Plantas")
+    if not tem_form: faltam.append("Formulario RT")
+    ts = datetime.now().strftime("%Y%m%d_%H%M%S")
+    pid = f"CBMGO_{processo}_{ts}" if processo else f"CBMGO_{ts}"
+    apto = not faltam
+    rel = (
+        "=== VERIFICACAO PACOTE SIAPI ===\n"
+        + f"Data: {datetime.now().strftime('%d/%m/%Y %H:%M')}\n"
+        + f"ID: {pid}\n"
+        + f"Edificacao: {nome} | RT: {rt_nome} | CREA: {rt_crea}\n\n"
+        + "DOCUMENTOS:\n"
+        + f"  {'[OK]' if tem_mem else '[FALTA]'} Memorial Descritivo\n"
+        + f"  {'[OK]' if tem_pran else '[FALTA]'} Pranchas / Plantas\n"
+        + f"  {'[OK]' if tem_form else '[FALTA]'} Formulario RT\n\n"
+        + ("STATUS: APTO PARA PROTOCOLO NO SIAPI\n\n" if apto
+           else f"STATUS: INCOMPLETO\nFaltam: {', '.join(faltam)}\n\n")
+        + "NOMENCLATURA PADRAO CBMGO:\n"
+        + "  [PROCESSO]_memorial_pscip.pdf\n"
+        + "  [PROCESSO]_prancha_01.dwg\n"
+        + "  [PROCESSO]_formulario_rt.pdf\n"
+        + "  [PROCESSO]_art.pdf\n\n"
+        + "ATENCAO: Protocolo final deve ser realizado pelo RT no portal SIAPI/CBMGO.\n"
+    )
+    return rel
+
+# ============================================================
+# INTERFACE GRADIO
+# ============================================================
+with gr.Blocks(title="Agente CBMGO - NT-01/2025", theme=gr.themes.Base()) as demo:
+    gr.Markdown(
+        "# Agente CBMGO\n"
+        "### Sistema Inteligente de Prevencao de Incendio - NT-01/2025\n"
+        "> Apoio a Responsaveis Tecnicos (RT) - CBMGO Goias"
+    )
+
+    with gr.Tabs():
+        with gr.TabItem("Calculadora NT-01"):
+            with gr.Row():
+                with gr.Column():
+                    i_nome = gr.Textbox(label="Nome da Edificacao")
+                    i_area = gr.Number(label="Area Total (m2)", value=500)
+                    i_alt  = gr.Number(label="Altura (m)", value=6)
+                    i_ocup = gr.Dropdown(
+                        label="Ocupacao",
+                        choices=["comercial","residencial","industrial","educacional",
+                                 "saude","servicos","reuniao","esporte","deposito"],
+                        value="comercial"
+                    )
+                    i_lot  = gr.Number(label="Lotacao (pessoas)", value=50)
+                    btn1   = gr.Button("Calcular Exigencias", variant="primary")
+                with gr.Column():
+                    o_calc = gr.Textbox(label="Calculos NT-01/2025", lines=14)
+            with gr.Row():
+                o_exig = gr.Textbox(label="Exigencias Identificadas", lines=10)
+                o_mem  = gr.Textbox(label="Memorial Descritivo Gerado", lines=10)
+            btn1.click(aba_calculadora,
+                       inputs=[i_nome, i_area, i_alt, i_ocup, i_lot],
+                       outputs=[o_calc, o_exig, o_mem])
+        with gr.TabItem("Auditoria de Memorial"):
+            with gr.Row():
+                with gr.Column():
+                    a_mem  = gr.Textbox(label="Memorial para Auditar", lines=14,
+                                        placeholder="Cole o texto do memorial descritivo aqui...")
+                    a_area = gr.Number(label="Area (m2)", value=500)
+                    a_alt  = gr.Number(label="Altura (m)", value=6)
+                    a_ocup = gr.Dropdown(
+                        label="Ocupacao",
+                        choices=["comercial","residencial","industrial","educacional",
+                                 "saude","servicos","reuniao","esporte","deposito"],
+                        value="comercial"
+                    )
+                    a_lot  = gr.Number(label="Lotacao", value=50)
+                    btn2   = gr.Button("Auditar Memorial", variant="primary")
+                with gr.Column():
+                    o_aud  = gr.Textbox(label="Relatorio de Auditoria", lines=20)
+            btn2.click(aba_auditoria,
+                       inputs=[a_mem, a_area, a_alt, a_ocup, a_lot],
+                       outputs=[o_aud])
+
+        with gr.TabItem("Preparacao SIAPI"):
+            with gr.Row():
+                with gr.Column():
+                    s_nome = gr.Textbox(label="Nome da Edificacao")
+                    s_proc = gr.Textbox(label="N. Processo CBMGO")
+                    s_rt   = gr.Textbox(label="Nome do Responsavel Tecnico")
+                    s_crea = gr.Textbox(label="CREA/CAU")
+                    gr.Markdown("**Documentos do Pacote:**")
+                    s_mem  = gr.Checkbox(label="Memorial Descritivo (PDF)")
+                    s_pran = gr.Checkbox(label="Pranchas/Plantas (DWG/PDF)")
+                    s_form = gr.Checkbox(label="Formulario RT")
+                    btn3   = gr.Button("Verificar Pacote", variant="primary")
+                with gr.Column():
+                    o_siapi = gr.Textbox(label="Relatorio SIAPI", lines=20)
+            btn3.click(aba_siapi,
+                       inputs=[s_nome, s_proc, s_rt, s_crea, s_mem, s_pran, s_form],
+                       outputs=[o_siapi])
+        with gr.TabItem("Sobre"):
+            gr.Markdown("""
+## Agente CBMGO
+Ferramenta de apoio para elaboracao de projetos de PSCIP conforme NT-01/2025 CBMGO.
+
+### Modulos Disponiveis
+- **Calculadora NT-01**: Calculos de extintores, hidrantes, reserva tecnica e lista de exigencias
+- **Auditoria**: Verifica conformidade do memorial descritivo com NT-01/2025
+- **Preparacao SIAPI**: Valida completude do pacote para protocolo no SIAPI
+
+### Em Desenvolvimento
+- RAG Normativo com FAISS (consulta semantica a NT-01/2025)
+- Classificador LLM (fine-tune Llama-3.1-8B via LoRA)
+- Gerador de Memorial LLM (fine-tune Mistral-7B)
+- Assistente CAD (scripts AutoCAD / Dynamo Revit)
+
+### Referencias Normativas
+NT-01/2025 CBMGO | ABNT NBR 13714 | NBR 10897 | NBR 13434 | NBR 10898 | NBR 5419
+
+Repositorio: https://huggingface.co/carlosh10/CERCON
+
+*Aviso Legal: Ferramenta auxiliar. O RT deve verificar conformidade com legislacao vigente.*
+""")
+
+if __name__ == "__main__":
+    demo.launch()