Update app.py

app.py

@@ -53,7 +53,123 @@ DICAS_OAB_MELHORADO = {
 HISTORICO = []
 CORRECAO_CACHE = {}
 
-# Funções auxiliares (processamento de arquivos) permanecem iguais, omitidas aqui por brevidade
+def preprocess_image_for_ocr(img):
+    try:
+        gray = img.convert('L')
+        np_img = np.array(gray)
+        _, thresh = cv2.threshold(np_img, 150, 255, cv2.THRESH_BINARY + cv2.THRESH_OTSU)
+        return Image.fromarray(thresh)
+    except Exception as e:
+        return f"❌ Erro no pré-processamento: {str(e)}"
+
+def extract_text_from_pdf(file_path_or_obj):
+    try:
+        if isinstance(file_path_or_obj, str):
+            with open(file_path_or_obj, 'rb') as f:
+                text = ""
+                with pdfplumber_open(f) as pdf:
+                    for page in pdf.pages:
+                        text += (page.extract_text() or "") + "\n"
+                if not text.strip():
+                    images = pdfplumber_open(f).pages
+                    for i, image in enumerate(images):
+                        try:
+                            img = image.to_image()
+                            preprocessed_img = preprocess_image_for_ocr(img)
+                            text += pytesseract.image_to_string(preprocessed_img) + "\n"
+                        except Exception as e:
+                            text += f"[Erro na página {i+1}: {str(e)}]\n"
+        else:
+            text = ""
+            with pdfplumber_open(file_path_or_obj) as pdf:
+                for page in pdf.pages:
+                    text += (page.extract_text() or "") + "\n"
+                if not text.strip():
+                    images = pdfplumber_open(file_path_or_obj).pages
+                    for i, image in enumerate(images):
+                        try:
+                            img = image.to_image()
+                            preprocessed_img = preprocess_image_for_ocr(img)
+                            text += pytesseract.image_to_string(preprocessed_img) + "\n"
+                        except Exception as e:
+                            text += f"[Erro na página {i+1}: {str(e)}]\n"
+        return text if text.strip() else "⚠️ Não foi possível extrair texto do PDF."
+    except Exception as e:
+        return f"❌ Erro ao processar o PDF: {str(e)}"
+
+def extract_text_from_docx(file_path_or_obj):
+    try:
+        if isinstance(file_path_or_obj, str):
+            with open(file_path_or_obj, 'rb') as f:
+                doc = docx.Document(f)
+                text = ""
+                for para in doc.paragraphs:
+                    text += para.text + "\n"
+                for table in doc.tables:
+                    for row in table.rows:
+                        for cell in row.cells:
+                            text += cell.text + " "
+                        text += "\n"
+        else:
+            doc = docx.Document(file_path_or_obj)
+            text = ""
+            for para in doc.paragraphs:
+                text += para.text + "\n"
+            for table in doc.tables:
+                for row in table.rows:
+                    for cell in row.cells:
+                        text += cell.text + " "
+                    text += "\n"
+        return text if text.strip() else "⚠️ Não foi possível extrair texto do DOCX."
+    except Exception as e:
+        return f"❌ Erro ao processar o DOCX: {str(e)}"
+
+def extract_text_from_txt(file_path_or_obj):
+    try:
+        if isinstance(file_path_or_obj, str):
+            with open(file_path_or_obj, 'rb') as f:
+                content = f.read()
+                encodings = ['utf-8', 'latin-1', 'cp1252', 'iso-8859-1']
+                for encoding in encodings:
+                    try:
+                        return content.decode(encoding)
+                    except UnicodeDecodeError:
+                        continue
+                return "❌ Não foi possível decodificar o arquivo TXT."
+        else:
+            content = file_path_or_obj.read()
+            encodings = ['utf-8', 'latin-1', 'cp1252', 'iso-8859-1']
+            for encoding in encodings:
+                try:
+                    return content.decode(encoding)
+                except UnicodeDecodeError:
+                    continue
+            return "❌ Não foi possível decodificar o arquivo TXT."
+    except Exception as e:
+        return f"❌ Erro ao processar o TXT: {str(e)}"
+
+def process_file(file_obj):
+    if file_obj is None:
+        return "Nenhum arquivo enviado."
+    try:
+        if hasattr(file_obj, 'name'):
+            file_path = file_obj.name
+            filename = os.path.basename(file_path).lower()
+            file_size = os.path.getsize(file_path)
+            if file_size > 10 * 1024 * 1024:
+                return "⚠️ O arquivo é muito grande. Máx. 10 MB."
+            if filename.endswith('.pdf'):
+                return extract_text_from_pdf(file_path)
+            elif filename.endswith('.docx'):
+                return extract_text_from_docx(file_path)
+            elif filename.endswith('.txt'):
+                return extract_text_from_txt(file_path)
+            else:
+                return "⚠️ Formato de arquivo não suportado. Use PDF, DOCX ou TXT."
+        else:
+            return "❌ Erro: Objeto de arquivo inválido."
+    except Exception as e:
+        return f"❌ Erro ao processar o arquivo: {str(e)}"
 
 def estruturar_prompt_analise(caso, peticao, modo_treinamento=False):
     """Estrutura o prompt apenas para a análise da petição."""
@@ -288,7 +404,127 @@ def extrair_peticao_corrigida(correcao_completa):
     except Exception as e:
         return f"❌ Erro ao extrair petição corrigida: {str(e)}"
 
-# Outras funções (comparar_peticao_melhorada, interface_gradio, etc.) permanecem iguais, omitidas por brevidade
+def fornecer_dica_contextual(texto):
+    dicas = [f"💡 {chave}: {dica}" for chave, dica in DICAS_OAB_MELHORADO.items() if chave.lower() in texto.lower()]
+    return "\n".join(dicas) if dicas else "💡 Escreva mais para receber dicas."
+
+def calcular_progresso():
+    if not HISTORICO:
+        return "📊 Nenhum progresso registrado."
+    notas = []
+    for entry in HISTORICO:
+        for line in entry["correcao_completa"].split("\n"):
+            if "Nota Final" in line:
+                try:
+                    nota = float(line.split()[-1].split("/")[0].replace(",", "."))
+                    notas.append(nota)
+                except:
+                    continue
+    media = sum(notas) / len(notas) if notas else 0
+    tendencia = "⬆️ Melhorando" if len(notas) > 1 and notas[-1] > notas[-2] else "⬇️ Estável ou piorando"
+    return f"📊 Progresso: Média {media:.1f}/10 | Tentativas: {len(notas)} | Tendência: {tendencia}"
+
+def comparar_peticao_melhorada(caso, peticao_texto, correcao):
+    if not caso:
+        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.", "", "", "", ""
+    if not correcao:
+        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.", "", "", "", ""
+
+    peticao_corrigida = extrair_peticao_corrigida(ultima_correcao)
+    if "⚠️" in peticao_corrigida or "❌" in peticao_corrigida:
+        return peticao_corrigida, "", "", "", ""
+
+    orig_lines = peticao.split('\n')
+    corr_lines = peticao_corrigida.split('\n')
+    start_orig = next((i for i, line in enumerate(orig_lines) if "AÇÃO" in line.upper() or "EXCELENTÍSSIMO" in line.upper()), 0)
+    start_corr = next((i for i, line in enumerate(corr_lines) if "AÇÃO" in line.upper() or "EXCELENTÍSSIMO" in line.upper()), 0)
+
+    orig_body = "\n".join(orig_lines[start_orig:]).strip()
+    corr_body = "\n".join(corr_lines[start_corr:]).strip()
+
+    diff = difflib.ndiff(orig_body.splitlines(), corr_body.splitlines())
+
+    resultado_diff = []
+    criticas = []
+    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}
+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."
+
+    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)}"
+    finally:
+        if 'temp_path' in locals() and os.path.exists(temp_path):
+            os.unlink(temp_path)
 
 def interface_gradio():
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
@@ -445,5 +681,44 @@ def interface_gradio():
 
     return demo
 
