Update app.py

app.py

@@ -44,17 +44,21 @@ PECAS_CONSTITUCIONAL = [
     "Recurso Ordinário (ROC)", "Recurso Extraordinário (RE)", "Reclamação"
 ]
 
+PECAS_PENAL = [
+    "Aleatória (Simulado Real)", "Relaxamento de Prisão em Flagrante", "Liberdade Provisória",
+    "Queixa-Crime", "Resposta à Acusação", "Memoriais", "Apelação",
+    "Recurso em Sentido Estrito (RESE)", "Agravo em Execução", "Revisão Criminal"
+]
+
 # --- FUNÇÕES UTILITÁRIAS ---
 
 def limpar_json(texto_bruto):
     if not texto_bruto: return {}
-    # Remove blocos markdown ```json
     texto = re.sub(r'^```json', '', texto_bruto, flags=re.MULTILINE)
     texto = re.sub(r'^```', '', texto, flags=re.MULTILINE)
     try:
         return json.loads(texto)
     except:
-        # Tenta encontrar o primeiro objeto JSON válido se houver texto em volta
         match = re.search(r'\{.*\}', texto, re.DOTALL)
         if match: return json.loads(match.group(0))
         return {}
@@ -63,98 +67,75 @@ def contar_palavras(texto):
     if not texto: return "0 palavras"
     return f"{len(texto.split())} palavras"
 
-# --- PDF ENGINE (COM TABELAS) ---
+# --- PDF ENGINE ---
 
 class JusTutorPDF(FPDF):
     def header(self):
-        self.set_font('Helvetica', 'B', 14)
+        self.set_font('Arial', 'B', 14)
         self.set_text_color(50, 50, 50)
         self.cell(0, 10, 'JusTutor Pro - Simulado OAB 2a Fase', 0, 1, 'C')
         self.ln(5)
 
     def footer(self):
         self.set_y(-15)
-        self.set_font('Helvetica', 'I', 8)
+        self.set_font('Arial', 'I', 8)
         self.cell(0, 10, f'Pagina {self.page_no()}', 0, 0, 'C')
 
 def safe_text(text):
-    """Trata caracteres para evitar erros no FPDF"""
     if not text: return ""
     if not isinstance(text, str): text = str(text)
-    # Substituições básicas de tipografia
     replacements = {'–': '-', '—': '-', '“': '"', '”': '"', '’': "'", '‘': "'", '…': '...'}
     for k, v in replacements.items(): text = text.replace(k, v)
-    # Codificação para Latin-1 (padrão FPDF)
     return text.encode('latin-1', 'replace').decode('latin-1')
 
 def draw_table_row(pdf, item, valor, nota, motivo):
-    """Desenha uma linha da tabela estilo FGV"""
-    pdf.set_font("Helvetica", "", 10)
-    
+    pdf.set_font("Arial", "", 10)
     line_height = 5
     x_start = pdf.get_x()
     y_start = pdf.get_y()
     
-    # Calcular alturas
     pdf.multi_cell(90, line_height, safe_text(item), split_only=True)
     h_item = len(pdf.multi_cell(90, line_height, safe_text(item), split_only=True)) * line_height
-    
     pdf.multi_cell(60, line_height, safe_text(motivo), split_only=True)
     h_motivo = len(pdf.multi_cell(60, line_height, safe_text(motivo), split_only=True)) * line_height
-    
     row_height = max(h_item, h_motivo, 10)
     
-    # Quebra de página se necessário
     if y_start + row_height > 270:
         pdf.add_page()
         y_start = pdf.get_y()
     
-    # Col 1: Item
     pdf.set_xy(x_start, y_start)
     pdf.multi_cell(90, line_height, safe_text(item), border=1)
-    
-    # Col 2: Valor
     pdf.set_xy(x_start + 90, y_start)
     pdf.cell(20, row_height, safe_text(str(valor)), border=1, align='C')
-    
-    # Col 3: Nota
     pdf.set_xy(x_start + 110, y_start)
     pdf.cell(20, row_height, safe_text(str(nota)), border=1, align='C')
