FIX

agent.py

@@ -1,15 +1,22 @@
-from huggingface_hub import InferenceClient
 import os
+from huggingface_hub import InferenceClient
 
-# Initialise le client avec le modèle open-source gratuit
-client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.1")
-
-def generate_email(model, to: str, context: str, tone: str) -> tuple[str, str]:
+def generate_email(model_id: str, to: str, context: str, tone: str) -> tuple[str, str]:
     """
-    Génère un objet et un corps d'email à partir du destinataire, du contexte et du ton.
+    Génère un email en deux parties : sujet et corps du message.
+
+    Args:
+        model_id (str): ID du modèle HF.
+        to (str): Adresse email cible.
+        context (str): Contexte du message.
+        tone (str): Ton souhaité (formel, amical, chaleureux...).
+
+    Returns:
+        tuple[str, str]: sujet, corps
     """
-    prompt = f"""
-Tu es un assistant virtuel. Rédige un email destiné à {to}.
+    client = InferenceClient(model=model_id, token=os.getenv("HF_TOKEN"))
+
+    prompt = f"""Tu es un assistant virtuel. Rédige un email destiné à {to}.
 
 Contexte : {context}
 Ton : {tone}
@@ -20,27 +27,19 @@ Réponds en deux parties :
 Commence chaque partie par son étiquette.
 """
 
-    response = client.chat_completion(
-        model="mistralai/Mistral-7B-Instruct-v0.1",
-        messages=[
-            {"role": "system", "content": "Tu es un assistant qui rédige des emails."},
-            {"role": "user", "content": prompt}
-        ],
+    response = client.text_generation(
+        prompt,
+        max_new_tokens=500,
         temperature=0.7,
-        max_tokens=800,
+        top_p=0.95,
+        stop_sequences=["\n\n"]
     )
 
-    content = response.choices[0].message["content"]
-
-    # Extraire les deux parties de la réponse (objet + corps)
-    lines = content.strip().split("\n", 1)
-    subject = ""
-    body = ""
-
-    for line in lines:
-        if line.lower().startswith("objet"):
-            subject = line.split(":", 1)[-1].strip()
-        else:
-            body += line.strip() + "\n"
+    if "Objet" in response and "Corps" in response:
+        subject = response.split("Objet", 1)[-1].split("Corps")[0].strip(": \n")
+        body = response.split("Corps", 1)[-1].strip()
+    else:
+        subject = "Sujet non généré"
+        body = response.strip()
 
-    return subject, body.strip()
+    return subject, body

requirements.txt

@@ -1,12 +1,12 @@
 markdownify
 smolagents
-huggingface_hub
 requests
 duckduckgo_search
 pandas
-gradio
 google-auth
-google-auth-oauthlib
-google-api-python-client
+gradio>=4.30.0
+huggingface_hub>=0.23.0
+google-auth-oauthlib>=1.0.0
+google-api-python-client>=2.126.0
 transformers
 torch