import os
import fitz  # PyMuPDF  instalar o pip install pymupdf
#difgitar no terminal portifolio/port28_AI/curriculos criar as pastas curriculos e curriculos_filtrados
# Lista de palavras-chave para filtragem
keywords = ["Grajaú", "js", "css", "html"]

# Pasta onde os currículos estão salvos
input_folder = "curriculos/"
output_folder = "curriculos_filtrados/"

# Criar pasta de saída se não existir
os.makedirs(output_folder, exist_ok=True)

# Função para extrair texto do PDF
def extract_text_from_pdf(pdf_path):
    doc = fitz.open(pdf_path)
    text = ""
    for page in doc:
        text += page.get_text("text")
    return text.lower()

# Processar currículos
for file in os.listdir(input_folder):
    if file.endswith(".pdf"):
        pdf_path = os.path.join(input_folder, file)
        text = extract_text_from_pdf(pdf_path)
        
        # Verifica se pelo menos uma palavra-chave está no texto
        if any(keyword.lower() in text for keyword in keywords):
            os.rename(pdf_path, os.path.join(output_folder, file))
            print(f"✔ {file} movido para {output_folder}")

print("Processo concluído!")