Update app.py

app.py

@@ -8,59 +8,60 @@ from pathlib import Path
 import json
 import dotenv
 
-# Carrega variáveis de ambiente do arquivo .env se existir
+# Carrega variáveis de ambiente
 dotenv.load_dotenv()
 
 # Configuração do logging
 logging.basicConfig(
     level=logging.INFO,
-    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+    handlers=[
+        logging.FileHandler("app.log"),
+        logging.StreamHandler()
+    ]
 )
 logger = logging.getLogger(__name__)
 
 def load_token() -> str:
     """Carrega o token HF de várias fontes possíveis"""
-    # Prioridade: 1. Variável de ambiente, 2. Arquivo .env, 3. arquivo config.json
-    
-    # Verifica variável de ambiente
+    # Tenta carregar do ambiente
     token = os.getenv('HF_TOKEN')
     if token:
         return token
         
-    # Verifica arquivo config.json
+    # Tenta carregar do arquivo de configuração
     config_file = Path('config.json')
     if config_file.exists():
         try:
             with open(config_file, 'r') as f:
                 config = json.load(f)
-                return config.get('hf_token', '')
+                if token := config.get('hf_token'):
+                    return token
         except Exception as e:
             logger.error(f"Erro ao ler config.json: {e}")
     
     return ''
 
-# Carrega o token
-HF_API_TOKEN = load_token()
-
 class DocumentGenerator:
     """Gerencia a geração de documentos usando HF Inference API"""
     
     def __init__(self, api_key: str):
         if not api_key:
-            raise ValueError("API Key não encontrada. Configure HF_TOKEN nas variáveis de ambiente ou no arquivo config.json")
-        
+            raise ValueError("Token HF não configurado!")
+            
         self.client = InferenceClient(api_key=api_key)
         self.model = "mistralai/Mistral-7B-Instruct-v0.2"
     
     def generate(self, doc_type: str, context: Dict[str, str]) -> str:
         """Gera o documento usando o modelo"""
         try:
-            # Prepara as mensagens para o chat
+            # Prepara as mensagens
             messages = [
                 {
                     "role": "system",
-                    "content": """Você é um advogado criminalista brasileiro experiente.
-                    Gere documentos jurídicos formais no formato do direito brasileiro."""
+                    "content": """Você é um advogado criminalista brasileiro altamente experiente, 
+                    especializado em redigir peças processuais. Gere documentos jurídicos formais, 
+                    tecnicamente precisos e no formato do direito brasileiro."""
                 },
                 {
                     "role": "user",
@@ -78,11 +79,17 @@ class DocumentGenerator:
                     FUNDAMENTOS JURÍDICOS:
                     {context.get('legal_basis')}
                     
-                    Use formato jurídico adequado, com todas as partes necessárias."""
+                    O documento deve seguir a formatação jurídica padrão brasileira, incluindo:
+                    1. Cabeçalho com endereçamento correto
+                    2. Qualificação completa das partes
+                    3. Exposição clara dos fatos
+                    4. Fundamentação jurídica sólida
+                    5. Pedidos específicos
+                    6. Fechamento formal com local, data e espaço para assinatura"""
                 }
             ]
             
