Update app.py

app.py

@@ -1,10 +1,9 @@
-import os
 import gradio as gr
-from huggingface_hub import InferenceClient
+import os
 from datetime import datetime
-from typing import Dict, Optional
+from huggingface_hub import InferenceClient
 
-class LegalDocumentGenerator:
+class DocumentGenerator:
     def __init__(self):
         self.api_key = os.getenv("HF_API_KEY")
         if not self.api_key:
@@ -23,310 +22,117 @@ class LegalDocumentGenerator:
         except Exception as e:
             raise Exception(f"Erro na geração da seção {section}: {str(e)}")
 
-    def generate_document(self, case_info: str, doc_type: str, client_data: Dict) -> str:
-        if not all([case_info, doc_type, client_data]):
-            return "Erro: Dados incompletos"
-
-        templates = {
-            "habeas corpus": self._get_habeas_corpus_template(),
-            "defesa prévia": self._get_defesa_previa_template(),
-            "apelação criminal": self._get_apelacao_template(),
-            "petição inicial": self._get_peticao_inicial_template()
-        }
-
-        template = templates.get(doc_type.lower())
-        if not template:
-            return "Erro: Tipo de documento não suportado"
-
-        try:
-            messages = [
-                {
-                    "role": "system",
-                    "content": (
-                        "Você é um advogado criminalista especializado. Gere uma peça jurídica observando:\n"
-                        "1. Formalidade e técnica jurídica\n"
-                        "2. Fundamentação legal precisa e atualizada\n"
-                        "3. Jurisprudência relevante\n"
-                        "4. Adequação ao caso concreto\n"
-                        f"Tipo de peça: {doc_type}\n"
-                        f"Informações do caso: {case_info}"
-                    )
-                }
-            ]
-
-            sections = {}
-            for section in template["required_sections"]:
-                sections[section] = self.generate_section(messages, section)
-
-            document = template["structure"].format(
-                **sections,
-                **client_data,
-                data=datetime.now().strftime("%d/%m/%Y")
-            )
-
-            return document
-
-        except Exception as e:
-            return f"Erro na geração do documento: {str(e)}"
-
-    def _get_habeas_corpus_template(self):
-        return {
-            "structure": (
-                "EXCELENTÍSSIMO(A) SENHOR(A) DOUTOR(A) JUIZ(A) DE DIREITO DA {vara} DA {comarca}\n\n"
-                "HABEAS CORPUS COM PEDIDO LIMINAR\n\n"
-                "QUALIFICAÇÃO DO PACIENTE\n{qualificacao}\n\n"
-                "DOS FATOS\n{fatos}\n\n"
-                "DO DIREITO\n{fundamentacao}\n\n"
-                "DO CONSTRANGIMENTO ILEGAL\n{constrangimento}\n\n"
-                "DA MEDIDA LIMINAR\n{liminar}\n\n"
-                "DOS PEDIDOS\n{pedidos}\n\n"
-                "PROVAS E DOCUMENTOS\n{provas}\n\n"
-                "{comarca}, {data}\n\n"
-                "{advogado}\n"
-                "OAB/{estado} {numero_oab}"
-            ),
-            "required_sections": [
-                "qualificacao", "fatos", "fundamentacao", 
-                "constrangimento", "liminar", "pedidos", "provas"
-            ]
-        }
-
-    def _get_defesa_previa_template(self):
-        return {
-            "structure": (
-                "EXCELENTÍSSIMO(A) SENHOR(A) DOUTOR(A) JUIZ(A) DE DIREITO DA {vara} DA {comarca}\n\n"
-                "DEFESA PRÉVIA\n\n"
-                "Autos nº {numero_processo}\n\n"
-                "QUALIFICAÇÃO DO ACUSADO\n{qualificacao}\n\n"
-                "DOS FATOS\n{fatos}\n\n"
-                "DO DIREITO\n{fundamentacao}\n\n"
-                "DAS PRELIMINARES\n{preliminares}\n\n"
-                "DO MÉRITO\n{merito}\n\n"
-                "DAS PROVAS\n{provas}\n\n"
-                "DOS PEDIDOS\n{pedidos}\n\n"
-                "{comarca}, {data}\n\n"
-                "{advogado}\n"
-                "OAB/{estado} {numero_oab}"
-            ),
-            "required_sections": [
-                "qualificacao", "fatos", "fundamentacao", "preliminares",
-                "merito", "provas", "pedidos"
-            ]
-        }
-
-    def _get_apelacao_template(self):
-        return {
-            "structure": (
-                "EXCELENTÍSSIMO(A) SENHOR(A) DOUTOR(A) DESEMBARGADOR(A) RELATOR(A) DO TRIBUNAL DE JUSTIÇA DO ESTADO DE {estado}\n\n"
-                "RAZÕES DE APELAÇÃO\n\n"
-                "Autos nº {numero_processo}\n\n"
-                "QUALIFICAÇÃO DO APELANTE\n{qualificacao}\n\n"
-                "DOS FATOS\n{fatos}\n\n"
-                "DAS PRELIMINARES\n{preliminares}\n\n"
-                "DO MÉRITO\n{merito}\n\n"
-                "DO DIREITO\n{fundamentacao}\n\n"
-                "DOS PEDIDOS\n{pedidos}\n\n"
-                "{comarca}, {data}\n\n"
-                "{advogado}\n"
-                "OAB/{estado} {numero_oab}"
-            ),
-            "required_sections": [
-                "qualificacao", "fatos", "preliminares", "merito",
-                "fundamentacao", "pedidos"
-            ]
-        }
-
-    def _get_peticao_inicial_template(self):
-        return {
-            "structure": (
-                "EXCELENTÍSSIMO(A) SENHOR(A) DOUTOR(A) JUIZ(A) DE DIREITO DA {vara} DA {comarca}\n\n"
-                "PETIÇÃO INICIAL\n\n"
-                "QUALIFICAÇÃO DO AUTOR\n{qualificacao}\n\n"
-                "DOS FATOS\n{fatos}\n\n"
-                "DO DIREITO\n{fundamentacao}\n\n"
-                "DO PEDIDO LIMINAR\n{liminar}\n\n"
-                "DOS PEDIDOS\n{pedidos}\n\n"
-                "DAS PROVAS\n{provas}\n\n"
-                "DO VALOR DA CAUSA\n{valor_causa}\n\n"
-                "{comarca}, {data}\n\n"
-                "{advogado}\n"
-                "OAB/{estado} {numero_oab}"
-            ),
-            "required_sections": [
-                "qualificacao", "fatos", "fundamentacao", "liminar",
-                "pedidos", "provas", "valor_causa"
-            ]
-        }
+    def gerar_documento(self, tipo_documento: str, dados: dict) -> str:
+        messages = [
+            {"role": "system", "content": "Você é um assistente especializado em peças jurídicas criminais."},
+            {"role": "user", "content": f"Dados para geração de {tipo_documento}:\n{str(dados)}"}
+        ]
+        
+        if tipo_documento == "denuncia":
+            sections = ["qualificacao", "narrativa", "tipificacao", "pedido"]
+        else:
+            sections = ["relatorio", "fundamentacao", "pedido"]
+            
+        documento = ""
+        for section in sections:
+            documento += self.generate_section(messages, section) + "\n\n"
+            
+        dados['data'] = datetime.now().strftime("%d/%m/%Y")
+        return documento
 
