Update app.py

app.py

@@ -1,87 +1,66 @@
 import os
 import gradio as gr
-from huggingface_hub import InferenceClient
+from huggingface_hub import HfApi, InferenceApi
 from datetime import datetime
 import logging
-from typing import Dict, List, Optional
+from typing import Dict, List
 
 # Configuração do logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Token já definido no ambiente HF Spaces
-HF_TOKEN = os.environ.get("HF_TOKEN")
-
 class DocumentGenerator:
-    """Gerencia a geração de documentos usando Claude"""
+    """Gerencia a geração de documentos usando LLM"""
     
     def __init__(self):
-        self.client = InferenceClient()
-        # Usando um modelo melhor em português
-        self.model = "bigcode/starcoder"  # Modelo mais robusto para português
+        # Usa a API key do ambiente
+        self.api = HfApi(token=os.environ.get("HF_TOKEN"))
+        # Usa modelo otimizado para português
+        self.model = "NeuralBeagle14-7B"  
     
     def generate(self, doc_type: str, context: Dict[str, str]) -> str:
-        """Gera o documento usando o modelo"""
+        """Gera o documento usando LLM"""
         try:
-            messages = [
-                {
-                    "role": "system",
-                    "content": """Você é um advogado criminalista brasileiro experiente, especializado em redigir peças processuais.
-                    Crie documentos jurídicos formais com linguagem técnica apropriada, seguindo rigorosamente o formato 
-                    das peças processuais brasileiras. Mantenha sempre a formatação e estrutura jurídica adequada."""
-                },
-                {
-                    "role": "user",
-                    "content": f"""Por favor, gere uma peça jurídica do tipo {doc_type} com as seguintes informações:
-
-DADOS DO PROCESSO:
-Nome: {context.get('client_name')}
-Número: {context.get('process_number')}
-Tribunal: {context.get('court')}
-Comarca: {context.get('jurisdiction')}
-
-EXPOSIÇÃO DOS FATOS:
-{context.get('facts')}
-
-FUNDAMENTOS JURÍDICOS:
-{context.get('legal_basis')}
-
-O documento deve ter:
-1. Endereçamento correto à autoridade
-2. Qualificação completa das partes
-3. Exposição clara e objetiva dos fatos
-4. Fundamentação jurídica
-5. Pedidos específicos
-6. Fechamento formal com data e assinatura
-
-Utilize linguagem jurídica formal e técnica."""
-                }
-            ]
+            base_prompt = f"""Atue como um advogado criminalista brasileiro experiente. 
+            Você deve gerar uma peça processual do tipo {doc_type} em português, formato jurídico brasileiro.
+            
+            DADOS:
+            - Cliente: {context.get('client_name')}
+            - Processo: {context.get('process_number')}
+            - Tribunal: {context.get('court')}
+            - Comarca: {context.get('jurisdiction')}
+            
+            FATOS:
+            {context.get('facts')}
+            
+            FUNDAMENTOS JURÍDICOS:
+            {context.get('legal_basis')}
+            
+            INSTRUÇÕES ESPECÍFICAS:
+            1. Use linguagem jurídica formal
+            2. Siga estritamente o formato de peças processuais brasileiras
+            3. Mantenha a estrutura com endereçamento, qualificação, fatos, direito e pedidos
+            4. Inclua local e data ao final
+            5. Use apenas os dados fornecidos
+            6. Evite repetições desnecessárias
+            7. Cite corretamente artigos e jurisprudência
+            8. Use formatação profissional
+            
+            COMECE A GERAR O DOCUMENTO AGORA:"""
 
