Spaces:

GabrielGD
/

CAPI

Runtime error

App Files Files Community

GabrielGD commited on Mar 3, 2025

Commit

740aded

verified ·

1 Parent(s): 4478f81

Update app.py

Browse files

Files changed (1) hide show

app.py +41 -18

app.py CHANGED Viewed

@@ -17,9 +17,7 @@ from datetime import datetime
 from ratelimiter import RateLimiter
 import difflib
 from html import escape
-from reportlab.lib.pagesizes import letter
-from reportlab.platypus import SimpleDocTemplate, Paragraph
-from reportlab.lib.styles import getSampleStyleSheet
 load_dotenv()
@@ -453,7 +451,7 @@ def formatar_correção(texto):
         if match:
             nota = match.group(1)
             texto_formatado = texto_formatado.replace(f"### ⭐ Nota Final\n\n{nota}", f"### ⭐ Nota Final: {nota}/10\n\n")
-    return texto_formatado
 def extrair_peticao_corrigida(correcao_completa):
     try:
@@ -515,22 +513,22 @@ def comparar_peticao_melhorada(caso, peticao_texto, correcao):
     for linha in diff:
         if linha.startswith('  '):
             texto = linha[2:]
-            resultado_diff.append(f"- {escape(texto)}")  # Simplificado para texto puro
         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} (inconsistente ou incorreto)")
-            resultado_diff.append(f"- REMOVIDO: {escape(texto)}")
         elif linha.startswith('+ '):
             texto = linha[2:]
             if any(kw in texto.lower() for kw in ["custas", "art. 319"]):
                 criticas.append(f"🟢 ADICIONADO: {texto} (essencial para conformidade)")
-            resultado_diff.append(f"- ADICIONADO: {escape(texto)}")
         elif linha.startswith('? '):
             continue
-    total_removidas = len([linha for linha in resultado_diff if "REMOVIDO" in linha])
-    total_adicionadas = len([linha for linha in resultado_diff if "ADICIONADO" in linha])
     resumo_estatistico = f"""
 Linhas Removidas: {total_removidas}
@@ -539,24 +537,49 @@ Diferença Líquida: {total_adicionadas - total_removidas} linhas
 """
     criticas_texto = "\n".join(criticas) if criticas else "Nenhuma mudança crítica detectada."
-    visualizacao_detalhada = "\n".join(resultado_diff)
     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)
-    content = f"Estatísticas:\n{resumo_estatistico}\n\nMudanças Críticas:\n{criticas_texto}\n\nAlterações Detalhadas:\n{visualizacao_detalhada}"
     try:
         with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp:
             temp_path = tmp.name
-            doc = SimpleDocTemplate(temp_path, pagesize=letter)
-            styles = getSampleStyleSheet()
-            story = [Paragraph(content.replace('\n', '<br/>'), styles['Normal'])]
-            doc.build(story)
             return temp_path
     except Exception as e:
-        return f"❌ Erro ao gerar PDF: {str(e)}"
     finally:
         if 'temp_path' in locals() and os.path.exists(temp_path):
             os.unlink(temp_path)
@@ -634,7 +657,7 @@ def interface_gradio():
                 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.Textbox(label="📝 Alterações Detalhadas", lines=10)  # Mudado para Textbox
         with gr.Row():
             download_btn = gr.Button("📥 Baixar Diferenças em PDF", visible=False)

 from ratelimiter import RateLimiter
 import difflib
 from html import escape
+from weasyprint import HTML  # Para gerar PDF com cores
 load_dotenv()
         if match:
             nota = match.group(1)
             texto_formatado = texto_formatado.replace(f"### ⭐ Nota Final\n\n{nota}", f"### ⭐ Nota Final: {nota}/10\n\n")
+    return texto_formatado  # Corrigido de 'texto_formatada' para 'texto_formatado'
 def extrair_peticao_corrigida(correcao_completa):
     try:
     for linha in diff:
         if linha.startswith('  '):
             texto = linha[2:]
+            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} (inconsistente ou incorreto)")
+            resultado_diff.append(f"<div class='linha-removida'><del>{escape(texto)}</del></div>")
         elif linha.startswith('+ '):
             texto = linha[2:]
             if any(kw in texto.lower() for kw in ["custas", "art. 319"]):
                 criticas.append(f"🟢 ADICIONADO: {texto} (essencial para conformidade)")
+            resultado_diff.append(f"<div class='linha-adicionada'><ins>{escape(texto)}</ins></div>")
         elif linha.startswith('? '):
             continue
+    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_estatistico = f"""
 Linhas Removidas: {total_removidas}
 """
     criticas_texto = "\n".join(criticas) if criticas else "Nenhuma mudança crítica detectada."
+    visualizacao_detalhada = f"""
+<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>
+{"".join(resultado_diff)}
+"""
     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>
+"""
     try:
         with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp:
             temp_path = tmp.name
+            HTML(string=html_content).write_pdf(temp_path)
             return temp_path
     except Exception as e:
+        return f"❌ Erro ao gerar PDF: {str(e)}"  # Retorna erro detalhado
     finally:
         if 'temp_path' in locals() and os.path.exists(temp_path):
             os.unlink(temp_path)
                 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)