Update app.py

app.py

@@ -160,11 +160,58 @@ class LegalDocumentGenerator:
 
 def create_interface():
     generator = LegalDocumentGenerator()
+    
+    example_cases = {
+        "habeas corpus": {
+            "case_info": "Cliente João da Silva foi preso em flagrante no dia 15/12/2023 por suposto furto em loja, mas há provas em vídeo mostrando que ele estava em outro local no momento do crime.",
+            "client_data": {
+                "name": "João da Silva",
+                "nationality": "Brasileiro",
+                "profession": "Comerciante",
+                "id": "12.345.678-9",
+                "address": "Rua das Flores, 123",
+                "city": "São Paulo",
+                "state": "SP",
+                "zip": "01234-567",
+                "court": "1ª Vara Criminal",
+                "jurisdiction": "São Paulo",
+                "process_number": "0123456-78.2023.8.26.0001",
+                "lawyer_name": "Dr. Pedro Santos",
+                "lawyer_oab": "123456",
+                "lawyer_state": "SP"
+            }
+        },
+        "defesa prévia": {
+            "case_info": "Cliente Maria Oliveira foi denunciada por tráfico de drogas após revista policial sem mandado em sua residência, onde foram encontradas substâncias ilícitas.",
+            "client_data": {
+                "name": "Maria Oliveira",
+                "nationality": "Brasileira",
+                "profession": "Professora",
+                "id": "98.765.432-1",
+                "address": "Av. Principal, 456",
+                "city": "São Paulo",
+                "state": "SP",
+                "zip": "04321-765",
+                "court": "2ª Vara Criminal",
+                "jurisdiction": "São Paulo",
+                "process_number": "0987654-32.2023.8.26.0002",
+                "lawyer_name": "Dra. Ana Costa",
+                "lawyer_oab": "654321",
+                "lawyer_state": "SP"
+            }
+        }
+    }
 
     with gr.Blocks() as interface:
         gr.Markdown("# Gerador de Peças Jurídicas Criminais")
         
         with gr.Tab("Dados do Caso"):
+            example_selector = gr.Dropdown(
+                label="Exemplos",
+                choices=["Selecione um exemplo...", "Exemplo Habeas Corpus", "Exemplo Defesa Prévia"],
+                value="Selecione um exemplo..."
+            )
+            
             case_info = gr.Textbox(
                 label="Informações do Caso",
                 placeholder="Descreva o caso detalhadamente",
@@ -215,37 +262,48 @@ def create_interface():
             placeholder="O documento gerado aparecerá aqui"
         )
 
-        def generate(
-            case_info, doc_type, 
-            name, nationality, profession, id_num,
-            address, city, state, zip_code,
-            court, jurisdiction, process_num,
-            lawyer_name, lawyer_oab, lawyer_state
-        ):
-            client_data = {
-                "nome": name,
-                "nacionalidade": nationality,
-                "profissao": profession,
-                "rg": id_num,
-                "endereco": address,
-                "cidade": city,
-                "estado": state,
-                "cep": zip_code,
-                "vara": court,
-                "comarca": jurisdiction,
-                "numero_processo": process_num,
-                "advogado": lawyer_name,
-                "numero_oab": lawyer_oab,
-                "estado_oab": lawyer_state
-            }
-            
-            return generator.generate_document(case_info, doc_type, client_data)
+        def load_example(example_choice):
+            if example_choice == "Exemplo Habeas Corpus":
+                data = example_cases["habeas corpus"]
+            elif example_choice == "Exemplo Defesa Prévia":
+                data = example_cases["defesa prévia"]
+            else:
+                return [""] * 16
+
+            client = data["client_data"]
+            return [
+                data["case_info"],
+                example_choice.lower().replace("exemplo ", ""),
+                client["name"],
+                client["nationality"],
+                client["profession"],
+                client["id"],
+                client["address"],
+                client["city"],
+                client["state"],
+                client["zip"],
+                client["court"],
+                client["jurisdiction"],
+                client["process_number"],
+                client["lawyer_name"],
+                client["lawyer_oab"],
+                client["lawyer_state"]
+            ]
 
-        def clear_fields():
-            return [""] * 16
+        example_selector.change(
+            load_example,
+            inputs=[example_selector],
+            outputs=[
+                case_info, doc_type,
+                client_name, client_nationality, client_profession, client_id,
+                client_address, client_city, client_state, client_zip,
+                court, jurisdiction, process_number,
+                lawyer_name, lawyer_oab, lawyer_state
+            ]
+        )
 
         generate_btn.click(
-            generate,
+            generator.generate_document,
             inputs=[
                 case_info, doc_type,
                 client_name, client_nationality, client_profession, client_id,
@@ -257,7 +315,7 @@ def create_interface():
         )
 
         clear_btn.click(
-            clear_fields,
+            lambda: [""] * 16,
             outputs=[
                 case_info, doc_type,
                 client_name, client_nationality, client_profession, client_id,
@@ -268,7 +326,7 @@ def create_interface():
         )
 
     return interface
-
+    
 if __name__ == "__main__":
     interface = create_interface()
     interface.launch()