Update app.py

app.py

@@ -470,39 +470,6 @@ Diferença Líquida: {total_adicionadas - total_removidas} linhas
 """
     return peticao, peticao_corrigida, resumo_estatistico, criticas_texto, visualizacao_detalhada
 
-def exportar_diferencas_html(caso, peticao_texto, correcao):
-    _, _, resumo_estatistico, criticas_texto, visualizacao_detalhada = comparar_peticao_melhorada(caso, peticao_texto, correcao)
-    
-    html_content = f"""
-<html>
-<head>
-    <title>Diferenças na Petição</title>
-    <style>
-        body {{ font-family: Arial, sans-serif; margin: 20px; }}
-        .linha-igual {{ color: #000000; margin: 2px 0; }}
-        .linha-removida {{ color: #ffffff; background-color: #b30000; margin: 2px 0; padding: 2px; }}
-        .linha-adicionada {{ color: #ffffff; background-color: #006600; margin: 2px 0; padding: 2px; }}
-        h2 {{ color: #333; }}
-        pre {{ white-space: pre-wrap; }}
-    </style>
-</head>
-<body>
-    <h2>Estatísticas</h2>
-    <pre>{resumo_estatistico}</pre>
-    <h2>Mudanças Críticas</h2>
-    <pre>{criticas_texto}</pre>
-    <h2>Alterações Detalhadas</h2>
-    {visualizacao_detalhada}
-</body>
-</html>
-"""
-    buffer = io.BytesIO()
-    buffer.write(html_content.encode('utf-8'))
-    buffer.seek(0)
-    
-    nome_arquivo = f"diferencas_peticao_{datetime.now().strftime('%Y%m%d_%H%M%S')}.html"
-    return gr.File(value=buffer, label=nome_arquivo)
-
 def interface_gradio():
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
         gr.Markdown("# 📝 Corretor Avançado de Petições Iniciais (Critérios OAB)")
@@ -560,6 +527,7 @@ def interface_gradio():
                         arquivo_preview = gr.Textbox(label="Prévia do Conteúdo do Arquivo", lines=5, interactive=False)
                         btn_arquivo = gr.Button("🔍 Corrigir Petição (Arquivo)", variant="primary")
                         peticao_arquivo.change(fn=process_file, inputs=[peticao_arquivo], outputs=[arquivo_preview])
+                processing_message = gr.Markdown("", visible=False)  # Aviso de processamento
 
         with gr.Row():
             clear_btn = gr.Button("🗑️ Limpar Todos os Campos")
@@ -579,8 +547,7 @@ def interface_gradio():
                 visualizacao_detalhada_output = gr.HTML(label="📝 Alterações Detalhadas")
 
         with gr.Row():
-            download_btn = gr.Button("📥 Baixar Diferenças em HTML", visible=False)
-            download_file = gr.File(label="Download", visible=False)
+            copy_btn = gr.Button("📋 Copiar Diferenças", visible=False)
 
         with gr.Accordion("📜 Histórico de Correções", open=False):
             historico_output = gr.Markdown()
@@ -590,23 +557,44 @@ def interface_gradio():
                 outputs=historico_output
             )
 
+        def copy_differences(visualizacao_detalhada):
+            texto = "\n".join([linha for linha in visualizacao_detalhada.split('\n') if not linha.strip().startswith(('<style>', '</style>'))])
+            texto = texto.replace('<div class="linha-igual">', '').replace('<div class="linha-removida"><del>', 'REMOVIDO: ').replace('<div class="linha-adicionada"><ins>', 'ADICIONADO: ').replace('</div>', '').replace('</del>', '').replace('</ins>', '')
+            return texto
+
         caso_btn.click(fn=gerar_caso_ficticio, inputs=[], outputs=[caso_input])
         btn_texto.click(
+            fn=lambda: gr.update(value="⏳ Aguarde enquanto a correção é gerada...", visible=True),
+            inputs=None,
+            outputs=[processing_message]
+        ).then(
             fn=lambda caso, texto, treinamento: corrigir_peticao(caso, texto, None, treinamento),
             inputs=[caso_input, peticao_texto, modo_treinamento],
             outputs=[resultado_output]
+        ).then(
+            fn=lambda: gr.update(value="", visible=False),
+            inputs=None,
+            outputs=[processing_message]
         )
         btn_arquivo.click(
+            fn=lambda: gr.update(value="⏳ Aguarde enquanto a correção é gerada...", visible=True),
+            inputs=None,
+            outputs=[processing_message]
+        ).then(
             fn=lambda caso, arquivo, treinamento: corrigir_peticao(caso, None, arquivo, treinamento),
             inputs=[caso_input, peticao_arquivo, modo_treinamento],
             outputs=[resultado_output]
+        ).then(
+            fn=lambda: gr.update(value="", visible=False),
+            inputs=None,
+            outputs=[processing_message]
         )
         clear_btn.click(
             fn=lambda: [None, None, None, None, "", "", "", "", "", gr.update(visible=False), gr.update(visible=False)],
             inputs=None,
             outputs=[caso_input, peticao_texto, peticao_arquivo, arquivo_preview, resultado_output, dica_contextual,
                      peticao_aluno_output, peticao_corrigida_output, resumo_alteracoes_output, criticas_output,
-                     visualizacao_detalhada_output, download_btn, download_file]
+                     visualizacao_detalhada_output, copy_btn, processing_message]
         )
         comparar_btn.click(
             fn=comparar_peticao_melhorada,
@@ -615,15 +603,13 @@ def interface_gradio():
                      visualizacao_detalhada_output]
         ).then(
             fn=lambda: gr.update(visible=True),
-            outputs=[download_btn]
+            outputs=[copy_btn]
         )
-        download_btn.click(
-            fn=exportar_diferencas_html,
-            inputs=[caso_input, peticao_texto, resultado_output],
-            outputs=[download_file]
-        ).then(
-            fn=lambda x: gr.update(visible=True, value=x),
-            outputs=[download_file]
+        copy_btn.click(
+            fn=copy_differences,
+            inputs=[visualizacao_detalhada_output],
+            outputs=[gr.Textbox(visible=False)],
+            _js="function(text) { navigator.clipboard.writeText(text); alert('Diferenças copiadas para a área de transferência!'); return text; }"
         )
 
         with gr.Accordion("📘 Critérios Detalhados", open=False):