Update app.py

app.py

@@ -46,99 +46,50 @@ def gerar_peca_juridica_criminal(informacoes_caso: str, tipo_peca: str) -> str:
     if not template:
         return "Tipo de peça não suportado."
 
-    mensagens_estrutura = [
-        {
-            "role": "system",
-            "content": (
-                f"Como advogado criminalista, redija {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": (
-                f"Elabore seguindo estrutura:\n"
-                f"{chr(10).join(template['estrutura'])}\n\n"
-                f"Requisitos:\n{template['prompt']}\n\n"
-                f"Caso:\n{informacoes_caso}"
-            )
-        }
-    ]
-
-    try:
-        resposta_estrutura = client.chat.completions.create(
-            model="mistralai/Mistral-7B-Instruct-v0.3",
-            messages=mensagens_estrutura,
-            max_tokens=template['max_tokens'],
-            temperature=0.7
-        )
-        estrutura_base = resposta_estrutura.choices[0].message["content"]
-    except Exception as e:
-        return f"Erro ao gerar a estrutura: {str(e)}"
-
-    mensagens_elaboracao = [
-        {
-            "role": "system",
-            "content": (
-                f"Como especialista criminal, aprimore {tipo_peca} com:\n"
-                "1. Legislação pertinente\n"
-                "2. Jurisprudência atual\n"
-                "3. Doutrina relevante\n"
-                "4. Fundamentação técnica"
-            )
-        },
-        {
-            "role": "user",
-            "content": f"Revise e complemente:\n{estrutura_base}"
-        }
-    ]
-
-    try:
-        resposta_elaboracao = client.chat.completions.create(
-            model="mistralai/Mistral-7B-Instruct-v0.3",
-            messages=mensagens_elaboracao,
-            max_tokens=template['max_tokens'],
-            temperature=0.7
-        )
-    except Exception as e:
-        return f"Erro ao elaborar a peça: {str(e)}"
-
-    mensagens_final = [
-        {
-            "role": "system",
-            "content": (
-                f"Finalize {tipo_peca} garantindo:\n"
-                "1. Estrutura correta\n"
-                "2. Numeração parágrafos\n"
-                "3. Formatação adequada\n"
-                "4. Citações precisas"
-            )
-        },
-        {
-            "role": "user",
-            "content": (
-                f"Versão final seguindo:\n"
-                f"{chr(10).join(template['estrutura'])}\n\n"
-                f"Conteúdo base:\n{estrutura_base}\n\n"
-                f"Complementos:\n{resposta_elaboracao.choices[0].message['content']}"
-            )
-        }
-    ]
-
-    try:
-        resposta_final = client.chat.completions.create(
-            model="mistralai/Mistral-7B-Instruct-v0.3",
-            messages=mensagens_final,
-            max_tokens=max(template['max_tokens'] + 100, 500),
-            temperature=0.5
-        )
-    except Exception as e:
-        return f"Erro ao finalizar a peça: {str(e)}"
+    # Preencher os campos da peça jurídica com base nas informações do caso
+    informacoes = {
+        "Nome do Cliente": "João da Silva",  # Exemplo
+        "Data de Nascimento": "01/01/1990",  # Exemplo
+        "Local de Nascimento": "São Paulo",
+        "Tipo de Documento de Identidade": "RG",
+        "Número do Documento de Identidade": "123456789",
+        "Endereço de Domicílio do Cliente": "Rua Exemplo, 123, São Paulo - SP",
+        "Local do Crime": "São Paulo",
+        "Data da Detenção": "15/12/2023",
+        "Data do Crime": "14/12/2023",
+        "Horas do Crime": "14:00",
+        "Local Distante do Crime": "Campinas"
+    }
 
-    return resposta_final.choices[0].message["content"]
+    # Substituir os placeholders na estrutura
+    estrutura_final = "\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...'"
+    )
+
+    # 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
     
 # Função para redefinir os campos
 def redefinir_campos():