Update app.py

app.py

@@ -22,7 +22,7 @@ GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 if GEMINI_API_KEY:
     genai.configure(api_key=GEMINI_API_KEY)
 
-# --- CONFIGURAÇÃO IA ---
+# --- CONFIGURAÇÃO IA (USE UM MODELO ESTÁVEL) ---
 MODEL_NAME = "gemini-3-flash-preview" 
 
 GENERATION_CONFIG = {
@@ -215,60 +215,43 @@ def get_system_instruction(area, modo="geracao"):
     if modo == "geracao":
         return base + """
         OBJETIVO: Criar uma prova de NÍVEL DIFÍCIL, com múltiplas teses e pegadinhas processuais.
-        
         ESTRUTURA JSON DE SAÍDA:
         {
-            "peca_enunciado": "Texto longo (min. 400 palavras) com datas, valores, locais e conflito detalhado...",
+            "peca_enunciado": "Texto longo (min. 400 palavras)...",
             "q1_enunciado": "...", "q2_enunciado": "...", "q3_enunciado": "...", "q4_enunciado": "...",
             "gabarito_oculto": { ... }
         }
         """
-    else: # MODO CORREÇÃO E GABARITO 
+    else: # MODO CORREÇÃO
         return base + """
-        OBJETIVO: Agir como o MELHOR PROFESSOR DE CURSINHO PREPARATÓRIO DO BRASIL.
-        
-        DIRETRIZES PARA O GABARITO MODELO (FUNDAMENTAL):
-        1. NÃO RESUMA. Escreva a peça INTEIRA, linha por linha.
-        2. Use linguagem jurídica culta e técnica (Ex: "Data venia", "Impende destacar", "Fulcro no art...").
-        3. Estruture em Tópicos: I - Preliminares; II - Prejudiciais; III - Mérito; IV - Pedidos.
-        4. Na fundamentação, use o SILOGISMO:
-           - Premissa Maior (A Lei/Súmula diz X);
-           - Premissa Menor (O caso narra Y);
-           - Conclusão (Logo, deve-se aplicar Z).
+        OBJETIVO: Agir como o MELHOR PROFESSOR DE CURSINHO.
         
-        DIRETRIZES PARA CORREÇÃO:
-        1. Seja rigoroso na pontuação (0.00 a 10.00).
-        2. Se o aluno não citou o artigo exato, desconte ponto.
+        DIRETRIZES CORREÇÃO:
+        1. Gere uma lista JSON 'tabela_notas' onde cada item tem 'item', 'valor', 'nota_aluno' (numérico) e 'motivo'.
+        2. 'nota_final' deve ser a soma exata.
         
-        ESTRUTURA JSON DE SAÍDA:
+        ESTRUTURA JSON OBRIGATÓRIA:
         {
-            "tabela_notas": [ ... ],
-            "nota_final": "...",
-            "gabarito_modelo": "TEXTO COMPLETO DA PEÇA IDEAL (Como se fosse para protocolar no tribunal) + RESPOSTAS COMPLETAS DAS QUESTÕES.",
+            "tabela_notas": [
+                {"item": "Endereçamento", "valor": "0.10", "nota_aluno": "0.10", "motivo": "Correto"},
+                {"item": "Mérito - Tese X", "valor": "0.50", "nota_aluno": "0.00", "motivo": "Não citou Art. Y"}
+            ],
+            "nota_final": "4.50",
+            "gabarito_modelo": "Texto completo...",
             "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 baseada no histórico FGV."
-    return f"""
-    Gere um SIMULADO COMPLETO de {area}.
-    {instrucao}.
-    IMPORTANTE: O caso deve ser rico em detalhes fáticos para permitir uma defesa/acusação extensa.
-    """
+    instrucao = f"Peça obrigatória: {peca_alvo}" if "Aleatória" not in peca_alvo else "Escolha peça aleatória."
+    extra_penal = "Para Penal: Defina crime, pena em abstrato e o exato momento processual." if area == "Direito Penal" else ""
+    return f"Gere simulado de {area}. {instrucao}. Inclua datas, valores e pegadinhas. {extra_penal}"
 
 def get_prompt_correcao(area, gabarito, respostas):
     return f"""
-    REALIZA A CORREÇÃO DETALHADA E A GERAÇÃO DO GABARITO PERFEITO.
-    
-    GABARITO ESPERADO (TÓPICOS): {json.dumps(gabarito, ensure_ascii=False)}
-    RESPOSTAS DO ALUNO: {json.dumps(respostas, ensure_ascii=False)}
-    
-    ATENÇÃO AO GERAR O 'GABARITO MODELO':
-    Eu quero que você ESCREVA A PEÇA INTEIRA que gabaritaria a prova.
-    Comece pelo Endereçamento ("Excelentíssimo Senhor..."), pule linhas, faça a Qualificação ("Fulano, já qualificado..."), abra tópico das Preliminares (escreva o parágrafo inteiro argumentando), tópico do Mérito (argumente fato+fundamento+pedido) e os Pedidos Finais (alínea por alínea).
-    
-    Nas questões (Q1 a Q4), escreva a resposta completa: "A resposta é [Sim/Não]. O fundamento legal é o Artigo X, que dispõe Y. No caso concreto, ocorreu Z..."
+    Corrija a prova comparando com o Gabarito.
+    GABARITO: {json.dumps(gabarito, ensure_ascii=False)}
+    RESPOSTAS ALUNO: {json.dumps(respostas, ensure_ascii=False)}
     """
 
 # --- BACKEND ---
@@ -306,7 +289,19 @@ def corrigir_simulado(gabarito, area, r_peca, r_q1, r_q2, r_q3, r_q4, enc_peca,
     try:
         response = model.generate_content(get_prompt_correcao(area, gabarito, respostas))
         dados = limpar_json(response.text)
-        tabela_md = "| Item | Valor | Nota | Motivo |\n|---|---|---|---|\n"
+        
+        # Garante o calculo da nota final se a IA falhar
+        nota_final = dados.get("nota_final", "0.0")
+        if nota_final == "0.0" or nota_final == "":
+            try:
+                soma = sum([float(x.get('nota_aluno', 0)) for x in dados.get('tabela_notas', [])])
+                nota_final = f"{soma:.2f}"
+                dados['nota_final'] = nota_final
+            except: pass
+
+        # Constrói o Markdown para a tela com a nota em destaque
+        tabela_md = f"# 🏆 NOTA FINAL: {nota_final} / 10.0\n\n"
+        tabela_md += "| Item Avaliado | Valor Máx | Sua Nota | Comentário |\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"