Essai sup

agent.py

@@ -1,19 +1,43 @@
-# agent.py
-from transformers import pipeline
+from smolagent import SmolAgent, OpenAIProvider
+from smolagent.schema import ChatPrompt, Prompt
 
-def build_email_agent():
-    # Charge un modèle de génération de texte léger (ex: distilgpt2, ou un modèle Hugging Face)
-    # Remplace par un modèle plus puissant si tu veux (et dispo)
-    generator = pipeline("text-generation", model="gpt2")
-    return generator
+# Crée le système de prompt soigné pour générer email professionnel
+system_prompt = Prompt(
+    role="system",
+    content=(
+        "Tu es un assistant IA qui génère des emails professionnels adaptés au contexte, "
+        "au destinataire et au ton donné. "
+        "Rédige d’abord un objet court et pertinent, puis le corps du mail complet."
+    ),
+)
+
+# Prompt utilisateur (input)
+user_prompt = Prompt(
+    role="user",
+    content=(
+        "Destinataire : {to_email}\n"
+        "Contexte : {context}\n"
+        "Ton souhaité : {tone}\n\n"
+        "Génère un objet et un corps d'email."
+    ),
+)
 
-def generate_email(agent, prompt):
-    # Génère un texte à partir du prompt
-    outputs = agent(prompt, max_length=300, num_return_sequences=1)
-    text = outputs[0]['generated_text']
+def build_email_agent():
+    provider = OpenAIProvider(model="gpt-4o-mini")  # ou autre modèle disponible
+    prompt = ChatPrompt(prompts=[system_prompt, user_prompt])
+    agent = SmolAgent(provider=provider, prompt=prompt)
+    return agent
 
-    # Pour simplifier, on peut séparer l’objet (première ligne) et le corps (le reste)
-    lines = text.strip().split('\n')
-    subject = lines[0][:78] if lines else "Objet automatique"
-    body = "\n".join(lines[1:]).strip() if len(lines) > 1 else text
+def generate_email(agent, to_email, context, tone):
+    # Formate le prompt avec les valeurs utilisateur
+    inputs = {
+        "to_email": to_email,
+        "context": context,
+        "tone": tone,
+    }
+    response = agent.chat(inputs)
+    # La réponse devrait contenir objet + corps, on sépare à la première ligne vide
+    parts = response.strip().split('\n\n', 1)
+    subject = parts[0].strip() if parts else "Objet automatique"
+    body = parts[1].strip() if len(parts) > 1 else response.strip()
     return subject, body

app.py

@@ -1,39 +1,35 @@
-# app.py
 import gradio as gr
 from agent import build_email_agent, generate_email
 from email_sender import send_email
 
 agent = build_email_agent()
 
-def handle_generate(to_email, context, tone):
-    prompt = f"Écris un email professionnel au destinataire {to_email}.\n" \
-             f"Contexte : {context}\n" \
-             f"Ton : {tone}\n\nObjet et corps du mail :"
-    subject, body = generate_email(agent, prompt)
+def on_generate(to_email, context, tone):
+    subject, body = generate_email(agent, to_email, context, tone)
     return subject, body
 
-def handle_send(to_email, subject, body):
-    result = send_email(to_email, subject, body)
-    return result
+def on_send(to_email, subject, body):
+    success, message = send_email(to_email, subject, body)
+    return message
 
-with gr.Blocks(title="Assistant Email Professionnel") as demo:
-    gr.Markdown("# ✉️ Assistant Email Professionnel")
+with gr.Blocks(title="Assistant Email Pro avec SmolAgent") as demo:
+    gr.Markdown("# ✉️ Assistant Email Professionnel SmolAgent")
 
-    to_email = gr.Textbox(label="Destinataire", placeholder="ex: nom@entreprise.com")
-    context = gr.Textbox(label="Contexte ou objectif de l'email", lines=4,
-                         placeholder="Par exemple : demander un rendez-vous, relancer un client...")
-    tone = gr.Dropdown(choices=["Formel", "Chaleureux", "Direct", "Poli"], value="Formel", label="Ton souhaité")
+    to_email = gr.Textbox(label="Destinataire", placeholder="exemple@exemple.com")
+    context = gr.Textbox(label="Contexte ou objectif", lines=4,
+                         placeholder="Demander un rendez-vous, relancer un client...")
+    tone = gr.Dropdown(choices=["Formel", "Chaleureux", "Direct", "Poli"], value="Formel", label="Ton")
 
     subject = gr.Textbox(label="Objet de l'email")
     body = gr.Textbox(label="Corps de l'email", lines=10)
 
     with gr.Row():
-        generate_btn = gr.Button("📝 Générer le mail")
-        send_btn = gr.Button("📨 Envoyer le mail")
+        generate_btn = gr.Button("📝 Générer l'email")
+        send_btn = gr.Button("📨 Envoyer l'email")
 
     status = gr.Markdown()
 
-    generate_btn.click(handle_generate, inputs=[to_email, context, tone], outputs=[subject, body])
-    send_btn.click(handle_send, inputs=[to_email, subject, body], outputs=status)
+    generate_btn.click(on_generate, inputs=[to_email, context, tone], outputs=[subject, body])
+    send_btn.click(on_send, inputs=[to_email, subject, body], outputs=status)
 
 demo.launch()

email_sender.py

@@ -1,4 +1,3 @@
-# email_sender.py
 import os
 import base64
 import pickle
@@ -10,7 +9,6 @@ from googleapiclient.discovery import build
 TOKEN_PATH = "token.pkl"
 CREDENTIALS_PATH = "client_secret.json"
 
-# Écrire client_secret.json depuis variable d'environnement (Hugging Face Secret)
 client_secret_content = os.getenv("CLIENT_SECRET_JSON")
 if client_secret_content and not os.path.exists(CREDENTIALS_PATH):
     with open(CREDENTIALS_PATH, "w") as f:
@@ -42,6 +40,6 @@ def send_email(to_email, subject, body):
     send_message = {"raw": raw_message}
     try:
         sent = service.users().messages().send(userId="me", body=send_message).execute()
-        return f"✅ Email envoyé à {to_email} (ID: {sent['id']})"
+        return True, f"✅ Email envoyé à {to_email} (ID: {sent['id']})"
     except Exception as e:
-        return f"❌ Échec de l'envoi : {e}"
+        return False, f"❌ Échec de l'envoi : {e}"