Update src/streamlit_app.py

src/streamlit_app.py

@@ -5,8 +5,8 @@ from urllib.parse import urlparse
 import io
 
 # --- CONFIGURAÇÕES ---
-# ⚠️ IMPORTANTE: Substitua "SEU_USUARIO/NOME_DO_SEU_MODELO" pelo seu ID real no Hugging Face
-MODEL_ID = "SEU_USUARIO/NOME_DO_SEU_MODELO"
+# Seu ID do modelo (Correto conforme você enviou)
+MODEL_ID = "asilvamaia/ident_br"
 
 st.set_page_config(page_title="Validador .BR", page_icon="🇧🇷")
 
@@ -49,7 +49,7 @@ def load_model():
     try:
         tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
         model = AutoModelForSequenceClassification.from_pretrained(MODEL_ID)
-        # Força uso de CPU no Space gratuito
+        # Força uso de CPU no Space gratuito para evitar erros
         model.to("cpu")
         model.eval()
         return tokenizer, model
@@ -81,6 +81,56 @@ if uploaded_file and tokenizer:
         for i, linha in enumerate(linhas):
             original = linha.strip()
             
-            # --- CORREÇÃO DA LINHA PROBLEMÁTICA ---
-            # Verifica se a linha é vazia ou é metadado 
-            if not original or original.startswith("")
+            # --- CORREÇÃO DA LINHA QUE ESTAVA COM ERRO ---
+            # Verifica se a linha é vazia ou se começa com metadados [source: ...]
+            if not original or original.startswith("[source:"): 
+                continue
+
+            # Aplica limpeza
+            limpo = limpar_entrada(original)
+            
+            # Inferência (Classificação)
+            inputs = tokenizer(limpo, return_tensors="pt", truncation=True, max_length=128)
+            with torch.no_grad():
+                outputs = model(**inputs)
+                pred = torch.argmax(outputs.logits, dim=1).item()
+            
+            # REGRA DE APROVAÇÃO:
+            # 1. Modelo diz que é Classe 1 (BR)
+            # 2. Tem ponto na string
+            # 3. Não tem @
+            if pred == 1 and "." in limpo and "@" not in limpo:
+                validos.append(limpo)
+            else:
+                rejeitados.append(original)
+            
+            # Atualiza barra de progresso a cada 10 itens
+            if i % 10 == 0:
+                progress_bar.progress(min((i + 1) / total_linhas, 1.0))
+        
+        progress_bar.progress(1.0)
+        
+        # --- DEDUPLICAÇÃO (Remove repetidos) ---
+        validos_unicos = sorted(list(set(validos)))
+        rejeitados_unicos = sorted(list(set(rejeitados)))
+        
+        st.success("Processamento Concluído!")
+        
+        # Exibe métricas
+        col1, col2 = st.columns(2)
+        with col1:
+            st.metric("✅ Aprovados (Únicos)", len(validos_unicos), help="Domínios .br válidos")
+        with col2:
+            st.metric("🔴 Rejeitados (Únicos)", len(rejeitados_unicos), help="Itens descartados")
+            
+        # Botão de Download
+        st.download_button(
+            label="⬇️ Baixar Lista Limpa (.txt)",
+            data="\n".join(validos_unicos),
+            file_name="dominios_limpos.txt",
+            mime="text/plain"
+        )
+        
+        # Mostra amostra
+        with st.expander("Ver amostra dos resultados"):
+            st.write(validos_unicos[:50])