-    
-    # Col 4: Motivo
     pdf.set_xy(x_start + 130, y_start)
-    pdf.set_font("Helvetica", "I", 9)
+    pdf.set_font("Arial", "I", 9)
     pdf.multi_cell(60, line_height, safe_text(motivo), border=1)
-    
     pdf.set_y(y_start + row_height)
 
 def gerar_pdf_caderno(area, enc_peca, q1, q2, q3, q4):
-    """Gera o caderno de questões"""
     pdf = JusTutorPDF()
     pdf.set_auto_page_break(auto=True, margin=15)
     pdf.add_page()
-    
-    pdf.set_font("Helvetica", "B", 12)
+    pdf.set_font("Arial", "B", 12)
     pdf.cell(0, 10, safe_text(f"CADERNO DE QUESTOES - {area}"), 0, 1, 'C')
     pdf.cell(0, 10, safe_text(f"Gerado em: {datetime.now().strftime('%d/%m/%Y')}"), 0, 1, 'C')
     pdf.ln(10)
     
     def add_q(title, body):
         if not body: return
-        pdf.set_font("Helvetica", "B", 12)
+        pdf.set_font("Arial", "B", 12)
         pdf.set_fill_color(230, 230, 230)
         pdf.cell(0, 10, safe_text(title), 0, 1, 'L', fill=True)
         pdf.ln(2)
-        pdf.set_font("Helvetica", "", 11)
+        pdf.set_font("Arial", "", 11)
         pdf.multi_cell(0, 6, safe_text(body))
         pdf.ln(8)
 
     add_q("PECA PRATICO-PROFISSIONAL", enc_peca)
-    add_q("QUESTAO 1", q1)
-    add_q("QUESTAO 2", q2)
-    add_q("QUESTAO 3", q3)
-    add_q("QUESTAO 4", q4)
-    
+    add_q("QUESTAO 1", q1); add_q("QUESTAO 2", q2); add_q("QUESTAO 3", q3); add_q("QUESTAO 4", q4)
     filename = f"Caderno_Prova_{datetime.now().strftime('%H%M%S')}.pdf"
     pdf.output(filename)
     return filename
