IO-AI / Io-main.py
Mauriciotuks's picture
Io-main.py
5f19fee verified
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)