Update app.py

app.py

@@ -481,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')
@@ -501,52 +501,23 @@ def comparar_peticao_melhorada(caso, peticao_texto, correcao):
     corr_body = "\n".join(corr_lines[start_corr:]).strip()
 
     diff = difflib.ndiff(orig_body.splitlines(), corr_body.splitlines())
-    secoes = {
-        "ENDEREÇAMENTO": [], "QUALIFICAÇÃO": [], "FATOS": [], "FUNDAMENTOS": [], "PEDIDOS": [], "FECHAMENTO": [], "OUTROS": []
-    }
-
-    def identificar_secao(linha):
-        linha_upper = linha.upper()
-        if "EXCELENTÍSSIMO" in linha_upper or "JUIZ" in linha_upper or "VARA" in linha_upper:
-            return "ENDEREÇAMENTO"
-        elif "QUALIFICAÇÃO" in linha_upper or "AUTOR" in linha_upper or "RÉU" in linha_upper:
-            return "QUALIFICAÇÃO"
-        elif "DOS FATOS" in linha_upper:
-            return "FATOS"
-        elif "DIREITO" in linha_upper or "FUNDAMENTO" in linha_upper:
-            return "FUNDAMENTOS"
-        elif "PEDIDO" in linha_upper or "REQUER" in linha_upper:
-            return "PEDIDOS"
-        elif "TERMOS" in linha_upper and "DEFERIMENTO" in linha_upper:
-            return "FECHAMENTO"
-        else:
-            return "OUTROS"
 
-    secao_atual = "OUTROS"
     resultado_diff = []
     criticas = []
     for linha in diff:
         if linha.startswith('  '):
             texto = linha[2:]
-            nova_secao = identificar_secao(texto)
-            if nova_secao != "OUTROS":
-                secao_atual = nova_secao
             resultado_diff.append(f"<div class='linha-igual'>{escape(texto)}</div>")
         elif linha.startswith('- '):
             texto = linha[2:]
             if any(kw in texto.lower() for kw in ["invenção", "custas", "art. 319"]):
                 criticas.append(f"🔴 REMOVIDO: {texto}")
             resultado_diff.append(f"<div class='linha-removida'><del>{escape(texto)}</del></div>")
-            secoes[secao_atual].append(f"- REMOVIDO: {texto}")
         elif linha.startswith('+ '):
             texto = linha[2:]
-            nova_secao = identificar_secao(texto)
-            if nova_secao != "OUTROS":
-                secao_atual = nova_secao
             if any(kw in texto.lower() for kw in ["custas", "art. 319"]):
                 criticas.append(f"🟢 ADICIONADO: {texto}")
             resultado_diff.append(f"<div class='linha-adicionada'><ins>{escape(texto)}</ins></div>")
-            secoes[secao_atual].append(f"+ ADICIONADO: {texto}")
         elif linha.startswith('? '):
             continue
 
@@ -560,23 +531,6 @@ 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("-"))
-            alteracoes_texto += f"""
-### {secao}
-| Tipo         | Quantidade |
-|--------------|------------|
-| Adicionadas  | {adicionadas}      |
-| Removidas    | {removidas}      |
-"""
-            for alt in alteracoes[:5]:
-                texto = alt[:60] + "..." if len(alt) > 60 else alt
-                alteracoes_texto += f"  {texto}\n"
-            if len(alteracoes) > 5:
-                alteracoes_texto += f"  - ... e mais {len(alteracoes) - 5} alterações\n"
 
     visualizacao_detalhada = f"""
 <style>
@@ -586,18 +540,21 @@ Diferença Líquida: {total_adicionadas - total_removidas} linhas
 </style>
 {"".join(resultado_diff)}
 """
-    return peticao, peticao_corrigida, resumo_estatistico, criticas_texto, alteracoes_texto, visualizacao_detalhada
+    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)
+    _, _, 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>
@@ -610,7 +567,7 @@ def exportar_diferencas_html(caso, peticao_texto, correcao):
 </body>
 </html>
 """
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp:
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf", mode='w') as tmp:
         HTML(string=html_content).write_pdf(tmp.name)
         return tmp.name
 
@@ -684,7 +641,7 @@ def interface_gradio():
 
         with gr.Row():
             with gr.Column(scale=1):
-                alteracoes_texto_output = gr.Markdown(label="🔍 Alterações por Seção")
+                alteracoes_texto_output = gr.Markdown(label="🔍 Alterações por Seção", visible=False)
             with gr.Column(scale=2):
                 resumo_alteracoes_output = gr.Markdown(label="📊 Resumo das Alterações")
                 criticas_output = gr.Markdown(label="🚨 Mudanças Críticas")
@@ -724,7 +681,7 @@ def interface_gradio():
             fn=comparar_peticao_melhorada,
             inputs=[caso_input, peticao_texto, resultado_output],
             outputs=[peticao_aluno_output, peticao_corrigida_output, resumo_alteracoes_output, criticas_output, 
-                     alteracoes_texto_output, visualizacao_detalhada_output]
+                     visualizacao_detalhada_output]
         ).then(
             fn=lambda: gr.update(visible=True),
             outputs=[download_btn]
@@ -733,6 +690,9 @@ def interface_gradio():
             fn=exportar_diferencas_html,
             inputs=[caso_input, peticao_texto, resultado_output],
             outputs=[download_file]
+        ).then(
+            fn=lambda x: [gr.update(visible=True, value=x), gr.update(visible=True)],
+            outputs=[download_file, download_btn]
         )
 
         with gr.Accordion("📘 Critérios Detalhados", open=False):