-            # Faz a chamada à API
+            # Gera o documento
             completion = self.client.chat.completions.create(
                 model=self.model,
                 messages=messages,
@@ -101,13 +108,23 @@ class DocumentGenerator:
         """Formata o texto gerado"""
         if not text:
             return "Erro: Nenhum texto gerado"
-        return text.strip()
+        
+        # Remove linhas vazias extras
+        lines = [line.strip() for line in text.split('\n') if line.strip()]
+        formatted_text = '\n\n'.join(lines)
+        
+        return formatted_text.strip()
 
 class WebInterface:
     """Interface Gradio para o gerador de documentos"""
     
     def __init__(self):
-        self.generator = DocumentGenerator(api_key=HF_API_TOKEN)
+        # Carrega o token
+        token = load_token()
+        if not token:
+            raise ValueError("Token HF não encontrado!")
+            
+        self.generator = DocumentGenerator(api_key=token)
         self.create_interface()
     
     def create_interface(self):
@@ -115,12 +132,13 @@ class WebInterface:
         
         with gr.Blocks(theme=gr.themes.Soft()) as self.app:
             gr.Markdown("""
-            # Criminal.ai - Gerador de Documentos Jurídicos
-            Sistema de geração de peças processuais criminais
+            # Criminal.ai - Gerador de Peças Processuais
+            ### Sistema Inteligente para Geração de Documentos Jurídicos
             """)
             
             with gr.Row():
                 with gr.Column():
+                    # Tipo de documento
                     doc_type = gr.Dropdown(
                         choices=[
                             "Habeas Corpus",
@@ -131,48 +149,123 @@ class WebInterface:
                             "Apelação Criminal"
                         ],
                         label="Tipo de Documento",
-                        value="Habeas Corpus"
+                        value="Habeas Corpus",
+                        info="Selecione o tipo de peça processual"
                     )
                     
+                    # Informações do processo
                     with gr.Group():
                         gr.Markdown("### Informações do Processo")
+                        
                         client_name = gr.Textbox(
                             label="Nome do Cliente",
-                            placeholder="Nome completo"
+                            placeholder="Nome completo do cliente/paciente",
+                            info="Digite o nome completo"
                         )
+                        
                         process_number = gr.Textbox(
                             label="Número do Processo",
-                            placeholder="NNNNNNN-NN.NNNN.N.NN.NNNN"
+                            placeholder="NNNNNNN-NN.NNNN.N.NN.NNNN",
+                            info="Número CNJ do processo"
                         )
+                        
                         court = gr.Textbox(
                             label="Tribunal",
-                            value="TRIBUNAL DE JUSTIÇA DO ESTADO"
+                            value="TRIBUNAL DE JUSTIÇA DO ESTADO",
+                            info="Nome completo do tribunal"
                         )
+                        
                         jurisdiction = gr.Textbox(
-                            label="Comarca"
+                            label="Comarca",
+                            placeholder="Nome da comarca",
+                            info="Comarca onde tramita o processo"
                         )
                     
+                    # Detalhes do caso
                     with gr.Group():
                         gr.Markdown("### Detalhes do Caso")
+                        
                         facts = gr.Textbox(
                             label="Fatos",
                             lines=5,
-                            placeholder="Descreva os fatos..."
+                            placeholder="Descreva os fatos relevantes do caso...",
+                            info="Descreva os fatos de forma clara e objetiva"
                         )
+                        
                         legal_basis = gr.Textbox(
                             label="Fundamentos Jurídicos",
-                            lines=3
+                            lines=3,
+                            placeholder="Fundamentos legais, jurisprudência...",
+                            info="Artigos, jurisprudência e doutrina aplicáveis"
                         )
                     
-                    generate_btn = gr.Button("Gerar Documento", variant="primary")
+                    # Botões
+                    with gr.Row():
+                        generate_btn = gr.Button(
+                            "📝 Gerar Documento",
+                            variant="primary",
+                            scale=2
+                        )
+                        
+                        clear_btn = gr.Button(
+                            "🗑️ Limpar Campos",
+                            variant="secondary",
+                            scale=1
+                        )
+                    
+                    status = gr.Textbox(
+                        label="Status",
+                        interactive=False
+                    )
                 
+                # Coluna do documento gerado
                 with gr.Column():
                     output = gr.Textbox(
                         label="Documento Gerado",
                         lines=30,
                         show_copy_button=True
                     )
-                    status = gr.Textbox(label="Status")
+                    
+                    with gr.Row():
+                        save_btn = gr.Button("💾 Salvar como TXT")
+                        copy_btn = gr.Button("📋 Copiar")
+            
+            # Exemplos de uso
+            gr.Examples(
+                examples=[
+                    [
+                        "Habeas Corpus",
+                        "João da Silva Santos",
+                        "0000123-45.2024.8.26.0000",
+                        "TRIBUNAL DE JUSTIÇA DO ESTADO DE SÃO PAULO",
+                        "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=[
+                    doc_type, client_name, process_number,
+                    court, jurisdiction, facts, legal_basis
+                ]
+            )
             
             # Eventos
             generate_btn.click(
@@ -183,6 +276,41 @@ class WebInterface:
                 ],
                 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
+                ]
+            )
+            
+            def save_to_file(text):
+                if not text:
+                    return "Nenhum documento para salvar"
+                filename = f"documento_{datetime.now().strftime('%Y%m%d_%H%M%S')}.txt"
+                try:
+                    with open(filename, 'w', encoding='utf-8') as f:
+                        f.write(text)
+                    return f"✅ Documento salvo como {filename}"
+                except Exception as e:
+                    return f"❌ Erro ao salvar: {str(e)}"
+            
+            save_btn.click(
+                fn=save_to_file,
+                inputs=[output],
+                outputs=[status]
+            )
+            
+            copy_btn.click(
+                fn=lambda x: x,
+                inputs=[output],
+                outputs=[output]
+            )
     
     def _generate_document(
         self, doc_type: str, client_name: str,
@@ -191,9 +319,11 @@ class WebInterface:
         legal_basis: str
     ) -> tuple:
         """Gera o documento com os parâmetros fornecidos"""
+        
         try:
+            # Validação
             if not all([client_name, process_number, facts, legal_basis]):
-                return "Erro: Preencha todos os campos obrigatórios", "⚠️ Campos incompletos"
+                return "Erro: Todos os campos obrigatórios devem ser preenchidos", "⚠️ Campos incompletos"
             
             context = {
                 "client_name": client_name,
@@ -204,8 +334,13 @@ class WebInterface:
                 "legal_basis": legal_basis
             }
             
+            # Atualiza status
+            yield "", "⏳ Gerando documento..."
+            
+            # Gera documento
             result = self.generator.generate(doc_type, context)
-            return result, "✅ Documento gerado com sucesso"
+            
+            return result, "✅ Documento gerado com sucesso!"
             
         except Exception as e:
             logger.error(f"Erro: {str(e)}")