Update app.py

app.py

@@ -11,15 +11,11 @@ if not api_key:
 client = InferenceClient(api_key=api_key)
 
 # Função para gerar a peça jurídica em português do Brasil
+# def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
 def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
-    if not tipo_peca:
-        return "Por favor, selecione o tipo de peça jurídica."
-        
-    # Validação de entrada
-    if not informacoes_caso:
-        return "Por favor, forneça as informações do caso."
+    if not tipo_peca or not informacoes_caso:
+        return "Por favor, forneça todas as informações necessárias."
 
-    # Template específico para cada tipo de peça
     templates = {
         "habeas corpus": {
             "estrutura": [
@@ -34,52 +30,40 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
             ],
             "prompt": "Gere um Habeas Corpus completo e tecnicamente preciso considerando:\n"
                      "- Qualificação completa do paciente\n"
-                     "- Narrativa clara e objetiva dos fatos\n"
-                     "- Base legal específica (Art. 647 e 648 do CPP)\n"
-                     "- Jurisprudência atualizada dos tribunais superiores\n"
+                     "- Narrativa clara dos fatos\n"
+                     "- Base legal (Art. 647 e 648 do CPP)\n"
+                     "- Jurisprudência atual\n"
                      "- Doutrina pertinente\n"
-                     "- Pedido liminar fundamentado\n"
-                     "- Pedido principal detalhado\n\n"
-                     "1. Fundamentos constitucionais (Art. 5º, LXVIII, CF)\n"
-                     "2. Requisitos específicos do HC\n"
-                     "3. Jurisprudência relevante do STF/STJ\n"
-                     "4. Pedidos liminares e principais\n"
+                     "- Pedidos liminar e principal\n"
         },
         "petição inicial": {
             "estrutura": [
-                "QUALIFICAÇÃO DAS PARTES",
+                "QUALIFICAÇÃO",
                 "DOS FATOS",
                 "DO DIREITO",
                 "DOS PEDIDOS"
             ],
-            "prompt": "Elabore uma petição inicial criminal contemplando:\n"
-                     "1. Elementos essenciais (Art. 41, CPP)\n"
-                     "2. Fundamentação legal específica\n"
-                     "3. Provas e documentos relevantes\n"
-                     "4. Pedidos principais e subsidiários\n"
+            "prompt": "Elabore uma petição inicial criminal contemplando os requisitos do Art. 41 do CPP"
         }
     }
 
     template = templates.get(tipo_peca.lower(), {
         "estrutura": ["INTRODUÇÃO", "DESENVOLVIMENTO", "CONCLUSÃO"],
-        "prompt": f"Elabore uma {tipo_peca} criminal completa e fundamentada."
+        "prompt": f"Elabore uma {tipo_peca} criminal fundamentada."
     })
 
-    # Primeira chamada: Estruturação básica
     mensagens_estrutura = [
         {
             "role": "system",
             "content": (
-                "Você é um advogado criminal especialista brasileiro. "
-                "Formate a peça seguindo rigorosamente a estrutura jurídica brasileira. "
-                "Use linguagem técnica apropriada e citações legais precisas."
+                "Você é um advogado criminalista especialista. "
+                "Formate a peça conforme a estrutura jurídica brasileira."
             )
         },
         {
             "role": "user",
             "content": f"{template['prompt']}\n\nCaso: {informacoes_caso}\n\n"
-                      f"Estruture a peça com as seguintes seções:\n"
-                      f"{chr(10).join(template['estrutura'])}"
+                      f"Estruture com:\n{chr(10).join(template['estrutura'])}"
         }
     ]
 
@@ -92,56 +76,48 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
 
     estrutura_base = resposta_estrutura.choices[0].message["content"]
 
-    # Segunda chamada: Contextualização e fundamentação
     mensagens_elaboracao = [
         {
             "role": "system",
             "content": (
-                "Você é um experiente advogado criminalista brasileiro com profundo conhecimento em habeas corpus. "
-                "Desenvolva cada seção da peça com máxima precisão técnica e fundamentação sólida. "
-                "Para cada argumento jurídico, inclua:\n"
-                "1. Base legal específica e atualizada\n"
-                "2. Jurisprudência recente dos tribunais superiores\n"
-                "3. Doutrina pertinente de autores reconhecidos\n"
-                "4. Conexão clara entre fatos, direito e pedidos"
+                "Como advogado criminalista experiente, desenvolva cada seção com:\n"
+                "1. Base legal específica\n"
+                "2. Jurisprudência atual\n"
+                "3. Doutrina relevante\n"
+                "4. Conexão entre fatos e direito"
             )
         },
         {
             "role": "user",
-            "content": f"Aprimore esta peça com fundamentação legal:\n{estrutura_base}"
+            "content": f"Aprimore esta peça:\n{estrutura_base}"
         }
     ]
 
-    resposta_legal = client.chat.completions.create(
+    resposta_elaboracao = client.chat.completions.create(
         model="mistralai/Mistral-7B-Instruct-v0.3",
-        messages=mensagens_legal,
+        messages=mensagens_elaboracao,
         max_tokens=1500,
         temperature=0.7
     )
 
-    # Terceira chamada: Finalização e formatação técnica
-    mensagens_revisao = [
+    mensagens_final = [
         {
             "role": "system",
-            "content": (
-                "Você é um revisor jurídico expert. "
-                "Faça a revisão final garantindo clareza, coerência "
-                "e formatação adequada da peça."
-            )
+            "content": "Faça a revisão técnica final garantindo precisão e formatação."
         },
         {
             "role": "user",
             "content": (
-                f"Revise e formate esta peça jurídica:\n"
+                f"Revise e formate:\n"
                 f"{estrutura_base}\n\n"
-                f"Complementos legais:\n{resposta_legal.choices[0].message['content']}"
+                f"Complementos:\n{resposta_elaboracao.choices[0].message['content']}"
             )
         }
     ]
 
     resposta_final = client.chat.completions.create(
         model="mistralai/Mistral-7B-Instruct-v0.3",
-        messages=mensagens_revisao,
+        messages=mensagens_final,
         max_tokens=1500,
         temperature=0.5
     )
@@ -149,7 +125,6 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
     return resposta_final.choices[0].message["content"]
 
 
-
 # Função para redefinir os campos
 def redefinir_campos():
     return "", "", ""