Update app.py

app.py

@@ -11,6 +11,44 @@ class DefenseGenerator:
         self.client = InferenceClient(api_key=self.api_key)
         
         self.templates = {
+            "habeas_corpus": {
+                "system": "Você é um advogado criminalista especializado em Habeas Corpus.",
+                "sections": {
+                    "cabecalho": """EXCELENTÍSSIMO(A) SENHOR(A) DOUTOR(A) DESEMBARGADOR(A) DO EGRÉGIO TRIBUNAL DE JUSTIÇA DO ESTADO DE {estado}""",
+                    "qualificacao": """{nome_reu}, {qualificacao_reu}, vem, por seu advogado, impetrar HABEAS CORPUS, com fundamento no art. 5º, LXVIII, da Constituição Federal e arts. 647 e seguintes do Código de Processo Penal, em face de ato do MM. Juiz de Direito da {vara} Vara Criminal da Comarca de {comarca}, pelos fatos e fundamentos a seguir expostos:""",
+                    "fundamentos": "{fundamentos}",
+                    "pedidos": """Ante o exposto, requer:
+
+{pedidos}
+
+Termos em que,
+Pede deferimento.
+
+{comarca}, {data}
+
+Advogado(a)
+OAB/{estado} {numero_oab}"""
+                }
+            },
+            "revogacao_prisao_preventiva": {
+                "system": "Você é um advogado criminalista especializado em pedidos de revogação de prisão preventiva.",
+                "sections": {
+                    "cabecalho": """EXCELENTÍSSIMO(A) SENHOR(A) DOUTOR(A) JUIZ(A) DE DIREITO DA {vara} VARA CRIMINAL DA COMARCA DE {comarca}
+
+Processo nº: {numero_processo}""",
+                    "qualificacao": """{nome_reu}, já qualificado nos autos do processo em epígrafe, por seu advogado que esta subscreve, vem, respeitosamente, à presença de Vossa Excelência, com fundamento no art. 316 do Código de Processo Penal, requerer a REVOGAÇÃO DA PRISÃO PREVENTIVA, pelos motivos de fato e de direito a seguir expostos:""",
+                    "fundamentos": "{fundamentos}",
+                    "pedidos": """Ante o exposto, requer-se a revogação da prisão preventiva, com a expedição do competente alvará de soltura.
+
+Termos em que,
+Pede deferimento.
+
+{comarca}, {data}
+
+Advogado(a)
+OAB/{estado} {numero_oab}"""
+                }
+            },
             "resposta_acusacao": {
                 "system": "Você é um advogado criminalista experiente, especializado em respostas à acusação.",
                 "sections": {
@@ -100,6 +138,18 @@ OAB/{estado} {numero_oab}"""
 def interface():
     gerador = DefenseGenerator()
     
+    tipo_peca_descricoes = {
+        "resposta_acusacao": "Apresentar defesa inicial após o recebimento da denúncia",
+        "alegacoes_defesa": "Apresentar argumentos finais após a instrução",
+        "habeas_corpus": "Impetrar HC contra prisão ou ameaça à liberdade",
+        "recurso_em_sentido_estrito": "Recorrer de decisões interlocutórias",
+        "apelacao": "Recorrer da sentença",
+        "embargos_declaracao": "Esclarecer pontos omissos/contraditórios",
+        "revogacao_prisao_preventiva": "Requerer revogação de prisão preventiva",
+        "relaxamento_prisao": "Requerer relaxamento de prisão ilegal",
+        "liberdade_provisoria": "Requerer liberdade provisória"
+    }
+    
     def processar(tipo_doc, comarca, vara, estado, numero_processo, nome_reu, 
                  preliminares, merito, provas, pedidos, relatorio, numero_oab):
         dados = {
@@ -121,6 +171,23 @@ def interface():
     with gr.Blocks(title="Gerador de Peças da Defesa Criminal") as demo:
         gr.Markdown("# Gerador de Peças da Defesa Criminal")
         
+        with gr.Row():
+            gr.Markdown("""### Selecione o tipo de peça que deseja gerar:
+            
+Cada opção é adequada para diferentes momentos processuais e estratégias de defesa.""")
+            
+        with gr.Row():
+            tipo_peca = gr.Dropdown(
+                choices=list(tipo_peca_descricoes.keys()),
+                label="Tipo de Peça",
+                value="resposta_acusacao",
+                info="Selecione o tipo específico de peça criminal"
+            )
+            
+        with gr.Row():
+            gr.Markdown(lambda x: f"**Descrição:** {tipo_peca_descricoes.get(x, '')}", every=100)
+            tipo_peca.change(lambda x: f"**Descrição:** {tipo_peca_descricoes.get(x, '')}", tipo_peca)
+        
         with gr.Row():
             tipo_documento = gr.Radio(
                 choices=["resposta_acusacao", "alegacoes_defesa"],