Upload app.py with huggingface_hub

app.py

@@ -0,0 +1,353 @@
+"""
+🚗 Agente DETRAN-RJ - Assistente Virtual
+=========================================
+Assistente virtual do Departamento de Trânsito do Estado do Rio de Janeiro.
+Ajuda cidadãos com informações sobre habilitação, veículos, multas, taxas e serviços.
+
+Template Version: 1.0.0
+Based on: Prometheus Agent Pattern
+"""
+
+import gradio as gr
+import json
+from pathlib import Path
+
+# ============================================================================
+# BASE DE CONHECIMENTO
+# ============================================================================
+
+class KnowledgeBase:
+    """Sistema de busca na base de conhecimento do DETRAN-RJ."""
+    
+    def __init__(self):
+        self.faqs = []
+        self.faqs_duda = []
+        self.loaded = False
+        self.load_data()
+    
+    def load_data(self):
+        """Carrega as FAQs do arquivo JSON."""
+        try:
+            # Carrega FAQs gerais
+            faqs_path = Path("data/faqs.json")
+            if faqs_path.exists():
+                with open(faqs_path, "r", encoding="utf-8") as f:
+                    data = json.load(f)
+                    self.faqs = data.get("faqs", [])
+            
+            # Carrega FAQs de DUDA/taxas
+            duda_path = Path("data/faqs-duda.json")
+            if duda_path.exists():
+                with open(duda_path, "r", encoding="utf-8") as f:
+                    data = json.load(f)
+                    self.faqs_duda = data.get("faqs", [])
+            
+            total = len(self.faqs) + len(self.faqs_duda)
+            if total > 0:
+                self.loaded = True
+                print(f"✅ Carregadas {total} FAQs ({len(self.faqs)} gerais + {len(self.faqs_duda)} DUDA)")
+        except Exception as e:
+            print(f"⚠️ Erro ao carregar FAQs: {e}")
+    
+    def search(self, query: str) -> str:
+        """Busca resposta nas FAQs baseado em palavras-chave."""
+        query_lower = query.lower()
+        query_words = set(query_lower.split())
+        
+        best_match = None
+        best_score = 0
+        
+        # Busca em todas as FAQs
+        all_faqs = self.faqs + self.faqs_duda
+        
+        for faq in all_faqs:
+            keywords = faq.get("keywords", [])
+            question = faq.get("question", "").lower()
+            
+            # Score por keywords
+            score = sum(2 for kw in keywords if kw.lower() in query_lower)
+            # Score por palavras na pergunta
+            score += sum(1 for word in query_words if word in question and len(word) > 3)
+            
+            if score > best_score:
+                best_score = score
+                best_match = faq.get("answer", "")
+        
+        return best_match if best_score >= 2 else None
+
+# Inicializa a base de conhecimento
+kb = KnowledgeBase()
+
+# ============================================================================
+# RESPOSTAS PRÉ-DEFINIDAS
+# ============================================================================
+
+RESPONSES = {
+    "saudacao": """Olá! 👋 Sou o assistente virtual do **DETRAN-RJ**.
+
+Posso ajudá-lo com informações sobre:
+- 🚗 **CNH** - Primeira habilitação, renovação, segunda via
+- 🚙 **Veículos** - Licenciamento, transferência, vistoria
+- 🪪 **Identidade** - RG, CIN, segunda via
+- ⚠️ **Multas** - Consulta, recursos, pontuação
+- 💰 **Taxas** - Valores DUDA 2026, formas de pagamento
+- 📍 **Postos** - Localização e agendamento
+
+Como posso ajudá-lo hoje?""",
+
+    "contato": """## 📞 Contatos Oficiais DETRAN-RJ
+
+| Canal | Informação |
+|-------|------------|
+| **Site** | [www.detran.rj.gov.br](https://www.detran.rj.gov.br) |
+| **Telefone** | (21) 3460-4040 / 3460-4041 / 3460-4042 |
+| **Horário** | Segunda a sexta, 6h às 21h |
+| **Posto Digital** | Serviços online 24h |
+
+Para agendamento de atendimento presencial, acesse o site oficial.""",
+
+    "primeira_habilitacao": """## 🚗 Primeira Habilitação (CNH)
+
+**Requisitos:**
+- Ter no mínimo 18 anos
+- Ser alfabetizado
+- Possuir RG e CPF
+
+**Etapas do processo:**
+1. Abertura do processo no DETRAN ou autoescola
+2. Exame médico de aptidão física e mental
+3. Exame psicológico (avaliação psicotécnica)
+4. Curso teórico-técnico (45 horas/aula)
+5. Exame teórico sobre legislação
+6. Curso prático de direção (20 horas/aula)
+7. Exame prático de direção
+
+**Valores 2026:**
+| Item | Valor |
+|------|-------|
+| Taxa DUDA (código 201-1) | R$ 419,55 |
+| Exame médico | R$ 100 a R$ 200 |
+| Exame psicológico | R$ 100 a R$ 200 |
+| Autoescola (média) | R$ 800 a R$ 1.500 |
+
+**Custo total estimado:** R$ 1.700 a R$ 2.800""",
+
+    "renovacao_cnh": """## 🔄 Renovação da CNH
+
+**Prazos de validade por idade:**
+| Idade | Validade |
+|-------|----------|
+| Até 49 anos | 10 anos |
+| 50 a 69 anos | 5 anos |
+| 70 anos ou mais | 3 anos |
+
+**Como renovar:**
+1. Agende pelo site do DETRAN-RJ
+2. Realize o exame médico em clínica credenciada
+3. Pague a taxa DUDA (código 203-8): **R$ 209,78**
+4. Aguarde a nova CNH (Correios ou retirada)
+
+**Importante:** Inicie o processo até 30 dias antes do vencimento.""",
+
+    "licenciamento": """## 📋 Licenciamento Anual 2026
+
+**Calendário por final de placa:**
+| Final da Placa | Prazo |
+|----------------|-------|
+| 0, 1, 2 | Até 31/05 |
+| 3, 4, 5 | Até 30/06 |
+| 6, 7, 8, 9 | Até 31/07 |
+
+**Valor total:** R$ 293,71
+- Taxa de licenciamento: R$ 209,78
+- Emissão CRLV-e: R$ 83,93
+
+**Como pagar:**
+- Site do DETRAN-RJ
+- App Carteira Digital de Trânsito (CDT)
+- Rede bancária autorizada
+
+O CRLV digital fica disponível automaticamente no app CDT após o pagamento.""",
+
+    "transferencia": """## 🔄 Transferência de Veículo
+
+**Prazo:** 30 dias após a compra
+
+**Documentos necessários:**
+- CRV preenchido e assinado (firma reconhecida)
+- RG e CPF do comprador
+- Comprovante de residência
+- Vistoria do veículo
+
+**Valores 2026:**
+| Serviço | Código DUDA | Valor |
+|---------|-------------|-------|
+| Transferência | 014-0 | R$ 209,78 |
+| Gravame (se financiado) | 018-3 | R$ 233,09 |
+| Placas Mercosul (veículo) | 037-0 | R$ 240,34 |
+| Placas Mercosul (moto) | 041-8 | R$ 120,17 |
+
+**Importante:** O vendedor deve fazer a Comunicação de Venda em até 30 dias.""",
+
+    "multas": """## ⚠️ Multas e Infrações
+
+**Consultar multas:**
+- Site: www.detran.rj.gov.br
+- App: Carteira Digital de Trânsito (CDT)
+- Informe placa e Renavam
+
+**Pontuação na CNH:**
+| Gravidade | Pontos |
+|-----------|--------|
+| Leve | 3 pontos |
+| Média | 4 pontos |
+| Grave | 5 pontos |
+| Gravíssima | 7 pontos |
+
+**Limite para suspensão:** 40 pontos em 12 meses (ou 20 pontos para PPD)
+
+**Prazos para recurso:**
+- Defesa prévia: 30 dias após notificação
+- Recurso JARI: 30 dias após penalidade
+- Recurso CETRAN: 30 dias após decisão JARI""",
+
+    "duda_valores": """## 💰 Tabela DUDA 2026 - Principais Valores
+
+| Serviço | Código | Valor |
+|---------|--------|-------|
+| Primeira habilitação | 201-1 | R$ 419,55 |
+| Reexame (prova) | 202-0 | R$ 156,80 |
+| Renovação CNH | 203-8 | R$ 209,78 |
+| Mudança/adição categoria | 204-6 | R$ 209,78 |
+| Segunda via CNH | 205-4 | R$ 209,78 |
+| Transferência veículo | 014-0 | R$ 209,78 |
+| Gravame/financiamento | 018-3 | R$ 233,09 |
+| Placas Mercosul (veículo) | 037-0 | R$ 240,34 |
+| Placas Mercosul (moto) | 041-8 | R$ 120,17 |
+| PID (internacional) | 207-0 | R$ 209,78 |
+
+**Onde pagar:** Banco Bradesco (site ou agência) ou rede autorizada.
+**Prazo compensação:** 48h (Bradesco) ou 96h (outros bancos).""",
+
+    "identidade": """## 🪪 Carteira de Identidade (CIN/RG)
+
+**Documentos necessários:**
+- Certidão de nascimento ou casamento
+- CPF (obrigatório para CIN)
+- Comprovante de residência
+
+**Valores:**
+- Primeira via: **Gratuita**
+- Segunda via: Taxa aplicável (exceto roubo/furto com B.O.)
+
+**Prazo de entrega:** Até 15 dias úteis
+
+**Diferença RG x CIN:**
+A CIN (Carteira de Identidade Nacional) é o novo documento que usa o CPF como número único, tem QR Code e validade nacional. Os RGs antigos são válidos até 2032.
+
+**Agendamento:** Site ou app do DETRAN-RJ""",
+
+    "default": """Obrigado pela sua pergunta! 
+
+Para informações mais específicas, recomendo:
+
+📱 **Site oficial:** [www.detran.rj.gov.br](https://www.detran.rj.gov.br)
+📞 **Telefone:** (21) 3460-4040
+🏢 **Posto Digital:** Serviços online 24h
+
+**Posso ajudar com:**
+- CNH (renovação, primeira habilitação, segunda via)
+- Veículos (licenciamento, transferência, vistoria)
+- Identidade (RG, CIN)
+- Multas e infrações
+- Taxas e valores DUDA 2026
+- Localização de postos
+
+Como posso ajudá-lo?"""
+}
+
+# ============================================================================
+# DETECÇÃO DE INTENÇÃO
+# ============================================================================
+
+INTENT_KEYWORDS = {
+    "saudacao": ["olá", "oi", "hello", "hi", "bom dia", "boa tarde", "boa noite", "hey", "opa"],
+    "contato": ["telefone", "contato", "endereço", "horário", "site", "falar com"],
+    "primeira_habilitacao": ["primeira habilitação", "tirar carteira", "tirar cnh", "primeira cnh", "como tirar"],
+    "renovacao_cnh": ["renovar", "renovação", "vencida", "vencimento cnh", "prazo cnh"],
+    "licenciamento": ["licenciamento", "licenciar", "crlv", "documento do carro", "calendario"],
+    "transferencia": ["transferir", "transferência", "comprei carro", "vender carro", "comunicação de venda"],
+    "multas": ["multa", "infração", "pontos", "recurso", "jari", "suspensão", "lei seca"],
+    "duda_valores": ["duda", "valor", "preço", "quanto custa", "taxa", "tabela", "código"],
+    "identidade": ["identidade", "rg", "cin", "documento", "carteira de identidade"],
+}
+
+def detect_intent(message: str) -> str:
+    """Detecta a intenção do usuário baseado em palavras-chave."""
+    msg = message.lower()
+    
+    for intent, keywords in INTENT_KEYWORDS.items():
+        if any(kw in msg for kw in keywords):
+            return intent
+    
+    return "default"
+
+# ============================================================================
+# FUNÇÃO DE RESPOSTA
+# ============================================================================
+
+def respond(message: str, history: list) -> str:
+    """Gera resposta para o chatbot."""
+    if not message or not message.strip():
+        return "Por favor, digite uma mensagem para que eu possa ajudá-lo."
+    
+    # Detecta intenção
+    intent = detect_intent(message)
+    
+    # Se for default, tenta buscar nas FAQs
+    if intent == "default" and kb.loaded:
+        faq_response = kb.search(message)
+        if faq_response:
+            return faq_response
+    
+    # Retorna resposta pré-definida
+    return RESPONSES.get(intent, RESPONSES["default"])
+
+# ============================================================================
+# INTERFACE GRADIO
+# ============================================================================
+
+EXAMPLES = [
+    "Olá!",
+    "Como tirar a primeira habilitação?",
+    "Quanto custa renovar a CNH?",
+    "Qual o prazo do licenciamento 2026?",
+    "Como transferir um veículo?",
+    "Quais os valores do DUDA 2026?",
+    "Como consultar multas?",
+    "Como tirar a carteira de identidade?",
+]
+
+demo = gr.ChatInterface(
+    fn=respond,
+    title="🚗 Agente DETRAN-RJ",
+    description="""**Assistente Virtual do Departamento de Trânsito do Rio de Janeiro**
+
+Posso ajudar com informações sobre CNH, veículos, multas, taxas DUDA e serviços.
+
+📞 **Contato:** (21) 3460-4040 | 🌐 **Site:** [www.detran.rj.gov.br](https://www.detran.rj.gov.br)""",
+    examples=EXAMPLES,
+    theme=gr.themes.Soft(
+        primary_hue="blue",
+        secondary_hue="green",
+    ),
+    retry_btn=None,
+    undo_btn=None,
+)
+
+# ============================================================================
+# EXECUÇÃO
+# ============================================================================
+
+if __name__ == "__main__":
+    demo.launch()