Create app.py

app.py

@@ -0,0 +1,204 @@
+import gradio as gr
+from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
+import re
+
+# ==================== CHARGEMENT DU MODÈLE ====================
+print("🔄 Chargement du modèle...")
+
+model = AutoModelForCausalLM.from_pretrained(
+    "padufour/mistral-accor-emails",
+    torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
+    device_map="auto",
+    low_cpu_mem_usage=True
+)
+
+tokenizer = AutoTokenizer.from_pretrained("padufour/mistral-accor-emails")
+
+print("✅ Modèle chargé et prêt !")
+
+# ==================== FONCTION DE GÉNÉRATION ====================
+def generate_email(email_type, client_name, company, objective, 
+                   segment, priority, history, style, language):
+    
+    # Construire le prompt
+    prompt = f"""### Instruction : Rédige un email du type "{email_type}" pour un client professionnel. L'email doit être complet, sans placeholder ni crochet [], et ne doit PAS mentionner le segment de clientèle dans le contenu.
+
+### Input :
+Client Information:
+* Name: {client_name}
+* Company: {company}
+* Objective: {objective}
+* Priority: {priority}
+* History: {history}
+* Style: {style}
+* Language: {language}
+
+Important: Ne pas inclure de placeholders [], ne pas mentionner le segment, ne pas mettre les coordonnées du client.
+
+### Réponse :"""
+    
+    # Générer
+    inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
+    
+    with torch.no_grad():
+        outputs = model.generate(
+            **inputs,
+            max_new_tokens=800,
+            temperature=0.7,
+            top_p=0.9,
+            top_k=50,
+            repetition_penalty=1.1,
+            do_sample=True,
+            pad_token_id=tokenizer.pad_token_id,
+            eos_token_id=tokenizer.eos_token_id
+        )
+    
+    email = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    
+    # Extraire la réponse
+    if "### Réponse :" in email:
+        email = email.split("### Réponse :")[1].strip()
+    
+    # Nettoyer
+    email = re.sub(r'\[.*?\]', '', email)
+    email = re.sub(r'\b(MICE|AHBO|leisure group)\b', '', email, flags=re.IGNORECASE)
+    email = re.sub(r'at\s+[\w\.-]+@[\w\.-]+\.\w+', '', email)
+    email = re.sub(r'or\s+\+?\d[\d\s\(\)\-\.]+', '', email)
+    
+    # Ajouter signature
+    if "ACCOR Team" not in email and "Accor Team" not in email:
+        email += "\n\nBest regards,\nThe ACCOR Team"
+    
+    # Nettoyer espaces
+    email = re.sub(r'\n\s*\n\s*\n', '\n\n', email)
+    email = re.sub(r'  +', ' ', email)
+    
+    return email.strip()
+
+# ==================== INTERFACE GRADIO ====================
+with gr.Blocks(theme=gr.themes.Soft(), title="ACCOR Email Generator") as demo:
+    
+    gr.Markdown(
+        """
+        # 🎯 ACCOR Email Generator
+        **Powered by Mistral-7B Fine-tuned**
+        
+        Generate professional marketing emails for ACCOR clients in seconds.
+        """
+    )
+    
+    with gr.Row():
+        with gr.Column(scale=1):
+            gr.Markdown("### 📋 Email Configuration")
+            
+            email_type = gr.Dropdown(
+                choices=["Newsletter", "New offers", "theme", "solution", "Hotel opening"],
+                value="Newsletter",
+                label="Email Type *",
+                info="Type of marketing campaign"
+            )
+            
+            segment = gr.Dropdown(
+                choices=["MICE", "AHBO", "leisure group"],
+                value="MICE",
+                label="Segment",
+                info="Customer segment (used internally, not displayed)"
+            )
+            
+            gr.Markdown("### 👤 Client Information")
+            
+            client_name = gr.Textbox(
+                label="Client Name *",
+                placeholder="Marie Dupont",
+                value="Marie Dupont"
+            )
+            
+            company = gr.Textbox(
+                label="Company *",
+                placeholder="Event Solutions SA",
+                value="Event Solutions SA"
+            )
+            
+            objective = gr.Textbox(
+                label="Objective *",
+                placeholder="Promote winter events",
+                value="Promote winter events",
+                lines=2
+            )
+            
+            gr.Markdown("### ⚙️ Advanced Options")
+            
+            priority = gr.Radio(
+                choices=["High", "Medium", "Low"],
+                value="High",
+                label="Priority"
+            )
+            
+            history = gr.Textbox(
+                label="Client History",
+                placeholder="New customer",
+                value="New customer"
+            )
+            
+            style = gr.Textbox(
+                label="Style",
+                placeholder="Professional and engaging",
+                value="Professional and engaging"
+            )
+            
+            language = gr.Radio(
+                choices=["ENGLISH", "FRENCH"],
+                value="ENGLISH",
+                label="Language"
+            )
+            
+            generate_btn = gr.Button("🚀 Generate Email", variant="primary", size="lg")
+        
+        with gr.Column(scale=1):
+            gr.Markdown("### 📧 Generated Email")
+            
+            output = gr.Textbox(
+                label="Result",
+                placeholder="Your generated email will appear here...",
+                lines=25,
+                show_copy_button=True
+            )
+            
+            gr.Markdown(
+                """
+                ---
+                **Model:** [padufour/mistral-accor-emails](https://huggingface.co/padufour/mistral-accor-emails)
+                
+                **Tips for best results:**
+                - Provide detailed objectives
+                - Specify client history for personalization
+                - Use descriptive style preferences
+                """
+            )
+    
+    # Exemples
+    gr.Markdown("### 💡 Quick Examples")
+    
+    gr.Examples(
+        examples=[
+            ["Newsletter", "Marie Dupont", "Event Solutions SA", "Promote winter events", "MICE", "High", "Long-term client", "Professional and warm", "ENGLISH"],
+            ["New offers", "Pierre Martin", "Travel Agency Plus", "Showcase summer destinations", "leisure group", "Medium", "Regular customer", "Friendly and inspiring", "ENGLISH"],
+            ["theme", "Sophie Laurent", "Corporate Events Ltd", "Present emerging trends in 2025", "MICE", "High", "Industry leader", "Modern and innovative", "ENGLISH"],
+            ["solution", "Thomas Bernard", "Business Travel Corp", "Promote loyalty program benefits", "AHBO", "High", "Frequent traveler", "Professional and convincing", "ENGLISH"],
+        ],
+        inputs=[email_type, client_name, company, objective, segment, priority, history, style, language],
+        outputs=output,
+        fn=generate_email,
+        cache_examples=False,
+    )
+    
+    # Event
+    generate_btn.click(
+        fn=generate_email,
+        inputs=[email_type, client_name, company, objective, segment, priority, history, style, language],
+        outputs=output
+    )
+
+# Lancer
+demo.launch()