Update app.py

app.py

@@ -17,6 +17,7 @@ from datetime import datetime
 from ratelimiter import RateLimiter
 import difflib
 from html import escape
+from weasyprint import HTML  # Para gerar PDF com cores
 
 load_dotenv()
 
@@ -480,15 +481,15 @@ def calcular_progresso():
 
 def comparar_peticao_melhorada(caso, peticao_texto, correcao):
     if not caso:
-        return "⚠️ Forneça um caso antes de comparar.", "", "", "", ""
+        return "⚠️ Forneça um caso antes de comparar.", "", "", "", "", ""
     peticao = peticao_texto if peticao_texto else (HISTORICO[-1]["peticao"] if HISTORICO else "")
     if not peticao:
-        return "⚠️ Forneça uma petição para comparar ou corrija uma antes.", "", "", "", ""
+        return "⚠️ Forneça uma petição para comparar ou corrija uma antes.", "", "", "", "", ""
     if not correcao:
-        return "⚠️ Corrija a petição antes de comparar.", "", "", "", ""
+        return "⚠️ Corrija a petição antes de comparar.", "", "", "", "", ""
     ultima_correcao = HISTORICO[-1]["correcao_completa"] if HISTORICO else ""
     if not ultima_correcao:
-        return "⚠️ Nenhuma correção disponível no histórico.", "", "", "", ""
+        return "⚠️ Nenhuma correção disponível no histórico.", "", "", "", "", ""
     peticao_corrigida = extrair_peticao_corrigida(ultima_correcao)
 
     orig_lines = peticao.split('\n')
@@ -552,32 +553,31 @@ def comparar_peticao_melhorada(caso, peticao_texto, correcao):
     total_removidas = len([linha for linha in resultado_diff if "linha-removida" in linha])
     total_adicionadas = len([linha for linha in resultado_diff if "linha-adicionada" in linha])
 
-    resumo_alteracoes = []
+    resumo_estatistico = f"""
+Linhas Removidas: {total_removidas}
+Linhas Adicionadas: {total_adicionadas}
+Diferença Líquida: {total_adicionadas - total_removidas} linhas
+"""
+
+    criticas_texto = "\n".join(criticas) if criticas else "Nenhuma mudança crítica detectada."
+    alteracoes_texto = ""
     for secao, alteracoes in secoes.items():
         if alteracoes:
             adicionadas = sum(1 for alt in alteracoes if alt.startswith("+"))
             removidas = sum(1 for alt in alteracoes if alt.startswith("-"))
-            resumo_alteracoes.append(f"""
+            alteracoes_texto += f"""
 ### {secao}
 | Tipo         | Quantidade |
 |--------------|------------|
 | Adicionadas  | {adicionadas}      |
 | Removidas    | {removidas}      |
-""")
+"""
             for alt in alteracoes[:5]:
                 texto = alt[:60] + "..." if len(alt) > 60 else alt
-                resumo_alteracoes.append(f"  {texto}")
+                alteracoes_texto += f"  {texto}\n"
             if len(alteracoes) > 5:
-                resumo_alteracoes.append(f"  - ... e mais {len(alteracoes) - 5} alterações")
+                alteracoes_texto += f"  - ... e mais {len(alteracoes) - 5} alterações\n"
 
-    resumo_estatistico = f"""
-- **Linhas Removidas:** {total_removidas}
-- **Linhas Adicionadas:** {total_adicionadas}
-- **Diferença Líquida:** {total_adicionadas - total_removidas} linhas
-"""
-
-    criticas_texto = "\n".join(criticas) if criticas else "Nenhuma mudança crítica detectada."
-    alteracoes_texto = "".join(resumo_alteracoes)
     visualizacao_detalhada = f"""
 <style>
 .linha-igual {{ color: #000000; margin: 2px 0; }}
@@ -588,6 +588,32 @@ def comparar_peticao_melhorada(caso, peticao_texto, correcao):
 """
     return peticao, peticao_corrigida, resumo_estatistico, criticas_texto, alteracoes_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>
+        .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; }}
+    </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>
+"""
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp:
+        HTML(string=html_content).write_pdf(tmp.name)
+        return tmp.name
+
 def interface_gradio():
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
         gr.Markdown("# 📝 Corretor Avançado de Petições Iniciais (Critérios OAB)")
@@ -658,11 +684,16 @@ def interface_gradio():
 
         with gr.Row():
             with gr.Column(scale=1):
+                alteracoes_texto_output = gr.Markdown(label="🔍 Alterações por Seção")
+            with gr.Column(scale=2):
                 resumo_alteracoes_output = gr.Markdown(label="📊 Resumo das Alterações")
                 criticas_output = gr.Markdown(label="🚨 Mudanças Críticas")
-            with gr.Column(scale=2):
                 visualizacao_detalhada_output = gr.HTML(label="📝 Alterações Detalhadas")
 
+        with gr.Row():
+            download_btn = gr.Button("📥 Baixar Diferenças em PDF", visible=False)
+            download_file = gr.File(label="Download", visible=False)
+
         with gr.Accordion("📜 Histórico de Correções", open=False):
             historico_output = gr.Markdown()
             historico_btn = gr.Button("🔄 Ver Histórico")
@@ -683,17 +714,25 @@ def interface_gradio():
             outputs=[resultado_output]
         )
         clear_btn.click(
-            fn=lambda: [None, None, None, None, "", "", "", "", "", "", ""],
+            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]
+                     visualizacao_detalhada_output, download_btn, download_file]
         )
         comparar_btn.click(
             fn=comparar_peticao_melhorada,
             inputs=[caso_input, peticao_texto, resultado_output],
             outputs=[peticao_aluno_output, peticao_corrigida_output, resumo_alteracoes_output, criticas_output, 
-                     visualizacao_detalhada_output]
+                     alteracoes_texto_output, visualizacao_detalhada_output]
+        ).then(
+            fn=lambda: gr.update(visible=True),
+            outputs=[download_btn]
+        )
+        download_btn.click(
+            fn=exportar_diferencas_html,
+            inputs=[caso_input, peticao_texto, resultado_output],
+            outputs=[download_file]
         )
 
         with gr.Accordion("📘 Critérios Detalhados", open=False):