+def gerar_caso_ficticio():
+    prompt = estruturar_prompt_geracao_caso()
+    return chamar_api(prompt)
+
+def estruturar_prompt_geracao_caso():
+    return """
+    Crie um caso fictício detalhado para uma petição inicial cível, similar ao exemplo:
+    "Ana, modelo profissional, residente em Manaus, viajou para São Paulo, para o casamento de sua filha. Para lavar, pintar seus cabelos e realizar um penteado para o casamento, Ana procurou os serviços de João Macedo, cabeleireiro e dono do salão de beleza 'Hair', sediado na cidade de São Paulo, que lhe cobrou R$500,00 (quinhentos reais) pela prestação do serviço. Após lavar os cabelos de Ana, João aplicou-lhe uma tintura da marca francesa ABC, importada pela empresa Brasil Connection Ltda, sediada na cidade de Curitiba (PR). Meia hora após a aplicação da tintura, Ana sofreu uma reação alérgica, que demandou atendimento médico-hospitalar, no valor de R$1.000,00 (um mil reais), bem como dois dias de absoluto repouso que impossibilitou sua presença no casamento de sua filha. Além disso, perdeu grande parte de seu cabelo, tendo permanecido com manchas em seu rosto por dois meses, perdendo um ensaio fotográfico, para o qual já havia sido contratada, pelo valor de R$50.000,00 (cinquenta mil reais). Posteriormente constatou-se que a tintura utilizada continha substâncias químicas extremamente perigosas à vida e à saúde das pessoas e a fabricante ABC já havia sido condenada pela justiça francesa a encerrar a fabricação e comercialização do produto. Indignada com os danos sofridos, Ana procura um advogado para pleitear o devido ressarcimento."
+    O caso deve:
+    1. Conter uma situação clara de violação de direitos que justifique uma ação cível
+    2. Incluir valores específicos de danos materiais e/ou morais
+    3. Mencionar nomes completos, cidades e informações suficientes para identificar as partes
+    4. Estabelecer relação entre as partes (contratual, extracontratual, consumo, etc.)
+    5. Ter complexidade adequada para uma petição inicial de nível OAB
+    6. Ter entre 15-25 linhas de texto
+    7. Focar em um dos seguintes temas (escolha APENAS UM aleatoriamente):
+       - Responsabilidade civil por acidente de consumo
+       - Inadimplemento contratual com danos
+       - Responsabilidade civil por dano ambiental
+       - Violação de direitos da personalidade
+       - Relação de consumo com vício de produto/serviço
+       - Reparação por danos patrimoniais
+    Termine com: "Situação problema: Como advogado(a) de [NOME DO CLIENTE], promova a demanda cabível."
+    """
+
+def validar_peticao_basica(peticao):
+    problemas = []
+    if "EXCELENTÍSSIMO" not in peticao.upper():
+        problemas.append("⚠️ Endereçamento ausente ou incompleto (Art. 319, I, CPC).")
+    if "Art. 319" not in peticao and "CPC" not in peticao:
+        problemas.append("⚠️ Fundamentação jurídica não menciona o CPC (Art. 319).")
+    if "Dá-se à causa o valor" not in peticao:
+        problemas.append("⚠️ Valor da causa não especificado (Art. 319, V, CPC).")
+    if "juntada da guia de custas" not in peticao.lower():
+        problemas.append("⚠️ Pedido de juntada da guia de custas ausente (-1,0 ponto).")
+    if "Nestes termos" not in peticao:
+        problemas.append("⚠️ Fechamento formal ausente.")
+    return problemas
+
 app = interface_gradio()
 app.launch()