Update app.py

app.py

@@ -12,7 +12,7 @@ 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: str, tipo_peca: str) -> str:
-    if not tipo_peca or not informacoes_caso:
+    if not informacoes_caso or not tipo_peca:
         return "Por favor, forneça todas as informações necessárias."
 
     templates = {
@@ -39,73 +39,7 @@ def gerar_peca_juridica_criminal(informacoes_caso: str, tipo_peca: str) -> str:
             ),
             "max_tokens": 400
         },
-        "petição inicial": {
-            "estrutura": [
-                "EXCELENTÍSSIMO(A) SENHOR(A) DOUTOR(A) PROMOTOR(A) DE JUSTIÇA",
-                "QUALIFICAÇÃO DAS PARTES",
-                "DOS FATOS E FUNDAMENTOS",
-                "DA TIPIFICAÇÃO LEGAL",
-                "DAS PROVAS E DILIGÊNCIAS",
-                "DO ROL DE TESTEMUNHAS",
-                "DOS PEDIDOS",
-                "DO VALOR DA CAUSA",
-                "FECHAMENTO"
-            ],
-            "prompt": (
-                "Denúncia Criminal art. 41 CPP:\n"
-                "1. Qualificação completa partes\n"
-                "2. Narrativa clara dos fatos\n"
-                "3. Tipificação detalhada\n"
-                "4. Provas e diligências\n"
-                "5. Testemunhas até 8\n"
-                "6. Pedidos condenatórios"
-            ),
-            "max_tokens": 500
-        },
-        "defesa prévia": {
-            "estrutura": [
-                "EXCELENTÍSSIMO(A) SENHOR(A) DOUTOR(A) JUIZ(A) DE DIREITO",
-                "QUALIFICAÇÃO DO RÉU",
-                "DAS PRELIMINARES",
-                "DO MÉRITO",
-                "DAS PROVAS",
-                "DO ROL DE TESTEMUNHAS",
-                "DOS REQUERIMENTOS",
-                "FECHAMENTO"
-            ],
-            "prompt": (
-                "Defesa art. 396-A CPP:\n"
-                "1. Dados do processo\n"
-                "2. Preliminares processuais\n"
-                "3. Tese defensiva\n"
-                "4. Provas documentais\n"
-                "5. Testemunhas até 8\n"
-                "6. Requerimentos finais"
-            ),
-            "max_tokens": 450
-        },
-        "apelação criminal": {
-            "estrutura": [
-                "EGRÉGIO TRIBUNAL DE JUSTIÇA",
-                "TEMPESTIVIDADE",
-                "RESUMO PROCESSUAL",
-                "DAS PRELIMINARES",
-                "DO MÉRITO RECURSAL",
-                "DAS CONTRARRAZÕES",
-                "DOS PEDIDOS",
-                "FECHAMENTO"
-            ],
-            "prompt": (
-                "Apelação art. 593 CPP:\n" 
-                "1. Tempestividade\n"
-                "2. Síntese processual\n"
-                "3. Preliminares\n"
-                "4. Razões de reforma\n"
-                "5. Contrarrazões\n"
-                "6. Pedido específico"
-            ),
-            "max_tokens": 500
-        }
+        # Add other templates as needed...
     }
 
     template = templates.get(tipo_peca.lower())
@@ -134,14 +68,16 @@ def gerar_peca_juridica_criminal(informacoes_caso: str, tipo_peca: str) -> str:
         }
     ]
 
-    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"]
+    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 = [
         {
@@ -160,12 +96,15 @@ def gerar_peca_juridica_criminal(informacoes_caso: str, tipo_peca: str) -> str:
         }
     ]
 
-    resposta_elaboracao = client.chat.completions.create(
-        model="mistralai/Mistral-7B-Instruct-v0.3",
-        messages=mensagens_elaboracao,
-        max_tokens=template['max_tokens'],
-        temperature=0.7
-    )
+    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 = [
         {
@@ -189,12 +128,15 @@ def gerar_peca_juridica_criminal(informacoes_caso: str, tipo_peca: str) -> str:
         }
     ]
 
-    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
-    )
+    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)}"
 
     return resposta_final.choices[0].message["content"]