Update app.py

app.py

@@ -45,23 +45,41 @@ def gerar_peca_juridica_criminal(informacoes_caso, tipo_peca):
 
     return resposta.choices[0].message["content"]
 
-# Configuração do app Gradio com exemplos
+# Função para redefinir os campos
+def redefinir_campos():
+    return "", "", ""
+
+# Função para exportar a peça jurídica gerada
+def exportar_peca(peca_texto):
+    """
+    Exporta a peça jurídica gerada para um arquivo .txt.
+
+    Parâmetros:
+        peca_texto (str): Texto da peça jurídica gerada.
+    
+    Retorna:
+        str: Caminho do arquivo exportado.
+    """
+    arquivo_nome = "peca_juridica_gerada.txt"
+    with open(arquivo_nome, "w", encoding="utf-8") as arquivo:
+        arquivo.write(peca_texto)
+    return f"Arquivo salvo como: {arquivo_nome}"
+
+# Configuração do app Gradio com funcionalidades adicionais
 def gerar_interface_gradio():
     """
-    Configura a interface do Gradio para gerar peças jurídicas com exemplos pré-definidos.
+    Configura a interface do Gradio para gerar peças jurídicas com exemplos pré-definidos e funcionalidades adicionais.
     """
     with gr.Blocks() as interface:
         gr.Markdown("# Gerador de Peças Jurídicas Criminais")
         gr.Markdown(
-            "### Insira as informações detalhadas do caso ou escolha um exemplo, e selecione o tipo de peça jurídica que deseja gerar."
+            "### Preencha os campos abaixo 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 ou selecione um exemplo abaixo."
-                ),
+                placeholder="Descreva o caso detalhadamente ou selecione um exemplo abaixo.",
                 lines=5
             )
             
@@ -76,9 +94,10 @@ def gerar_interface_gradio():
                 interactive=True
             )
             
-            entrada_tipo = gr.Textbox(
+            entrada_tipo = gr.Dropdown(
                 label="Tipo de Peça Jurídica",
-                placeholder="Exemplo: habeas corpus, petição inicial, defesa prévia..."
+                choices=["habeas corpus", "petição inicial", "defesa prévia", "apelação criminal"],
+                placeholder="Selecione o tipo de peça jurídica"
             )
         
         def atualizar_caso(caso_exemplo):
@@ -93,7 +112,10 @@ def gerar_interface_gradio():
             outputs=entrada_caso
         )
         
-        botao_gerar = gr.Button("Gerar Peça Jurídica")
+        with gr.Row():
+            botao_gerar = gr.Button("Gerar Peça Jurídica")
+            botao_redefinir = gr.Button("Redefinir Campos")
+        
         saida_texto = gr.Textbox(
             label="Peça Jurídica Gerada",
             lines=15,
@@ -105,6 +127,26 @@ def gerar_interface_gradio():
             inputs=[entrada_caso, entrada_tipo], 
             outputs=saida_texto
         )
+
+        botao_redefinir.click(
+            redefinir_campos, 
+            inputs=None, 
+            outputs=[entrada_caso, entrada_tipo, saida_texto]
+        )
+        
+        with gr.Row():
+            botao_exportar = gr.Button("Exportar Peça Jurídica para .txt")
+            saida_arquivo = gr.Textbox(
+                label="Arquivo Exportado",
+                interactive=False,
+                placeholder="O caminho do arquivo exportado será exibido aqui."
+            )
+        
+        botao_exportar.click(
+            exportar_peca,
+            inputs=saida_texto,
+            outputs=saida_arquivo
+        )
     
     return interface