Update app.py

app.py

@@ -19,25 +19,29 @@ def gerar_peca_juridica_criminal(informacoes_caso: str, tipo_peca: str) -> str:
         "habeas corpus": {
             "estrutura": [
                 "EXCELENTÍSSIMO(A) SENHOR(A) DOUTOR(A) JUIZ(A) DE DIREITO DA [VARA] DA [COMARCA]",
-                "QUALIFICAÇÃO DO PACIENTE",
-                "DOS FATOS",
-                "DO DIREITO",
-                "DO CONSTRANGIMENTO ILEGAL",
-                "DA MEDIDA LIMINAR",
-                "DOS PEDIDOS",
-                "PROVAS E DOCUMENTOS",
-                "FECHAMENTO",
+                "QUALIFICAÇÃO DO PACIENTE\n"
+                "O acusado, [Nome do Cliente], é um cidadão brasileiro, nascido em [Data de Nascimento] no município de [Local de Nascimento], "
+                "titular do documento de identidade [Tipo de Documento de Identidade] no número [Número do Documento de Identidade], "
+                "efetivo domiciliado na [Endereço de Domicílio do Cliente].",
+                "DOS FATOS\n"
+                "A Polícia de [Local do Crime] detém o acusado desde [Data da Detenção], por motivos de suposto furto qualificado, "
+                "cometido no dia [Data do Crime] às [Horas do Crime], no local de [Local do Crime]. "
+                "Entretanto, há provas documentais e testemunha ocular que comprovam a presença do acusado na cidade distante [Local Distante do Crime] no momento do crime.",
+                "DO DIREITO\n"
+                "O artigo 5º LXVIII da Constituição Federal dispõe que 'toda pessoa pode ser presa quando há fundamentos razoáveis para acreditar que ela cometeu ou está prestes a cometer um crime'. "
+                "O Código de Processo Penal estabelece, nos artigos 647 e 648, que 'a prisão não pode ser imposta a menos que haja fundamentos razoáveis...'",
+                "DO CONSTRANGIMENTO ILEGAL\n"
+                "A prisão do acusado, sem a devida fundamentação, caracteriza constrangimento ilegal, uma vez que há evidências de sua inocência.",
+                "DA MEDIDA LIMINAR\n"
+                "Requer-se a concessão de medida liminar para a liberdade do paciente.",
+                "DOS PEDIDOS\n"
+                "Diante do exposto, requer-se a concessão do habeas corpus.",
+                "PROVAS E DOCUMENTOS\n"
+                "Anexamos documentos que comprovam a inocência do acusado.",
+                "FECHAMENTO\n"
+                "Nestes termos, pede deferimento."
             ],
-            "prompt": (
-                "HC técnico:\n"
-                "1. Qualificação completa\n"
-                "2. Fatos e provas do constrangimento\n"
-                "3. CF art 5º LXVIII e CPP 647/648\n"
-                "4. STF/STJ sobre o tema\n"
-                "5. Pedido liminar fundamentado\n"
-                "6. Rol de documentos"
-            ),
-            "max_tokens": 400
+            "max_tokens": 500
         },
         # Add other templates as needed...
     }
@@ -58,39 +62,50 @@ def gerar_peca_juridica_criminal(informacoes_caso: str, tipo_peca: str) -> str:
         "Data da Detenção": "15/12/2023",
         "Data do Crime": "14/12/2023",
         "Horas do Crime": "14:00",
-        "Local Distante do Crime": "Campinas"
+        "Local Distante do Crime": "Campinas",
+        "VARA": "1ª Vara Criminal",  # Exemplo
+        "COMARCA": "São Paulo"  # Exemplo
     }
 
-    # Substituir os placeholders na estrutura
-    estrutura_final = "\n".join([
+    # Montar o texto da peça jurídica
+    texto_final = "\n\n".join([
         linha.replace(placeholder, informacoes[placeholder])
         for linha in template["estrutura"]
         for placeholder in informacoes.keys()
         if placeholder in linha
     ])
 
-    # Adicionar as seções adicionais
-    direitos = (
-        "O artigo 5º LXVIII do Código Penal brasileiro dispõe que 'toda pessoa pode ser presa quando há fundamentos razoáveis para acreditar que ela cometeu ou está prestes a cometer um crime'. "
-        "Por outro lado, o Código Processual Penal estabelece, nos artigos 647 e 648, que 'a prisão não pode ser imposta a menos que haja fundamentos razoáveis...'"
-    )
+    # Realizar a requisição única para aprimoramento
+    try:
+        mensagens = [
+            {
+                "role": "system",
+                "content": (
+                    f"Como advogado criminalista, redija um {tipo_peca} técnica seguindo:\n"
+                    f"1. Estrutura formal\n"
+                    f"2. Linguagem jurídica\n"
+                    f"3. Fundamentação precisa\n"
+                    f"4. Formatação correta"
+                )
+            },
+            {
+                "role": "user",
+                "content": texto_final
+            }
+        ]
+
+        resposta_final = client.chat.completions.create(
+            model="mistralai/Mistral-7B-Instruct-v0.3",
+            messages=mensagens,
+            max_tokens=template['max_tokens'],
+            temperature=0.7
+        )
 
-    # Montar o texto da peça jurídica
-    texto_final = (
-        f"{estrutura_final}\n\n"
-        "DOS FATOS\n"
-        "A Polícia de {informacoes['Local do Crime']} detém o acusado desde {informacoes['Data da Detenção']}, por motivos de suposto furto qualificado...\n\n"
-        f"{direitos}\n\n"
-        "DO CONSTRANGIMENTO ILEGAL\n"
-        "A prisão do acusado, sem a devida fundamentação, caracteriza constrangimento ilegal, uma vez que há evidências de sua inocência...\n\n"
-        "DA MEDIDA LIMINAR\n"
-        "Requer-se a concessão de medida liminar para a liberdade do paciente...\n\n"
-        "DOS PEDIDOS\n"
-        "Diante do exposto, requer-se..."
-    )
-
-    return texto_final
-    
+        return resposta_final.choices[0].message["content"]
+
+    except Exception as e:
+        return f"Erro ao gerar a peça: {str(e)}"
+        
 # Função para redefinir os campos
 def redefinir_campos():
     return "", "", ""