Update app.py

app.py

@@ -69,7 +69,6 @@ def simplificar_dependente(nome):
     }
     return simplificacoes.get(nome, nome)
 
-# --- NOVA FUNÇÃO PARA FORMATAR APENAS PARA TEXTO PURO ---
 def formatar_documentos_para_whatsapp(nome, documentos):
     resultado = f"Documentos necessários para {nome}:\n"
     for doc in documentos:
@@ -77,7 +76,7 @@ def formatar_documentos_para_whatsapp(nome, documentos):
             resultado += f"\nDA EMPRESA:\n"
         else:
             resultado += f"- {doc}\n"
-    resultado += "\n" # Adiciona uma linha em branco para separar as seções
+    resultado += "\n"
     return resultado
 
 def obter_documentos_dependente(dependente, nascido_2010):
@@ -96,7 +95,6 @@ def obter_documentos_dependente(dependente, nascido_2010):
 
     return documentos
 
-# --- NOVA FUNÇÃO PRINCIPAL QUE GERA APENAS TEXTO PURO ---
 def processar_documentos_para_whatsapp(titular_nascido_2010, categoria, modalidade, titular_carencia, dependentes_info):
     texto_final = ""
     lista_completa = []
@@ -124,8 +122,7 @@ def processar_documentos_para_whatsapp(titular_nascido_2010, categoria, modalida
     for nome, documentos in lista_completa:
         texto_final += formatar_documentos_para_whatsapp(nome, documentos)
     
-    # Retorna o texto puro para o Textbox e torna o botão de cópia visível
-    return texto_final.strip(), gr.update(visible=True) 
+    return texto_final.strip(), gr.update(visible=True), gr.update(visible=True) # Retorna texto, e torna visíveis o btn_copiar e status_copia
 
 def atualizar_lista_inclusos(titular_nascido_2010, categoria, modalidade, titular_carencia, dependentes_info):
     lista_inclusos = []
@@ -145,6 +142,23 @@ def atualizar_lista_inclusos(titular_nascido_2010, categoria, modalidade, titula
     
     return lista_inclusos
 
+# --- NOVA FUNÇÃO PARA RESETAR TODOS OS CAMPOS ---
+def limpar_tudo():
+    return (
+        gr.update(value="PF"),                      # categoria
+        gr.update(choices=[], interactive=False, value=None), # modalidade (reinicia para vazio e desativado)
+        gr.update(value=False),                     # titular_nascido_2010
+        gr.update(value=False),                     # titular_carencia
+        gr.update(value=None),                      # dep_tipo
+        gr.update(value=False),                     # dep_nascido_2010
+        gr.update(value=False),                     # dep_carencia
+        gr.update(value=[["Titular (2009-)", "PF", ""]]), # lista_inclusos (volta ao estado inicial)
+        [],                                         # dependentes_info (limpa a lista de dependentes do estado)
+        "",                                         # output_textbox (limpa o texto de saída)
+        gr.update(visible=False),                   # btn_copiar (oculta o botão de cópia)
+        gr.update(visible=False, value="")          # status_copia (oculta e limpa o status da cópia)
+    )
+
 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")
@@ -204,30 +218,31 @@ def criar_interface():
                     )
                     dependentes_info = gr.State([])
                 
-                btn_submit = gr.Button("🔍 Gerar Lista de Documentos", variant="primary")
+                with gr.Row(): # Agrupando botões de ação
+                    btn_submit = gr.Button("🔍 Gerar Lista de Documentos", variant="primary")
+                    btn_limpar = gr.Button("🧹 Limpar Tudo", variant="secondary") # Novo botão Limpar
             
             with gr.Column():
-                # A única caixa de texto para saída e cópia
                 output_textbox = gr.Textbox(
                     label="Documentos Necessários (Pronto para copiar e colar no WhatsApp!)",
-                    interactive=False, # Não permite que o usuário digite
-                    lines=20, # Aumenta o espaço para visualizar
-                    elem_id="documentos-output-textbox" # ID para o JS, para o botão de cópia
+                    interactive=False,
+                    lines=20,
+                    elem_id="documentos-output-textbox"
                 )
                 btn_copiar = gr.Button("📋 Copiar Lista", visible=False) 
                 status_copia = gr.Textbox(label="Status da Cópia", interactive=False, visible=False)
         
         def atualizar_modalidades(categoria):
             if categoria == "CNPJ":
-                return gr.Dropdown(choices=["MEI", "OUTROS"], label="Modalidade CNPJ", interactive=True)
+                return gr.Dropdown(choices=["MEI", "OUTROS"], label="Modalidade CNPJ", interactive=True, value=None) # Resetar valor
             elif categoria == "ADESAO":
                 return gr.Dropdown(
                     choices=["ENTIDADE ABERTA", "ESTUDANTE", "TRABALHADOR", "PROFISSIONAL LIBERAL", "APOSENTADO"],
                     label="Modalidade Adesão",
-                    interactive=True
-                )
+                    interactive=True, value=None
+                ) # Resetar valor
             else:
-                return gr.Dropdown(choices=[], label="Modalidade (não aplicável)", interactive=False)
+                return gr.Dropdown(choices=[], label="Modalidade (não aplicável)", interactive=False, value=None) # Resetar valor
 
         categoria.change(
             atualizar_modalidades,
@@ -253,9 +268,8 @@ def criar_interface():
             outputs=lista_inclusos
         )
         
-        # O botão de submissão agora atualiza o único textbox e torna o botão de cópia visível
         btn_submit.click(
-            processar_documentos_para_whatsapp, # Funcao que gera texto puro
+            processar_documentos_para_whatsapp, 
             inputs=[
                 titular_nascido_2010,
                 categoria,
@@ -263,7 +277,7 @@ def criar_interface():
                 titular_carencia,
                 dependentes_info
             ],
-            outputs=[output_textbox, btn_copiar] 
+            outputs=[output_textbox, btn_copiar, status_copia] # Passamos status_copia também para que ele seja visível
         )
         
         btn_copiar.click(
@@ -272,13 +286,11 @@ def criar_interface():
             status_copia,
             js="""
             async () => {
-                // Pega o valor do textarea dentro do gr.Textbox usando o elem_id
                 const texto = document.getElementById("documentos-output-textbox").querySelector("textarea").value;
                 try {
                     await navigator.clipboard.writeText(texto);
                     return "✅ Copiado para área de transferência!";
                 } catch (err) {
-                    // Fallback para navegadores mais antigos ou erros
                     const textarea = document.createElement('textarea');
                     textarea.value = texto;
                     document.body.appendChild(textarea);
@@ -290,10 +302,30 @@ def criar_interface():
             }
             """
         ).then(
-            lambda: gr.update(visible=True),
+            lambda: gr.update(visible=True), # Aqui o status_copia já foi tornado visível acima. Isso só garante que se por algum motivo ele sumisse, voltaria.
             None,
             status_copia
         )
+
+        # --- CONEXÃO DO BOTÃO LIMPAR TUDO ---
+        btn_limpar.click(
+            limpar_tudo,
+            inputs=None, # Não precisa de inputs para limpar
+            outputs=[
+                categoria,
+                modalidade,
+                titular_nascido_2010,
+                titular_carencia,
+                dep_tipo,
+                dep_nascido_2010,
+                dep_carencia,
+                lista_inclusos,
+                dependentes_info, # O State precisa ser resetado!
+                output_textbox,
+                btn_copiar,
+                status_copia
+            ]
+        )
         
     return app