Update app.py

app.py

@@ -10,10 +10,10 @@ if not api_key:
 # Criação do cliente com a API Key
 client = InferenceClient(api_key=api_key)
 
-# Função para gerar a peça jurídica em português
+# Função para gerar a peça jurídica em português do Brasil
 def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
     """
-    Gera uma peça jurídica criminal em português com base no tipo e nas informações fornecidas.
+    Gera uma peça jurídica criminal em português do Brasil com base no tipo e nas informações fornecidas.
 
     Parâmetros:
         informacoes_caso (str): Descrição detalhada do caso.
@@ -27,7 +27,7 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
             "role": "system",
             "content": (
                 "Você é um especialista em direito criminal no Brasil e deve gerar peças jurídicas detalhadas. "
-                "Todas as respostas devem estar em português, considerando o ordenamento jurídico brasileiro."
+                "Todas as respostas devem estar em português do Brasil, considerando o ordenamento jurídico brasileiro."
             )
         },
         {
@@ -45,31 +45,54 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
 
     return resposta.choices[0].message["content"]
 
-# Configuração do app Gradio
+# Configuração do app Gradio com exemplos
 def gerar_interface_gradio():
     """
-    Configura a interface do Gradio para gerar peças jurídicas.
+    Configura a interface do Gradio para gerar peças jurídicas com exemplos pré-definidos.
     """
     with gr.Blocks() as interface:
         gr.Markdown("# Gerador de Peças Jurídicas Criminais")
         gr.Markdown(
-            "### Insira as informações detalhadas do caso e escolha o tipo de peça jurídica que deseja gerar."
+            "### Insira as informações detalhadas do caso ou escolha um exemplo, e selecione o tipo de peça jurídica que deseja gerar."
         )
         
         with gr.Row():
             entrada_caso = gr.Textbox(
                 label="Informações do Caso",
                 placeholder=(
-                    "Descreva o caso detalhadamente. Exemplo: O cliente foi acusado injustamente de furto qualificado. "
-                    "Os fatos ocorreram no dia 10 de dezembro de 2024, mas há provas de que ele estava em outro local."
+                    "Descreva o caso detalhadamente ou selecione um exemplo abaixo."
                 ),
                 lines=5
             )
+            
+            exemplo_caso = gr.Dropdown(
+                label="Exemplos de Casos",
+                choices=[
+                    "O cliente foi acusado injustamente de furto qualificado, mas há provas de que estava em outro local no momento do crime.",
+                    "O cliente foi preso em flagrante por porte de entorpecentes, mas os policiais não apresentaram o mandado de busca.",
+                    "O cliente foi acusado de agressão, mas existem testemunhas que comprovam sua ausência no local do ocorrido."
+                ],
+                value="",
+                interactive=True
+            )
+            
             entrada_tipo = gr.Textbox(
                 label="Tipo de Peça Jurídica",
-                placeholder="Exemplo: habeas corpus, petição inicial..."
+                placeholder="Exemplo: habeas corpus, petição inicial, defesa prévia..."
             )
         
+        def atualizar_caso(caso_exemplo):
+            """
+            Atualiza o campo de texto com o exemplo escolhido pelo usuário.
+            """
+            return caso_exemplo
+
+        exemplo_caso.change(
+            atualizar_caso, 
+            inputs=exemplo_caso, 
+            outputs=entrada_caso
+        )
+        
         botao_gerar = gr.Button("Gerar Peça Jurídica")
         saida_texto = gr.Textbox(
             label="Peça Jurídica Gerada",