GmailAiSort/main.py at main · Hugo-Galley/GmailAiSort · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
# Import des Librairies necessaire
import time
import requests

from google_auth_oauthlib.flow import InstalledAppFlow
from googleapiclient.discovery import build
import base64

# Url de connexion à l'API Google
SCOPES = ['https://www.googleapis.com/auth/gmail.modify']

# Authentification
flow = InstalledAppFlow.from_client_secrets_file('credentials.json', SCOPES)
creds = flow.run_local_server(port=0)
service = build('gmail', 'v1', credentials=creds)

# Recuperation de la liste des labels existant
listLabel = service.users().labels().list(userId='me').execute()

# Fonction pour récupérer le contenu du message
def get_email_content(msg):
    body = None
    if 'data' in msg['payload']['body']:
        body = msg['payload']['body']['data']
    elif 'parts' in msg['payload']:
        for part in msg['payload']['parts']:
            if part['mimeType'] == 'text/plain':
                body = part['body']['data']
                break
            elif part['mimeType'] == 'text/html':
                body = part['body']['data']
                break

    if body:
        return base64.urlsafe_b64decode(body).decode('utf-8')
    else:
        return "Aucun contenu disponible"

# Fonction pour attribuer un label a un email
def add_label(message_id, label_name):
    for label in listLabel['labels']:
        if label['name'] in label_name or label_name in label['name']:
            body = {
                'addLabelIds': label['id'],
            }
            service.users().messages().modify(userId="me",id=message_id, body=body).execute()
            print(f"Label {label['name']} ajouté à l'email {message_id}")
            return
    print("Aucun Label trouvé on skip")


# Fonction pour calculer le temps restant avant la fin du programme
def time_remaining(timeFor100Mails,nbrEmailScrap):
    mailRestant = nbrEmailScrap
    mailRestant -= 100
    tempsRestant = (mailRestant * timeFor100Mails) / 100
    if tempsRestant / 3600 > 1:
        tempsRestant /= 3600
        print(f"Temps restant {tempsRestant} H")
    elif tempsRestant / 60 > 1:
        tempsRestant /= 60
        print(f"Temps restant {tempsRestant} min")
    else:
        print(f"Temps restant {tempsRestant} sec")

# Fonction principale permettant de recuperer les mails afin de leur appliquer les modifs
def recup_email(nbrEmailScrap):
    nbrMail = 0
    page_token = None
    # Tant qu'il y'a des mails on recupere des messages
    while nbrMail <= nbrEmailScrap:
        results = service.users().messages().list(userId='me', pageToken=page_token).execute()
        messages = results.get('messages', [])
        timeBefore100Mails = time.time()
        # On parcours les messages
        for message in messages:
            # On recupere toute les infos necessaire sur le message
            msg = service.users().messages().get(userId='me', id=message['id']).execute()
            headers = msg['payload']['headers']
            from_header = next(header['value'] for header in headers if header['name'] == 'From')
            subject_header = next(header['value'] for header in headers if header['name'] == 'Subject')
            content = get_email_content(msg)
            # On demande le label au modele voulu puis on l'ajoute
            label = AskModelLabel(from_header,subject_header,content,"mistral-small")
            add_label(message['id'], label)
            nbrMail+=1
            print(f"Mail {nbrMail} sur {nbrEmailScrap} \n")
            timeAfter100Mails = time.time()
            timeFor100Mails =  timeAfter100Mails - timeBefore100Mails
        time_remaining(timeFor100Mails, nbrEmailScrap)


        # On passe a la page suivante jusque'a ce qu'il y'en ai plus
        page_token = results.get('nextPageToken')
        if not page_token:
            break


# Fonction pour demander au modele le label
def AskModelLabel(sender,subject,body,modelName):

    # Prompt envoyé au model pour retourner un des 5 labels
    question = f"Voici un e-mail avec l'émetteur : {sender}, le sujet : {subject}, et le contenu : {body}. Analyse-le et retourne uniquement l'une des 5 catégories suivantes : Travail, Personnel, SpamMail, Urgent, Promotion. Ne mets rien d'autre, aucune explication, juste le label."
    data = {
        "model": modelName,
        "messages": [{'role': 'user', "content": question}],
        'stream': False
    }
    url = "http://localhost:11434/api/chat"

    try:
        # Envoi de la requête POST
        response = requests.post(url, json=data)
        response.raise_for_status()
        response_json = response.json()

        # Récupération de la réponse du Modele
        ai_reply = response_json["message"]["content"]
        print(f"L'IA a trouvé le label : {ai_reply}")
        return ai_reply
    except requests.exceptions.RequestException as e:
        print("Une erreur a eu lieu:", e)

# Lancement du programme pour 100 mails
recup_email(100)