Update app.py

app.py

@@ -11,7 +11,6 @@ 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 or not informacoes_caso:
         return "Por favor, forneça todas as informações necessárias."
@@ -28,13 +27,17 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
                 "DOS PEDIDOS",
                 "FECHAMENTO"
             ],
-            "prompt": "Gere um Habeas Corpus completo e tecnicamente preciso considerando:\n"
-                     "- Qualificação completa do paciente\n"
-                     "- Narrativa clara dos fatos\n"
-                     "- Base legal (Art. 647 e 648 do CPP)\n"
-                     "- Jurisprudência atual\n"
-                     "- Doutrina pertinente\n"
-                     "- Pedidos liminar e principal\n"
+            "prompt": (
+                "Gere um Habeas Corpus tecnicamente preciso incluindo:\n"
+                "- Qualificação do paciente\n"
+                "- Narrativa dos fatos\n"
+                "- CPP Arts. 647/648 e jurisprudência:\n"
+                "  * STF HC 148.459/DF sobre excesso de prazo\n"
+                "  * STJ HC 509.030/SP sobre ausência de fundamentação\n"
+                "  * STF HC 137.728/RJ sobre presunção de inocência\n"
+                "- Súmulas 691 e 692 do STF\n"
+                "- Pedidos liminar e principal detalhados"
+            )
         },
         "petição inicial": {
             "estrutura": [
@@ -80,11 +83,12 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
         {
             "role": "system",
             "content": (
-                "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"
+                "Atue como advogado criminalista especialista em habeas corpus. "
+                "Para cada argumento, cite:\n"
+                "1. Artigos específicos do CPP e CF\n"
+                "2. Precedentes recentes do STF/STJ (últimos 5 anos)\n"
+                "3. Súmulas vinculantes aplicáveis\n"
+                "4. Correlação entre precedentes e o caso concreto"
             )
         },
         {