-            # Configuração para melhor resultado em PT-BR
-            result = self.client.text_generation(
-                self.model,
-                messages[-1]['content'],
-                max_new_tokens=2048,
-                temperature=0.2,
-                top_p=0.9,
-                repetition_penalty=1.2,
-                do_sample=True,
-                return_full_text=True
+            # Faz a chamada à API
+            inference = InferenceApi(repo_id=f"meta-llama/{self.model}", token=os.environ.get("HF_TOKEN"))
+            response = inference(
+                base_prompt,
+                max_length=4096,
+                temperature=0.3,
+                top_p=0.95,
+                repetition_penalty=1.15,
+                do_sample=True
             )
 
-            # Formatação final
-            output = self._format_output(result[0]['generated_text'])
+            return self._format_output(response[0]['generated_text'])
             
-            # Adiciona data atual
-            current_date = datetime.now().strftime('%d de %B de %Y').replace('May', 'Maio').replace('April', 'Abril')
-            if '[DATA]' in output:
-                output = output.replace('[DATA]', current_date)
-            else:
-                output = f"{output}\n\n{context.get('jurisdiction')}, {current_date}"
-            
-            return output
-
         except Exception as e:
             logger.error(f"Erro na geração: {str(e)}")
             return f"Erro na geração do documento: {str(e)}"
@@ -90,12 +69,20 @@ Utilize linguagem jurídica formal e técnica."""
         """Formata o texto gerado"""
         if not text:
             return "Erro: Nenhum texto gerado"
+
+        # Remove o prompt da resposta
+        text = text.split("COMECE A GERAR O DOCUMENTO AGORA:")[-1].strip()
         
-        # Remove espaços extras e formata parágrafos
+        # Ajusta formatação
         lines = [line.strip() for line in text.split('\n') if line.strip()]
         formatted_text = '\n\n'.join(lines)
         
-        return formatted_text.strip()
+        # Adiciona data se não existir
+        if "[DATA]" in formatted_text:
+            current_date = datetime.now().strftime('%d de %B de %Y')
+            formatted_text = formatted_text.replace("[DATA]", current_date)
+        
+        return formatted_text
 
 class WebInterface:
     """Interface Gradio para o gerador de documentos"""
@@ -115,7 +102,6 @@ class WebInterface:
             
             with gr.Row():
                 with gr.Column():
-                    # Seleção do tipo de documento
                     doc_type = gr.Dropdown(
                         choices=[
                             "Habeas Corpus",
@@ -129,12 +115,11 @@ class WebInterface:
                         value="Habeas Corpus"
                     )
                     
-                    # Dados do processo
                     with gr.Group():
                         gr.Markdown("### Informações do Processo")
                         client_name = gr.Textbox(
                             label="Nome do Cliente",
-                            placeholder="Nome completo do cliente"
+                            placeholder="Nome completo"
                         )
                         process_number = gr.Textbox(
                             label="Número do Processo",
@@ -149,7 +134,6 @@ class WebInterface:
                             placeholder="Nome da comarca"
                         )
                     
-                    # Detalhes do caso
                     with gr.Group():
                         gr.Markdown("### Detalhes do Caso")
                         facts = gr.Textbox(
@@ -160,12 +144,10 @@ class WebInterface:
                         legal_basis = gr.Textbox(
                             label="Fundamentos Jurídicos",
                             lines=3,
-                            placeholder="Indique os fundamentos legais..."
+                            placeholder="Fundamentos legais..."
                         )
                     
-                    with gr.Row():
-                        generate_btn = gr.Button("Gerar Documento", variant="primary")
-                        clear_btn = gr.Button("Limpar", variant="secondary")
+                    generate_btn = gr.Button("📝 Gerar Documento", variant="primary")
                 
                 with gr.Column():
                     output = gr.Textbox(
@@ -173,10 +155,8 @@ class WebInterface:
                         lines=30,
                         show_copy_button=True
                     )
-                    
-                    status = gr.Textbox(label="Status")
             
-            # Exemplos de uso
+            # Exemplos
             gr.Examples(
                 examples=[
                     [
@@ -187,24 +167,6 @@ class WebInterface:
                         "São Paulo",
                         "Paciente preso em flagrante no dia 25/12/2024 por suposto furto simples (art. 155, caput, CP). O paciente é primário, possui residência fixa no distrito da culpa e ocupação lícita comprovada. Não estão presentes os requisitos autorizadores da prisão preventiva.",
                         "Art. 5º, LXVIII, CF/88; Art. 647 do CPP; Ausência dos requisitos do Art. 312 do CPP. Súmula 308 STJ."
-                    ],
-                    [
-                        "Denúncia Criminal",
-                        "Pedro Oliveira Lima",
-                        "0000456-78.2024.8.26.0000",
-                        "TRIBUNAL DE JUSTIÇA DO ESTADO DE SÃO PAULO",
-                        "São Paulo",
-                        "No dia 20/12/2024, por volta das 14h, na Rua ABC, o denunciado subtraiu para si uma carteira contendo R$ 500,00, mediante grave ameaça exercida com simulacro de arma de fogo.",
-                        "Art. 157, caput, do Código Penal. Circunstâncias do Art. 59 do CP desfavoráveis."
-                    ],
-                    [
-                        "Alegações Finais",
-                        "Maria Santos Costa",
-                        "0000789-10.2024.8.26.0000",
-                        "TRIBUNAL DE JUSTIÇA DO ESTADO DE SÃO PAULO",
-                        "São Paulo",
-                        "A ré foi denunciada por suposto furto (art. 155, CP). As provas produzidas nos autos não confirmam a autoria. As testemunhas não reconheceram a acusada e não há registros de câmeras.",
-                        "Insuficiência probatória. Art. 386, VII do CPP. Princípio in dubio pro reo."
                     ]
                 ],
                 inputs=[
@@ -220,19 +182,7 @@ class WebInterface:
                     doc_type, client_name, process_number,
                     court, jurisdiction, facts, legal_basis
                 ],
-                outputs=[output, status]
-            )
-            
-            def clear_fields():
-                return [gr.update(value="") for _ in range(6)] + ["Campos limpos"]
-            
-            clear_btn.click(
-                fn=clear_fields,
-                inputs=[],
-                outputs=[
-                    client_name, process_number, court,
-                    jurisdiction, facts, legal_basis, status
-                ]
+                outputs=output
             )
     
     def _generate_document(
@@ -240,12 +190,12 @@ class WebInterface:
         process_number: str, court: str,
         jurisdiction: str, facts: str,
         legal_basis: str
-    ) -> tuple:
+    ) -> str:
         """Gera o documento com os parâmetros fornecidos"""
+        
         try:
-            # Validação
             if not all([client_name, process_number, facts, legal_basis]):
-                return "Erro: Todos os campos obrigatórios devem ser preenchidos", "⚠️ Campos incompletos"
+                return "Erro: Todos os campos obrigatórios devem ser preenchidos"
             
             context = {
                 "client_name": client_name,
@@ -256,18 +206,15 @@ class WebInterface:
                 "legal_basis": legal_basis
             }
             
-            # Gera documento
-            result = self.generator.generate(doc_type, context)
-            
-            return result, "✅ Documento gerado com sucesso!"
+            return self.generator.generate(doc_type, context)
             
         except Exception as e:
             logger.error(f"Erro: {str(e)}")
-            return "", f"❌ Erro: {str(e)}"
+            return f"Erro na geração: {str(e)}"
     
     def launch(self):
         """Inicia a interface web"""
-        self.app.launch()
+        self.app.launch(share=True)
 
 if __name__ == "__main__":
     interface = WebInterface()