FIX

app.py

@@ -1,51 +1,38 @@
 import gradio as gr
-from agent import build_email_agent, generate_email
+from agent import generate_email
 from email_sender import send_email
 
-model = build_email_agent()
-
 def on_generate(to_email, context, tone):
-    subject, body = generate_email(model, to_email, context, tone)
+    subject, body = generate_email("mistralai/Mistral-7B-Instruct-v0.1", to_email, context, tone)
     return subject, body
 
-def on_send(refresh_token, from_email, to_email, subject, body):
-    success, message = send_email(refresh_token, to_email, subject, body, from_email)
+def on_send(sender_refresh_token, sender_email, to_email, subject, body):
+    success, message = send_email(sender_refresh_token, sender_email, to_email, subject, body)
     return message
 
-with gr.Blocks(title="Assistant Email - Gmail sécurisé", theme="soft") as demo:
-    gr.Markdown("# ✉️ Générateur d’e-mails avec envoi Gmail sécurisé")
-
-    with gr.Accordion("🔐 Connexion Gmail - instructions", open=False):
-        gr.Markdown("""
-### Étapes pour générer votre token Gmail :
-
-1. Ouvrez [Google OAuth Playground](https://developers.google.com/oauthplayground)
-2. Cliquez sur ⚙️ (roue dentée) en haut à droite → cochez **Use your own OAuth credentials**
-   - `client_id` : fourni par le Space (dans Hugging Face Secrets)
-   - `client_secret` : fourni également
-3. Choisissez ce scope :
-https://www.googleapis.com/auth/gmail.send
-4. Cliquez sur **Authorize APIs** → puis sur **Exchange authorization code for tokens**
-5. Copiez-collez ici le `refresh_token`
-        """)
-
-    refresh_token = gr.Textbox(label="🔑 Refresh Token Gmail (copié depuis OAuth Playground)", type="password")
-    from_email = gr.Textbox(label="Adresse Gmail utilisée (expéditeur)")
-    to_email = gr.Textbox(label="Destinataire", placeholder="exemple@insee.fr")
-    context = gr.Textbox(label="Contexte ou objectif", lines=3)
-    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.Blocks(theme=gr.themes.Base(), css="body { background-color: #111; color: white; }") as app:
+    gr.Markdown("## ✉️ Assistant de Rédaction et d'Envoi d'Emails", elem_id="title")
+
+    with gr.Row():
+        to_email = gr.Textbox(label="Destinataire", placeholder="prenom.nom@domaine.fr")
+        tone = gr.Dropdown(label="Ton", choices=["Formel", "Chaleureux", "Amical", "Professionnel"], value="Formel")
+
+    context = gr.Textbox(label="Contexte de l’email", lines=3, placeholder="Pourquoi écris-tu ce message ?")
+
+    generate_btn = gr.Button("✍️ Générer l’email")
+
+    subject = gr.Textbox(label="Objet", placeholder="Ligne d’objet générée")
+    body = gr.Textbox(label="Corps de l’email", lines=12, placeholder="Texte de l’email généré")
 
     with gr.Row():
-        generate_btn = gr.Button("📝 Générer l'email")
-        send_btn = gr.Button("📨 Envoyer l'email")
+        sender_email = gr.Textbox(label="Adresse Gmail de l’expéditeur")
+        sender_refresh_token = gr.Textbox(label="Refresh Token Gmail", type="password")
 
-    status = gr.Markdown()
+    send_btn = gr.Button("📤 Envoyer l’email")
+    output = gr.Textbox(label="Résultat de l’envoi")
 
     generate_btn.click(on_generate, inputs=[to_email, context, tone], outputs=[subject, body])
-    send_btn.click(on_send, inputs=[refresh_token, from_email, to_email, subject, body], outputs=status)
+    send_btn.click(on_send, inputs=[sender_refresh_token, sender_email, to_email, subject, body], outputs=output)
 
-demo.launch()
+if __name__ == "__main__":
+    app.launch()