Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
-import os
 from datetime import datetime
 from huggingface_hub import InferenceClient
+import os
 
 class DefenseGenerator:
     def __init__(self):
@@ -80,7 +80,7 @@ Local, Data.
 [Advogado]
 OAB/[Estado] [Número]
 
-Dados básicos para formatação:
+Dados para formatação:
 Comarca: {dados_basicos['comarca']}
 Vara: {dados_basicos['vara']}
 Cliente: {dados_basicos['cliente']}
@@ -88,38 +88,64 @@ OAB: {dados_basicos['oab']}"""
 
         return self.generate_content(prompt)
 
-def interface():
-    gerador = DefenseGenerator()
-    
-    with gr.Blocks() as demo:
-        gr.Markdown("# Gerador de Peças Criminais")
+def create_interface():
+    with gr.Blocks(theme=gr.themes.Soft()) as demo:
+        gr.Markdown("# Sistema Criminal.AI", elem_classes="text-center text-2xl font-bold mb-4")
         
-        with gr.Row():
+        with gr.Group():
             tipo = gr.Dropdown(
-                choices=list(gerador.document_types.keys()),
+                choices=list(DefenseGenerator().document_types.keys()),
+                value="resposta_acusacao",
                 label="Tipo de Peça",
-                value="resposta_acusacao"
+                container=False,
+                scale=1
             )
         
         with gr.Row():
-            with gr.Column():
-                comarca = gr.Textbox(label="Comarca", placeholder="Ex: São Paulo")
-                vara = gr.Textbox(label="Vara", placeholder="Ex: 1ª")
-                
-            with gr.Column():
-                cliente = gr.Textbox(label="Nome do Cliente", placeholder="Ex: João da Silva")
-                oab = gr.Textbox(label="OAB", placeholder="Ex: SP123456")
+            with gr.Column(scale=1):
+                comarca = gr.Textbox(
+                    label="Comarca",
+                    placeholder="Ex: São Paulo",
+                    container=False
+                )
+                vara = gr.Textbox(
+                    label="Vara",
+                    placeholder="Ex: 1ª",
+                    container=False
+                )
+            
+            with gr.Column(scale=1):
+                cliente = gr.Textbox(
+                    label="Nome do Cliente",
+                    placeholder="Ex: João da Silva",
+                    container=False
+                )
+                oab = gr.Textbox(
+                    label="OAB",
+                    placeholder="Ex: SP123456",
+                    container=False
+                )
         
         fatos = gr.Textbox(
-            label="Descreva os fatos e o que deseja na peça",
+            label="Descreva os fatos",
             placeholder="Descreva a situação do cliente e o que precisa ser argumentado...",
-            lines=5
+            lines=5,
+            container=False
         )
         
-        btn = gr.Button("Gerar")
-        output = gr.Textbox(label="Peça Gerada", lines=20, show_copy_button=True)
-        
+        with gr.Row():
+            gerar_btn = gr.Button("Gerar Peça", variant="primary", scale=2)
+            limpar_btn = gr.Button("Limpar", scale=1)
+            
+        output = gr.TextArea(
+            label="Documento Gerado",
+            lines=20,
+            show_copy_button=True,
+            container=False
+        )
+
         def process(tipo, fatos, comarca, vara, cliente, oab):
+            gerador = DefenseGenerator()
             dados = {
                 "comarca": comarca,
                 "vara": vara,
@@ -128,22 +154,43 @@ def interface():
             }
             return gerador.gerar_documento(tipo, fatos, dados)
             
-        btn.click(
+        def clear():
+            return {
+                tipo: "resposta_acusacao",
+                comarca: "",
+                vara: "",
+                cliente: "",
+                oab: "",
+                fatos: "",
+                output: ""
+            }
+            
+        gerar_btn.click(
             fn=process,
             inputs=[tipo, fatos, comarca, vara, cliente, oab],
-            outputs=output
+            outputs=output,
+            api_name="gerar_peca"
+        )
+        
+        limpar_btn.click(
+            fn=clear,
+            outputs=[tipo, comarca, vara, cliente, oab, fatos, output],
+            api_name="limpar"
         )
+
+        exemplos = [
+            ["resposta_acusacao", "Cliente preso em flagrante por furto de celular iPhone 13 da vítima Maria em 10/01/2024. É primário, tem residência fixa e trabalho. A vítima não reconheceu com certeza.", "São Paulo", "1ª", "João Silva", "SP123456"],
+            ["habeas_corpus", "Cliente preso preventivamente há 90 dias por tráfico. Apreendidas 10g de maconha, é primário, tem residência e trabalho fixos. Não foram apreendidos outros materiais típicos de tráfico.", "São Paulo", "2ª", "Pedro Santos", "SP654321"]
+        ]
         
-        exemplo = gr.Examples(
-            examples=[
-                ["resposta_acusacao", "Cliente preso em flagrante por furto de celular iPhone 13 da vítima Maria em 10/01/2024. É primário, tem residência fixa e trabalho. A vítima não reconheceu com certeza.", "São Paulo", "1ª", "João Silva", "SP123456"],
-                ["habeas_corpus", "Cliente preso preventivamente há 90 dias por tráfico. Apreendidas 10g de maconha, é primário, tem residência e trabalho fixos. Não foram apreendidos outros materiais típicos de tráfico.", "São Paulo", "2ª", "Pedro Santos", "SP654321"]
-            ],
-            inputs=[tipo, fatos, comarca, vara, cliente, oab]
+        gr.Examples(
+            examples=exemplos,
+            inputs=[tipo, fatos, comarca, vara, cliente, oab],
+            label="Exemplos de Uso"
         )
-    
+
     return demo
 
 if __name__ == "__main__":
-    demo = interface()
+    demo = create_interface()
     demo.launch()