@@ -164,33 +145,28 @@ def gerar_pdf_completo(area, enc_peca, q1, q2, q3, q4, r_peca, r_q1, r_q2, r_q3,
     pdf.set_auto_page_break(auto=True, margin=15)
     pdf.add_page()
     
-    # 1. Respostas
-    pdf.set_font("Helvetica", "B", 14)
+    # Respostas
+    pdf.set_font("Arial", "B", 14)
     pdf.cell(0, 10, safe_text("FOLHA DE RESPOSTAS"), 0, 1, 'C')
     pdf.ln(5)
     
     def add_resp(title, content):
         if not content: content = "[Em branco]"
-        pdf.set_font("Helvetica", "B", 11)
+        pdf.set_font("Arial", "B", 11)
         pdf.cell(0, 8, safe_text(title), 0, 1)
         pdf.set_font("Courier", "", 10)
         pdf.multi_cell(0, 5, safe_text(content), border=1)
         pdf.ln(5)
         
-    add_resp("Sua Peca", r_peca)
-    add_resp("Questao 1", r_q1)
-    add_resp("Questao 2", r_q2)
-    add_resp("Questao 3", r_q3)
-    add_resp("Questao 4", r_q4)
+    add_resp("Sua Peca", r_peca); add_resp("Q1", r_q1); add_resp("Q2", r_q2); add_resp("Q3", r_q3); add_resp("Q4", r_q4)
     
-    # 2. Espelho (Tabela)
+    # Tabela
     pdf.add_page()
-    pdf.set_font("Helvetica", "B", 14)
-    pdf.cell(0, 10, safe_text("ESPELHO DE CORRECAO (PADRAO FGV)"), 0, 1, 'C')
+    pdf.set_font("Arial", "B", 14)
+    pdf.cell(0, 10, safe_text("ESPELHO DE CORRECAO"), 0, 1, 'C')
     pdf.ln(5)
-    
     pdf.set_fill_color(200, 200, 200)
-    pdf.set_font("Helvetica", "B", 10)
+    pdf.set_font("Arial", "B", 10)
     pdf.cell(90, 8, "ITEM AVALIADO", 1, 0, 'C', True)
     pdf.cell(20, 8, "VALOR", 1, 0, 'C', True)
     pdf.cell(20, 8, "NOTA", 1, 0, 'C', True)
@@ -198,35 +174,25 @@ def gerar_pdf_completo(area, enc_peca, q1, q2, q3, q4, r_peca, r_q1, r_q2, r_q3,
     
     try:
         if isinstance(dados_correcao, str):
-             pdf.set_font("Helvetica", "", 10)
-             pdf.multi_cell(0, 5, safe_text(dados_correcao))
+             pdf.set_font("Arial", "", 10); pdf.multi_cell(0, 5, safe_text(dados_correcao))
         else:
             tabela = dados_correcao.get("tabela_notas", [])
             nota_final = dados_correcao.get("nota_final", "0.0")
-            
             for linha in tabela:
                 draw_table_row(pdf, linha.get("item", ""), linha.get("valor", ""), linha.get("nota_aluno", ""), linha.get("motivo", ""))
-            
             pdf.ln(5)
-            pdf.set_font("Helvetica", "B", 12)
+            pdf.set_font("Arial", "B", 12)
             pdf.cell(0, 10, safe_text(f"NOTA FINAL: {nota_final} / 10.0"), 0, 1, 'R')
-
     except Exception as e:
-        pdf.multi_cell(0, 5, safe_text(f"Erro ao gerar tabela: {str(e)}"))
+        pdf.multi_cell(0, 5, safe_text(f"Erro tabela: {str(e)}"))
 
-    # 3. Gabarito
     if modelo:
-        pdf.add_page()
-        pdf.set_font("Helvetica", "B", 14); pdf.cell(0, 10, safe_text("GABARITO MODELO"), 0, 1, 'C')
-        pdf.set_font("Helvetica", "", 10)
-        pdf.multi_cell(0, 5, safe_text(str(modelo).replace("**", "")))
+        pdf.add_page(); pdf.set_font("Arial", "B", 14); pdf.cell(0, 10, safe_text("GABARITO MODELO"), 0, 1, 'C')
+        pdf.set_font("Arial", "", 10); pdf.multi_cell(0, 5, safe_text(str(modelo).replace("**", "")))
 
-    # 4. Mentor
     if mentor:
-        pdf.add_page()
-        pdf.set_font("Helvetica", "B", 14); pdf.cell(0, 10, safe_text("MENTOR VADE MECUM"), 0, 1, 'C')
-        pdf.set_font("Helvetica", "", 10)
-        pdf.multi_cell(0, 5, safe_text(str(mentor).replace("**", "")))
+        pdf.add_page(); pdf.set_font("Arial", "B", 14); pdf.cell(0, 10, safe_text("MENTOR VADE MECUM"), 0, 1, 'C')
+        pdf.set_font("Arial", "", 10); pdf.multi_cell(0, 5, safe_text(str(mentor).replace("**", "")))
 
     filename = f"Relatorio_Final_{datetime.now().strftime('%H%M%S')}.pdf"
     pdf.output(filename)
@@ -236,6 +202,16 @@ def gerar_pdf_completo(area, enc_peca, q1, q2, q3, q4, r_peca, r_q1, r_q2, r_q3,
 
 def get_system_instruction(area, modo="geracao"):
     base = f"Você é a BANCA EXAMINADORA FGV (OAB 2ª FASE - {area}). "
+    
+    # REGRAS ESPECÍFICAS DE PENAL
+    if area == "Direito Penal":
+        base += """
+        CRITÉRIOS DE PENAL:
+        1. Identificação da Peça: Focar no MOMENTO PROCESSUAL (Inquérito, Denúncia, Sentença, Trânsito em Julgado, Execução).
+        2. Estrutura: Endereçamento, Preliminares (Nulidade, Extinção da Punibilidade), Mérito (Absolvição, Desclassificação), Subsidiárias (Dosimetria, Regime, Substituição de Pena).
+        3. Datar no último dia do prazo (Súmula 710 STF para Processual Penal).
+        """
+
     if modo == "geracao":
         return base + """
         Crie PROVAS DIFÍCEIS e INÉDITAS.
@@ -244,29 +220,23 @@ def get_system_instruction(area, modo="geracao"):
         """
     else:
         return base + """
-        CORRIJA COM RIGOR ABSOLUTO.
-        IMPORTANTE: O Output "tabela_notas" deve ser uma LISTA DE OBJETOS JSON.
-        Estrutura JSON Obrigatória:
+        CORRIJA COM RIGOR ABSOLUTO. Output JSON OBRIGATÓRIO:
         {
-            "tabela_notas": [
-                {"item": "Endereçamento", "valor": "0.10", "nota_aluno": "0.10", "motivo": "Correto"},
-                {"item": "Q1 - Letra A", "valor": "0.65", "nota_aluno": "0.00", "motivo": "Não citou Artigo"}
-            ],
-            "nota_final": "Total",
-            "gabarito_modelo": "Texto...",
-            "mentor_vade_mecum": "Texto..."
+            "tabela_notas": [{"item": "...", "valor": "...", "nota_aluno": "...", "motivo": "..."}],
+            "nota_final": "...", "gabarito_modelo": "...", "mentor_vade_mecum": "..."
         }
         """
 
 def get_prompt_gerador(area, peca_alvo):
     instrucao = f"Peça obrigatória: {peca_alvo}" if "Aleatória" not in peca_alvo else "Escolha peça aleatória."
-    return f"Gere simulado de {area}. {instrucao}. Inclua datas, valores e pegadinhas."
+    extra_penal = "Para Penal: Defina crime, pena em abstrato e o exato momento processual (ex: intimado da sentença ontem)." if area == "Direito Penal" else ""
+    return f"Gere simulado de {area}. {instrucao}. Inclua datas, valores/penas e pegadinhas. {extra_penal}"
 
 def get_prompt_correcao(area, gabarito, respostas):
     return f"""
-    Corrija a prova.
+    Corrija a prova comparando com o Gabarito.
     GABARITO: {json.dumps(gabarito, ensure_ascii=False)}
-    RESPOSTAS: {json.dumps(respostas, ensure_ascii=False)}
+    RESPOSTAS ALUNO: {json.dumps(respostas, ensure_ascii=False)}
     """
 
 # --- BACKEND ---
@@ -279,22 +249,18 @@ def gerar_simulado(area, peca_treino):
     try:
         response = model.generate_content(get_prompt_gerador(area, peca_treino))
         dados = limpar_json(response.text)
-        
         pdf_caderno = gerar_pdf_caderno(area, dados.get("peca_enunciado"), dados.get("q1_enunciado"), dados.get("q2_enunciado"), dados.get("q3_enunciado"), dados.get("q4_enunciado"))
         
-        # AQUI FOI A CORREÇÃO: RETORNAR EXATAMENTE 16 ITENS PARA O GRADIO
         return (
-            dados.get("gabarito_oculto"), # 1
-            dados.get("peca_enunciado"), dados.get("peca_enunciado"), # 2, 3
-            dados.get("q1_enunciado"), dados.get("q1_enunciado"), # 4, 5 (Duplicado para state/markdown)
-            dados.get("q2_enunciado"), dados.get("q2_enunciado"), # 6, 7
-            dados.get("q3_enunciado"), dados.get("q3_enunciado"), # 8, 9
-            dados.get("q4_enunciado"), dados.get("q4_enunciado"), # 10, 11
-            gr.update(visible=True), # 12
-            gr.update(interactive=True), # 13
-            pdf_caderno, # 14
-            gr.update(visible=True), # 15
-            "✅ Prova Gerada! Baixe o caderno." # 16
+            dados.get("gabarito_oculto"), 
+            dados.get("peca_enunciado"), dados.get("peca_enunciado"),  
+            dados.get("q1_enunciado"), dados.get("q1_enunciado"),
+            dados.get("q2_enunciado"), dados.get("q2_enunciado"),
+            dados.get("q3_enunciado"), dados.get("q3_enunciado"),
+            dados.get("q4_enunciado"), dados.get("q4_enunciado"),
+            gr.update(visible=True), gr.update(interactive=True),
+            pdf_caderno, gr.update(visible=True),
+            "✅ Prova Gerada! Baixe o caderno."
         )
     except Exception as e:
         logger.error(f"Erro: {e}")
@@ -302,34 +268,29 @@ def gerar_simulado(area, peca_treino):
 
 def corrigir_simulado(gabarito, area, r_peca, r_q1, r_q2, r_q3, r_q4, enc_peca, enc_q1, enc_q2, enc_q3, enc_q4):
     if not gabarito: return [None]*4 + ["⚠️ Gere a prova antes."]
-    
     respostas = {"peca": r_peca, "q1": r_q1, "q2": r_q2, "q3": r_q3, "q4": r_q4}
     model = genai.GenerativeModel(model_name=MODEL_NAME, system_instruction=get_system_instruction(area, "correcao"), generation_config=GENERATION_CONFIG)
     
     try:
         response = model.generate_content(get_prompt_correcao(area, gabarito, respostas))
         dados = limpar_json(response.text)
-        
         tabela_md = "| Item | Valor | Nota | Motivo |\n|---|---|---|---|\n"
         for l in dados.get("tabela_notas", []):
             tabela_md += f"| {l.get('item','')} | {l.get('valor','')} | {l.get('nota_aluno','')} | {l.get('motivo','')} |\n"
         
         pdf_final = gerar_pdf_completo(area, enc_peca, enc_q1, enc_q2, enc_q3, enc_q4, r_peca, r_q1, r_q2, r_q3, r_q4, dados, dados.get("gabarito_modelo"), dados.get("mentor_vade_mecum"))
-
-        return (
-            tabela_md,
-            dados.get("gabarito_modelo", "Erro"),
-            dados.get("mentor_vade_mecum", "Erro"),
-            pdf_final,
-            "✅ Correção Realizada!"
-        )
+        return (tabela_md, dados.get("gabarito_modelo", "Erro"), dados.get("mentor_vade_mecum", "Erro"), pdf_final, "✅ Correção Realizada!")
     except Exception as e:
         return [f"Erro: {e}"]*3 + [None, "❌ Falha"]
 
 # --- ORÁCULO ---
 def analisar_pdf_historico(file_obj, area):
     if not file_obj: return None, "⚠️ Envie o PDF."
-    keywords = {"Direito do Trabalho": {"Reclamação": ["reclamação"], "Contestação": ["contestação"], "RO": ["recurso ordinário"]}, "Direito Constitucional": {"ADI": ["ação direta"], "MS": ["mandado de segurança"], "Ação Popular": ["ação popular"]}}
+    keywords = {
+        "Direito do Trabalho": {"Reclamação": ["reclamação"], "Contestação": ["contestação"], "RO": ["recurso ordinário"]},
+        "Direito Constitucional": {"ADI": ["ação direta"], "MS": ["mandado de segurança"], "Ação Popular": ["ação popular"]},
+        "Direito Penal": {"Apelação": ["apelação", "sentença condenatória"], "RESE": ["recurso em sentido estrito", "pronúncia"], "Memoriais": ["memoriais", "audiência de instrução"], "Agravo": ["agravo em execução"]}
+    }
     termos = keywords.get(area, {})
     contagem = {k: 0 for k in termos.keys()}
     try:
@@ -348,28 +309,23 @@ def analisar_pdf_historico(file_obj, area):
 
 def interface():
     with gr.Blocks(theme=gr.themes.Soft(), title="JusTutor Ultimate") as demo:
-        
         gabarito_state = gr.State()
-        
-        gr.Markdown("# ⚖️ JusTutor Pro: Preparação OAB 2ª Fase")
+        gr.Markdown("# ⚖️ JusTutor Pro: Trabalho, Constitucional e Penal")
         
         with gr.Tabs():
             with gr.TabItem("📝 Sala de Prova"):
                 with gr.Row(variant="panel"):
-                    area_sel = gr.Dropdown(["Direito do Trabalho", "Direito Constitucional"], value="Direito do Trabalho", label="Área")
-                    peca_sel = gr.Dropdown(PECAS_TRABALHO, value="Aleatória (Simulado Real)", label="Modo")
+                    area_sel = gr.Dropdown(["Direito do Trabalho", "Direito Constitucional", "Direito Penal"], value="Direito Penal", label="Área")
+                    peca_sel = gr.Dropdown(PECAS_PENAL, value="Aleatória (Simulado Real)", label="Modo")
                     btn_gerar = gr.Button("🚀 Iniciar Prova", variant="primary")
                     status = gr.Textbox(label="Status", value="Pronto.", interactive=False)
                 
                 with gr.Group(visible=False) as area_prova:
                     gr.Markdown("---")
-                    
-                    # BOTÃO DOWNLOAD CADERNO
                     with gr.Row():
                         btn_down_caderno = gr.DownloadButton("📥 Baixar Caderno de Questões (PDF)", visible=False)
                     
                     with gr.Row():
-                        # Caderno
                         with gr.Column(scale=1): 
                             gr.Markdown("### 📘 Caderno de Questões")
                             with gr.Accordion("PEÇA (5.0)", open=True):
@@ -384,7 +340,6 @@ def interface():
                             with gr.Accordion("Q4 (1.25)", open=False): 
                                 enc_q4 = gr.Markdown(); enc_q4_st = gr.State()
                         
-                        # Respostas
                         with gr.Column(scale=1):
                             gr.Markdown("### ✏️ Folha de Respostas")
                             with gr.Tabs():
@@ -407,7 +362,7 @@ def interface():
                             with gr.TabItem("📚 Gabarito Modelo"): res_modelo = gr.Markdown()
                             with gr.TabItem("🧭 Mentor Vade Mecum"): res_mentor = gr.Markdown()
                         
-                        btn_pdf_final = gr.DownloadButton("📄 Baixar Relatório Final (Correção + Tabela)", label="Baixar PDF Final")
+                        btn_pdf_final = gr.DownloadButton("📄 Baixar Relatório Final", label="Baixar PDF Final")
 
             with gr.TabItem("🔮 Oráculo"):
                 file_oracle = gr.File(label="PDF Provas Anteriores")
@@ -415,11 +370,13 @@ def interface():
                 plot_oracle = gr.Plot()
                 txt_oracle = gr.Markdown()
 
-        # --- LOGICA ---
-        def atualiza_lista(area): return gr.update(choices=PECAS_TRABALHO if area == "Direito do Trabalho" else PECAS_CONSTITUCIONAL)
+        def atualiza_lista(area):
+            if area == "Direito do Trabalho": return gr.update(choices=PECAS_TRABALHO)
+            elif area == "Direito Constitucional": return gr.update(choices=PECAS_CONSTITUCIONAL)
+            else: return gr.update(choices=PECAS_PENAL)
+        
         area_sel.change(atualiza_lista, area_sel, peca_sel)
 
-        # Gerar Prova
         btn_gerar.click(
             fn=lambda: "⏳ Gerando...", outputs=status
         ).then(
@@ -427,18 +384,11 @@ def interface():
             inputs=[area_sel, peca_sel],
             outputs=[
                 gabarito_state, 
-                enc_peca_st, enc_peca,
-                enc_q1_st, enc_q1, 
-                enc_q2_st, enc_q2,
-                enc_q3_st, enc_q3,
-                enc_q4_st, enc_q4,
-                area_prova, btn_corrigir, 
-                btn_down_caderno, btn_down_caderno, # Arquivo + Visibilidade
-                status
+                enc_peca_st, enc_peca, enc_q1_st, enc_q1, enc_q2_st, enc_q2, enc_q3_st, enc_q3, enc_q4_st, enc_q4,
+                area_prova, btn_corrigir, btn_down_caderno, btn_down_caderno, status
             ]
         )
 
-        # Corrigir
         btn_corrigir.click(
             fn=lambda: "⏳ Corrigindo...", outputs=status
         ).then(