-def create_interface():
-    generator = LegalDocumentGenerator()
+def interface():
+    gerador = DocumentGenerator()
     
-    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"
-            }
+    def processar(tipo_doc, comarca, vara, estado, nome_reu, qualificacao, narrativa, artigo, testemunhas, 
+                 numero_processo="", narrativa_processual="", fundamentacao="", pedido_final=""):
+        dados = {
+            "comarca": comarca,
+            "vara": vara,
+            "estado": estado,
+            "nome_reu": nome_reu,
+            "qualificacao_reu": qualificacao,
+            "narrativa": narrativa,
+            "artigo": artigo,
+            "testemunhas": testemunhas,
+            "numero_processo": numero_processo,
+            "narrativa_processual": narrativa_processual,
+            "fundamentacao": fundamentacao,
+            "pedido_final": pedido_final
         }
-    }
+        
+        return gerador.gerar_documento(tipo_doc, dados)
 
-    with gr.Blocks() as interface:
+    with gr.Blocks(title="Gerador de Peças Jurídicas Criminais") as demo:
         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..."
+        with gr.Row():
+            tipo_documento = gr.Radio(
+                choices=["denuncia", "alegacoes_finais"],
+                label="Tipo de Documento",
+                value="denuncia"
             )
             
-            case_info = gr.Textbox(
-                label="Informações do Caso",
-                placeholder="Descreva o caso detalhadamente",
-                lines=5
-            )
-            doc_type = gr.Dropdown(
-                label="Tipo de Peça",
-                choices=[
-                    "habeas corpus",
-                    "defesa prévia",
-                    "apelação criminal",
-                    "petição inicial"
-                ]
-            )
-
-        with gr.Tab("Dados do Cliente"):
-            with gr.Row():
-                client_name = gr.Textbox(label="Nome")
-                client_nationality = gr.Textbox(label="Nacionalidade")
-            with gr.Row():
-                client_profession = gr.Textbox(label="Profissão")
-                client_id = gr.Textbox(label="RG")
-            with gr.Row():
-                client_address = gr.Textbox(label="Endereço")
-                client_city = gr.Textbox(label="Cidade")
-            with gr.Row():
-                client_state = gr.Textbox(label="Estado")
-                client_zip = gr.Textbox(label="CEP")
-
-        with gr.Tab("Dados Processuais"):
-            with gr.Row():
-                court = gr.Textbox(label="Vara")
-                jurisdiction = gr.Textbox(label="Comarca")
-            with gr.Row():
-                process_number = gr.Textbox(label="Número do Processo")
-                lawyer_name = gr.Textbox(label="Nome do Advogado")
-            with gr.Row():
-                lawyer_oab = gr.Textbox(label="Número OAB")
-                lawyer_state = gr.Textbox(label="Estado OAB")
-
         with gr.Row():
