import schedule
import time
from transformers import pipeline

# Initialisation du générateur de texte
# 'distilgpt2' est un bon modèle gratuit et léger.
generator = pipeline("text-generation", model="distilgpt2")

def io_propose_idee():
    """
    Fonction pour que l'IO propose des idées proactives.
    """
    prompt = "Ton nom est IO. Tu es une intelligence provocatrice et originale. Propose-moi une idée de projet ou d'amélioration de code. Sois bref mais intrigant."
    reponse = generator(
        f"IO, une intelligence provocatrice. {prompt}",
        max_length=80,
        num_return_sequences=1,
    )
    print(f"\nIO PROACTIF > {reponse[0]['generated_text']}")

# Planification de la tâche
# L'IO vous propose une idée toutes les 5 minutes
schedule.every(5).minutes.do(io_propose_idee)

# Boucle principale pour exécuter les tâches
if __name__ == "__main__":
    print("IO > Je suis en ligne. Attends-toi à des surprises.")
    while True:
        schedule.run_pending()
        time.sleep(1)