Update app.py

app.py

@@ -4,68 +4,33 @@ from huggingface_hub import InferenceClient
 from datetime import datetime
 import logging
 from typing import Dict, List, Optional
-from pathlib import Path
-import json
-import dotenv
-
-# 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',
-    handlers=[
-        logging.FileHandler("app.log"),
-        logging.StreamHandler()
-    ]
-)
+logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-def load_token() -> str:
-    """Carrega o token HF de várias fontes possíveis"""
-    # Tenta carregar do ambiente
-    token = os.getenv('HF_TOKEN')
-    if token:
-        return token
-        
-    # 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)
-                if token := config.get('hf_token'):
-                    return token
-        except Exception as e:
-            logger.error(f"Erro ao ler config.json: {e}")
-    
-    return ''
+# Token já definido no ambiente HF Spaces
+HF_TOKEN = os.environ.get("HF_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("Token HF não configurado!")
-            
-        self.client = InferenceClient(api_key=api_key)
+    def __init__(self):
+        self.client = InferenceClient(api_key=HF_TOKEN)
         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
             messages = [
                 {
                     "role": "system",
-                    "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."""
+                    "content": """Você é um advogado criminalista brasileiro experiente. 
+                    Gere documentos jurídicos formais no formato do direito brasileiro."""
                 },
                 {
                     "role": "user",
-                    "content": f"""Gere um {doc_type} completo com os seguintes dados:
+                    "content": f"""Gere um {doc_type} completo e tecnicamente preciso com:
                     
                     QUALIFICAÇÃO:
                     Cliente: {context.get('client_name')}
@@ -77,15 +42,7 @@ class DocumentGenerator:
                     {context.get('facts')}
                     
                     FUNDAMENTOS JURÍDICOS:
-                    {context.get('legal_basis')}
-                    
-                    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"""
+                    {context.get('legal_basis')}"""
                 }
             ]
             
@@ -95,7 +52,7 @@ class DocumentGenerator:
                 messages=messages,
                 temperature=0.3,
                 top_p=0.85,
-                max_tokens=2048,
+                max_tokens=2048
             )
             
             return self._format_output(completion.choices[0].message.content)
@@ -108,23 +65,13 @@ class DocumentGenerator:
         """Formata o texto gerado"""
         if not text:
             return "Erro: Nenhum texto gerado"
-        
-        # 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()
+        return text.strip()
 
 class WebInterface:
     """Interface Gradio para o gerador de documentos"""
     
     def __init__(self):
-        # Carrega o token
-        token = load_token()
-        if not token:
-            raise ValueError("Token HF não encontrado!")
-            
-        self.generator = DocumentGenerator(api_key=token)
+        self.generator = DocumentGenerator()
         self.create_interface()
     
     def create_interface(self):
@@ -138,7 +85,7 @@ class WebInterface:
             
             with gr.Row():
                 with gr.Column():
-                    # Tipo de documento
+                    # Seleção do tipo de documento
                     doc_type = gr.Dropdown(
                         choices=[
                             "Habeas Corpus",
@@ -149,76 +96,47 @@ class WebInterface:
                             "Apelação Criminal"
                         ],
                         label="Tipo de Documento",
-                        value="Habeas Corpus",
-                        info="Selecione o tipo de peça processual"
+                        value="Habeas Corpus"
                     )
                     
-                    # Informações do processo
+                    # 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/paciente",
-                            info="Digite o nome completo"
+                            placeholder="Nome completo do cliente"
                         )
-                        
                         process_number = gr.Textbox(
                             label="Número do Processo",
-                            placeholder="NNNNNNN-NN.NNNN.N.NN.NNNN",
-                            info="Número CNJ do processo"
+                            placeholder="NNNNNNN-NN.NNNN.N.NN.NNNN"
                         )
-                        
                         court = gr.Textbox(
                             label="Tribunal",
-                            value="TRIBUNAL DE JUSTIÇA DO ESTADO",
-                            info="Nome completo do tribunal"
+                            value="TRIBUNAL DE JUSTIÇA DO ESTADO"
                         )
-                        
                         jurisdiction = gr.Textbox(
                             label="Comarca",
-                            placeholder="Nome da comarca",
-                            info="Comarca onde tramita o processo"
+                            placeholder="Nome da comarca"
                         )
                     
                     # Detalhes do caso
                     with gr.Group():
                         gr.Markdown("### Detalhes do Caso")
-                        
                         facts = gr.Textbox(
                             label="Fatos",
                             lines=5,
-                            placeholder="Descreva os fatos relevantes do caso...",
-                            info="Descreva os fatos de forma clara e objetiva"
+                            placeholder="Descreva os fatos relevantes..."
                         )
-                        
                         legal_basis = gr.Textbox(
                             label="Fundamentos Jurídicos",
                             lines=3,
-                            placeholder="Fundamentos legais, jurisprudência...",
-                            info="Artigos, jurisprudência e doutrina aplicáveis"
+                            placeholder="Indique os fundamentos legais..."
                         )
                     
-                    # 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
-                    )
+                        generate_btn = gr.Button("Gerar Documento", variant="primary")
+                        clear_btn = gr.Button("Limpar", variant="secondary")
                 
-                # Coluna do documento gerado
                 with gr.Column():
                     output = gr.Textbox(
                         label="Documento Gerado",
@@ -226,9 +144,7 @@ class WebInterface:
                         show_copy_button=True
                     )
                     
-                    with gr.Row():
-                        save_btn = gr.Button("💾 Salvar como TXT")
-                        copy_btn = gr.Button("📋 Copiar")
+                    status = gr.Textbox(label="Status")
             
             # Exemplos de uso
             gr.Examples(
@@ -288,29 +204,6 @@ class WebInterface:
                     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,
@@ -319,7 +212,6 @@ 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]):
@@ -334,9 +226,6 @@ class WebInterface:
                 "legal_basis": legal_basis
             }
             
-            # Atualiza status
-            yield "", "⏳ Gerando documento..."
-            
             # Gera documento
             result = self.generator.generate(doc_type, context)
             
@@ -348,21 +237,8 @@ class WebInterface:
     
     def launch(self):
         """Inicia a interface web"""
-        self.app.launch(share=True)
+        self.app.launch()
 
 if __name__ == "__main__":
-    # Verifica se o token está configurado
-    if not HF_API_TOKEN:
-        print("\n⚠️  Token HF não encontrado!")
-        print("\nPara configurar o token:")
-        print("1. Crie um arquivo .env com:")
-        print("   HF_TOKEN=seu_token_aqui")
-        print("\nOu")
-        print("2. Crie um arquivo config.json com:")
-        print('   {"hf_token": "seu_token_aqui"}')
-        print("\nOu")
-        print("3. Configure a variável de ambiente HF_TOKEN")
-        exit(1)
-    
     interface = WebInterface()
     interface.launch()