Update app.py

app.py

@@ -1,4 +1,6 @@
 import gradio as gr
+import pyperclip
+import re
 
 def obter_documentos_padrao_2010_mais():
     return [
@@ -21,10 +23,10 @@ def obter_documentos_carencia():
     return [
         "3 últimos boletos pagos e seus comprovantes",
         "Carteirinha",
-        "Carta de permanência (solicitar à administradora do seu plano)"
+        "Carta de permanência (solicitar à administradora do plano, geralmente por ligação)"
     ]
 
-def obter_documentos_titular(categoria, modalidade, documentos_padrao, incluir_carencia):
+def obter_documentos_titular(categoria, modalidade, documentos_padrao):
     documentos = documentos_padrao.copy()
     documentos.append("Comprovante de Residência (no nome do titular ou dos pais)")
 
@@ -36,18 +38,16 @@ def obter_documentos_titular(categoria, modalidade, documentos_padrao, incluir_c
             documentos.extend(["CNPJ", "Contrato Social e suas alterações"])
 
     elif categoria == "ADESAO":
-        if modalidade != "ENTIDADE ABERTA":
-            documentos_map = {
-                "ESTUDANTE": "Declaração Escolar",
-                "TRABALHADOR": "Contracheque",
-                "PROFISSIONAL LIBERAL": "Documento de Registro no Conselho na validade + Diploma",
-                "APOSENTADO": "Comprovante de Pensão"
-            }
-            documentos.append(documentos_map.get(modalidade, "Modalidade inválida!"))
-
-    if incluir_carencia:
-        documentos.append("\nDocumentação para Compra de Carência:")
-        documentos.extend(obter_documentos_carencia())
+        documentos_map = {
+            "ENTIDADE ABERTA": None,
+            "ESTUDANTE": "Declaração Escolar",
+            "TRABALHADOR": "Contracheque",
+            "PROFISSIONAL LIBERAL": "Documento de Registro no Conselho na validade + Diploma",
+            "APOSENTADO": "Comprovante de Pensão"
+        }
+        doc_extra = documentos_map.get(modalidade)
+        if doc_extra:
+            documentos.append(doc_extra)
 
     return documentos
 
@@ -70,7 +70,17 @@ def simplificar_dependente(nome):
     }
     return simplificacoes.get(nome, nome)
 
-def obter_documentos_dependente(dependente, nascido_2010, incluir_carencia):
+def formatar_documentos(nome, documentos):
+    resultado = f"<h3>Documentos necessários para {nome}:</h3><ul>"
+    for doc in documentos:
+        if doc == "DA EMPRESA:":
+            resultado += f"</ul><h4>{doc}</h4><ul>"
+        else:
+            resultado += f"<li>{doc}</li>"
+    resultado += "</ul>"
+    return resultado
+
+def obter_documentos_dependente(dependente, nascido_2010):
     documentos = obter_documentos_padrao_2010_mais() if nascido_2010 else obter_documentos_padrao_2009_menos()
 
     if dependente == "Cônjuge ou companheiro(a)":
@@ -84,86 +94,179 @@ def obter_documentos_dependente(dependente, nascido_2010, incluir_carencia):
     elif dependente == "Sogro(a)":
         documentos.append("Certidão de casamento do titular")
 
-    if incluir_carencia:
-        documentos.append("\nDocumentação para Compra de Carência:")
-        documentos.extend(obter_documentos_carencia())
-
     return documentos
 
-def imprimir_documentos(nome, documentos):
-    resultado = f"\nDocumentos necessários para {nome}:\n"
-    for doc in documentos:
-        if doc == "DA EMPRESA:" or "Documentação para Compra de Carência" in doc:
-            resultado += f"\n{doc}\n"
-        else:
-            resultado += f"- {doc}\n"
-    return resultado
-
-def gerar_documentos(titular_nascido_2010_mais, categoria, modalidade, compra_carencia_titular,
-                     nomes_dependentes, nascido_2010_dependentes, compra_carencia_dependentes):
+def atualizar_lista_inclusos(titular_nascido_2010, categoria, modalidade, titular_carencia, dependentes_info):
+    lista_inclusos = []
     
-    documentos_padrao_titular = obter_documentos_padrao_2010_mais() if titular_nascido_2010_mais else obter_documentos_padrao_2009_menos()
-    incluir_carencia_titular = compra_carencia_titular
-
-    resultado = ""
-    documentos_titular = obter_documentos_titular(categoria, modalidade, documentos_padrao_titular, incluir_carencia_titular)
-    resultado += imprimir_documentos("TITULAR", documentos_titular)
-
-    # Prepara lista de dependentes válidos
-    dependentes = []
-    for i in range(len(nomes_dependentes)):
-        nome = nomes_dependentes[i].strip()
-        if nome == "":
-            continue
-        dependentes.append((nome, nascido_2010_dependentes[i], compra_carencia_dependentes[i]))
-
-    if dependentes:
-        carencia_flags = [d[2] for d in dependentes]
-        if all(carencia_flags):
-            # Todos compram carência
-            documentos_todos = []
-            for dep, nascido_2010, _ in dependentes:
-                documentos_todos.extend(obter_documentos_dependente(dep, nascido_2010, incluir_carencia=True))
-            resultado += imprimir_documentos("Dependentes: TODOS", documentos_todos)
-        else:
-            # Individual
-            for dep, nascido_2010, carencia in dependentes:
-                if carencia:
-                    documentos_dep = obter_documentos_dependente(dep, nascido_2010, incluir_carencia=True)
-                    nome_simplificado = simplificar_dependente(dep)
-                    resultado += imprimir_documentos(f"Dependente: {nome_simplificado}", documentos_dep)
-
-    return resultado
-
-
-with gr.Blocks() as demo:
-    gr.Markdown("### Gerador de Documentos Saúde")
-
-    titular_nascido_2010_mais = gr.Checkbox(label="Titular nasceu em 2010 ou depois?")
-    categoria = gr.Dropdown(label="Categoria", choices=["CPF", "CNPJ", "ADESAO"], value="CPF")
-    modalidade = gr.Dropdown(label="Modalidade", choices=["MEI", "OUTROS", "ESTUDANTE", "TRABALHADOR", "PROFISSIONAL LIBERAL", "APOSENTADO", "ENTIDADE ABERTA"], value="MEI")
-    compra_carencia_titular = gr.Checkbox(label="Compra carência - Titular?")
-
-    gr.Markdown("### Dependentes (até 5)")
+    # Adiciona titular
+    tipo_titular = "Titular (2010+)" if titular_nascido_2010 else "Titular (2009-)"
+    if titular_carencia:
+        tipo_titular += " + carência"
+    lista_inclusos.append([tipo_titular, categoria, modalidade if modalidade else "PF"])
+    
+    # Adiciona dependentes
+    if dependentes_info:
+        for dep_info in dependentes_info:
+            dep_nascido_2010, dep_tipo, dep_carencia = dep_info
+            tipo_dep = f"Dep. {simplificar_dependente(dep_tipo)} (2010+)" if dep_nascido_2010 else f"Dep. {simplificar_dependente(dep_tipo)} (2009-)"
+            if dep_carencia:
+                tipo_dep += " + carência"
+            lista_inclusos.append([tipo_dep, "Dependente", dep_tipo])
+    
+    return lista_inclusos
 
-    nomes_dependentes = []
-    nascido_2010_dependentes = []
-    compra_carencia_dependentes = []
+def limpar_texto_para_copiar(texto):
+    # Remove tags HTML para ter um texto limpo
+    texto_limpo = re.sub('<[^<]+?>', '', texto)
+    texto_limpo = texto_limpo.replace('</ul>', '\n').replace('</li>', '\n')
+    texto_limpo = re.sub('\n+', '\n', texto_limpo).strip()
+    return texto_limpo
 
-    for i in range(5):
-        nomes_dependentes.append(gr.Textbox(label=f"Nome dependente {i+1}"))
-        nascido_2010_dependentes.append(gr.Checkbox(label=f"Dependente {i+1} nasceu em 2010 ou depois?"))
-        compra_carencia_dependentes.append(gr.Checkbox(label=f"Dependente {i+1} compra carência?"))
+def copiar_para_area_transferencia(texto):
+    try:
+        if not texto:
+            return "❌ Nada para copiar - gere a lista primeiro"
+        
+        texto_limpo = limpar_texto_para_copiar(texto)
+        pyperclip.copy(texto_limpo)
+        return "✅ Copiado para área de transferência!"
+    except Exception as e:
+        return f"❌ Erro ao copiar: {str(e)}"
 
-    btn_gerar = gr.Button("Gerar documentos")
-    output = gr.Textbox(label="Resultado", lines=30)
+def atualizar_modalidades(categoria):
+    if categoria == "CNPJ":
+        return gr.Dropdown(choices=["MEI", "OUTROS"], label="Modalidade CNPJ", interactive=True)
+    elif categoria == "ADESAO":
+        return gr.Dropdown(
+            choices=["ENTIDADE ABERTA", "ESTUDANTE", "TRABALHADOR", "PROFISSIONAL LIBERAL", "APOSENTADO"],
+            label="Modalidade Adesão",
+            interactive=True
+        )
+    else:
+        return gr.Dropdown(choices=[], label="Modalidade (não aplicável)", interactive=False)
 
-    btn_gerar.click(
-        gerar_documentos,
-        inputs=[titular_nascido_2010_mais, categoria, modalidade, compra_carencia_titular,
-                nomes_dependentes, nascido_2010_dependentes, compra_carencia_dependentes],
-        outputs=output
-    )
+def criar_interface():
+    with gr.Blocks(title="Documentos para Plano de Saúde", theme=gr.themes.Soft()) as app:
+        gr.Markdown("# 📋 Documentos Necessários para Plano de Saúde")
+        gr.Markdown("Preencha os dados abaixo para obter a lista de documentos necessários.")
+        
+        with gr.Row():
+            with gr.Column():
+                # Seção do Titular
+                with gr.Group():
+                    gr.Markdown("### Titular")
+                    categoria = gr.Radio(
+                        choices=["PF", "CNPJ", "ADESAO"], 
+                        label="Categoria do Titular",
+                        value="PF"
+                    )
+                    modalidade = gr.Dropdown(
+                        choices=[], 
+                        label="Modalidade (se aplicável)",
+                        interactive=False
+                    )
+                    with gr.Row():
+                        titular_nascido_2010 = gr.Checkbox(label="Nasceu em 2010 ou depois?")
+                        titular_carencia = gr.Checkbox(label="Fará compra de carência?")
+                
+                # Seção de Dependentes
+                with gr.Accordion("Adicionar Dependentes", open=True) as accordion_deps:
+                    with gr.Group():
+                        gr.Markdown("### Novo Dependente")
+                        dep_tipo = gr.Dropdown(
+                            choices=[
+                                "Cônjuge ou companheiro(a)",
+                                "Filhos(as) solteiros naturais, tutelados, enteados e adotivos",
+                                "Filhos(as) inválidos declarados no imposto de renda do titular",
+                                "Pai e mãe",
+                                "Padrasto e madrasta",
+                                "Irmãos(ãs) consanguíneos ou adotivos",
+                                "Netos",
+                                "Sobrinhos",
+                                "Primos",
+                                "Tios",
+                                "Cunhados(as)",
+                                "Sogro(a)",
+                                "Genros e noras",
+                                "Avô e avó"
+                            ],
+                            label="Tipo de Dependente"
+                        )
+                        with gr.Row():
+                            dep_nascido_2010 = gr.Checkbox(label="Nasceu em 2010 ou depois?")
+                            dep_carencia = gr.Checkbox(label="Fará compra de carência?")
+                        btn_adicionar_dep = gr.Button("➕ Adicionar Dependente", variant="primary")
+                    
+                    lista_inclusos = gr.Dataframe(
+                        headers=["Inclusos nessa cotação", "Tipo", "Detalhes"],
+                        datatype=["str", "str", "str"],
+                        interactive=False,
+                        label="Inclusos nessa cotação",
+                        value=[["Titular (2009-)", "PF", ""]]
+                    )
+                    dependentes_info = gr.State([])
+                
+                btn_submit = gr.Button("🔍 Gerar Lista de Documentos", variant="primary")
+            
+            with gr.Column():
+                output = gr.HTML(label="Documentos Necessários")
+                btn_copiar = gr.Button("📋 Copiar Lista")
+                status_copia = gr.Textbox(label="Status", interactive=False, visible=False)
+        
+        # Atualiza modalidades quando categoria muda
+        categoria.change(
+            atualizar_modalidades,
+            inputs=categoria,
+            outputs=modalidade
+        )
+        
+        # Atualiza lista de inclusos em tempo real
+        inputs_lista = [titular_nascido_2010, categoria, modalidade, titular_carencia, dependentes_info]
+        for input_component in inputs_lista:
+            input_component.change(
+                atualizar_lista_inclusos,
+                inputs=inputs_lista,
+                outputs=lista_inclusos
+            )
+        
+        # Adiciona dependentes à lista
+        btn_adicionar_dep.click(
+            lambda dep_info, dep_nasc, dep_tipo, dep_carencia: (dep_info + [(dep_nasc, dep_tipo, dep_carencia)]), 
+            inputs=[dependentes_info, dep_nascido_2010, dep_tipo, dep_carencia],
+            outputs=dependentes_info
+        ).then(
+            atualizar_lista_inclusos,
+            inputs=inputs_lista,
+            outputs=lista_inclusos
+        )
+        
+        # Processa o formulário
+        btn_submit.click(
+            processar_documentos,
+            inputs=[
+                titular_nascido_2010,
+                categoria,
+                modalidade,
+                titular_carencia,
+                dependentes_info
+            ],
+            outputs=output
+        )
+        
+        # Copia para área de transferência
+        btn_copiar.click(
+            copiar_para_area_transferencia,
+            inputs=output,
+            outputs=status_copia
+        ).then(
+            lambda: gr.update(visible=True),
+            None,
+            status_copia
+        )
+    
+    return app
 
-if __name__ == "__main__":
-    demo.launch()
+# Criar e lançar a interface
+interface = criar_interface()
+interface.launch()