-            generate_btn = gr.Button("Gerar Documento")
-            clear_btn = gr.Button("Limpar Campos")
-
-        output = gr.Textbox(
-            label="Documento Gerado",
-            lines=20,
-            placeholder="O documento gerado aparecerá aqui"
-        )
-
-        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"]
-            ]
-
-        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(
-            generator.generate_document,
+            with gr.Column():
+                comarca = gr.Textbox(label="Comarca")
+                vara = gr.Textbox(label="Vara")
+                estado = gr.Textbox(label="Estado")
+                nome_reu = gr.Textbox(label="Nome do Réu")
+                qualificacao = gr.Textbox(
+                    label="Qualificação do Réu",
+                    placeholder="brasileiro, casado, etc..."
+                )
+                
+            with gr.Column():
+                narrativa = gr.Textbox(
+                    label="Narrativa dos Fatos",
+                    lines=5
+                )
+                artigo = gr.Textbox(label="Artigo(s) do CP")
+                testemunhas = gr.Textbox(
+                    label="Testemunhas",
+                    lines=3,
+                    placeholder="1. Nome da testemunha\n2. Nome da testemunha"
+                )
+
+        with gr.Row(visible=False) as campos_alegacoes:
+            with gr.Column():
+                numero_processo = gr.Textbox(label="Número do Processo")
+                narrativa_processual = gr.Textbox(
+                    label="Narrativa Processual",
+                    lines=5
+                )
+                fundamentacao = gr.Textbox(
+                    label="Fundamentação",
+                    lines=5
+                )
+                pedido_final = gr.Textbox(
+                    label="Pedido Final",
+                    placeholder="condenação do réu nas sanções do art. X..."
+                )
+
+        def atualizar_campos(escolha):
+            return gr.Row(visible=(escolha == "alegacoes_finais"))
+
+        tipo_documento.change(atualizar_campos, tipo_documento, campos_alegacoes)
+        
+        with gr.Row():
+            gerar_btn = gr.Button("Gerar Documento")
+            
+        saida = gr.Textbox(label="Documento Gerado", lines=10)
+        
+        gerar_btn.click(
+            processar,
             inputs=[
-                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
+                tipo_documento, comarca, vara, estado, nome_reu, qualificacao,
+                narrativa, artigo, testemunhas, numero_processo,
+                narrativa_processual, fundamentacao, pedido_final
             ],
-            outputs=output
-        )
-
-        clear_btn.click(
-            lambda: [""] * 16,
-            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
-            ]
+            outputs=saida
         )
-
-    return interface
     
+    return demo
+
 if __name__ == "__main__":
-    interface = create_interface()
-    interface.launch()
+    demo = interface()
+    demo.launch()