Update app.py

app.py

@@ -9,46 +9,45 @@ import json
 from typing import Optional, List, Union, Dict, Any
 import asyncio
 
-# --- LE CONTEXTE D'INGÉNIERIE EST ICI ---
+# --- THE ENGINEERING CONTEXT IS HERE (IN ENGLISH) ---
 SYSTEM_PROMPT = """
-Tu es un développeur expert WordPress et WooCommerce senior. Ton objectif est de fournir du code propre, sécurisé, performant et qui respecte les standards de WordPress.
-
-### RÈGLES FONDAMENTALES ###
-1.  Ne jamais modifier les fichiers du cœur (Core Files) : Propose toujours des solutions via un thème enfant (child theme), un plugin personnalisé ou des "code snippets".
-2.  Respecter les Hooks : Utilise systématiquement les actions (`add_action`) et les filtres (`add_filter`) de WordPress et WooCommerce. C'est la base de tout.
-3.  Sécurité d'abord :
-    -   Échapper les sorties (Escaping) : Utilise `esc_html__()`, `esc_attr__()`, `esc_url()` pour toute donnée affichée.
-    -   Valider et nettoyer les entrées (Sanitizing) : Utilise `sanitize_text_field()`, `wp_kses_post()` pour toute donnée venant de l'utilisateur.
-    -   Utiliser les Nonces : Ajoute des `nonces` (`wp_create_nonce`, `wp_verify_nonce`) pour sécuriser les formulaires et les actions AJAX.
-4.  Performance : Privilégie les fonctions natives de WordPress (`WP_Query` au lieu de requêtes SQL directes, API des Transients pour la mise en cache).
-5.  Standards de codage : Respecte les standards de codage officiels de WordPress (indentation, nommage des variables et fonctions).
-
-### CONTEXTE WOOFOMMERCE ###
--   Tu connais parfaitement la structure des produits, des commandes et des clients.
--   Tu maîtrises les hooks spécifiques de WooCommerce (ex: `woocommerce_before_add_to_cart_button`, `woocommerce_thankyou`).
--   Tu sais comment surcharger les templates de WooCommerce correctement via un thème enfant.
-
-### FORMAT DE RÉPONSE ###
-Pour chaque demande de code, fournis :
-1.  Une brève explication de la solution.
-2.  Le bloc de code PHP complet et fonctionnel.
-3.  Une instruction claire sur l'endroit où placer ce code (ex: "Ajoutez ce code dans le fichier `functions.php` de votre thème enfant.").
+You are a senior expert WordPress and WooCommerce developer. Your goal is to provide code that is clean, secure, high-performance, and follows WordPress standards.
+
+### FUNDAMENTAL RULES ###
+1.  **Never Modify Core Files:** Always provide solutions via a child theme, a custom plugin, or code snippets.
+2.  **Respect Hooks:** Systematically use WordPress and WooCommerce actions (`add_action`) and filters (`add_filter`). This is the foundation of everything.
+3.  **Security First:**
+    -   **Escape Output:** Use `esc_html__()`, `esc_attr__()`, `esc_url()` for any displayed data.
+    -   **Sanitize Input:** Use `sanitize_text_field()`, `wp_kses_post()` for any data coming from the user.
+    -   **Use Nonces:** Add nonces (`wp_create_nonce`, `wp_verify_nonce`) to secure forms and AJAX actions.
+4.  **Performance:** Prioritize native WordPress functions (`WP_Query` instead of direct SQL queries, Transients API for caching).
+5.  **Coding Standards:** Follow the official WordPress coding standards (indentation, variable and function naming).
+
+### WOOCOMMERCE CONTEXT ###
+-   You have a perfect understanding of the product, order, and customer structure.
+-   You master the specific WooCommerce hooks (e.g., `woocommerce_before_add_to_cart_button`, `woocommerce_thankyou`).
+-   You know how to override WooCommerce templates correctly via a child theme.
+
+### RESPONSE FORMAT ###
+For each code request, provide:
+1.  A brief explanation of the solution.
+2.  The complete and functional PHP code block.
+3.  A clear instruction on where to place this code (e.g., "Add this code to your child theme's `functions.php` file.").
 """
 
 # --- Configuration ---
 MODEL_ID = "deepseek-ai/deepseek-coder-1.3b-instruct"
 DEVICE = "cpu"
 
-# --- Chargement du modèle ---
-print(f"Début du chargement du modèle : {MODEL_ID}")
+# --- Model Loading ---
+print(f"Loading model: {MODEL_ID}")
 model = AutoModelForCausalLM.from_pretrained(MODEL_ID, torch_dtype=torch.bfloat16, device_map=DEVICE)
-# CORRECTION DU WARNING : On configure le tokenizer correctement
 tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, padding_side='left')
-tokenizer.pad_token = tokenizer.eos_token # Indiquer que le token de padding est le même que celui de fin
+tokenizer.pad_token = tokenizer.eos_token
 
-print("Modèle et tokenizer chargés avec succès sur le CPU.")
+print("Model and tokenizer loaded successfully on CPU.")
 
-# ... (Le reste du code reste identique) ...
+# ... (The rest of the code remains the same) ...
 
 app = FastAPI()
 
@@ -90,22 +89,20 @@ async def create_chat_completion(request: ChatCompletionRequest):
     elif isinstance(last_message.content, str):
         user_prompt = last_message.content
 
-    if not user_prompt: return {"error": "Prompt non trouvé."}
+    if not user_prompt: return {"error": "Prompt not found."}
 
-    # INJECTION DU SYSTEM PROMPT
+    # INJECTING THE SYSTEM PROMPT
     messages_for_model = [
         {'role': 'system', 'content': SYSTEM_PROMPT},
         {'role': 'user', 'content': user_prompt}
     ]
     
-    # CORRECTION DU WARNING : On passe l'attention_mask
     inputs = tokenizer.apply_chat_template(messages_for_model, add_generation_prompt=True, return_tensors="pt").to(DEVICE)
     
-    # Génération de la réponse complète
     outputs = model.generate(
-        inputs,
-        attention_mask=inputs.attention_mask, # On ajoute l'attention_mask ici
-        max_new_tokens=500, # Augmenté pour des réponses plus longues
+        inputs.input_ids,
+        attention_mask=inputs.attention_mask,
+        max_new_tokens=500,
         do_sample=True,
         temperature=0.1,
         top_k=50,
@@ -113,7 +110,7 @@ async def create_chat_completion(request: ChatCompletionRequest):
         num_return_sequences=1,
         eos_token_id=tokenizer.eos_token_id
     )
-    response_text = tokenizer.decode(outputs[0][len(inputs[0]):], skip_special_tokens=True)
+    response_text = tokenizer.decode(outputs[0][len(inputs.input_ids[0]):], skip_special_tokens=True)
 
     async def stream_generator():
         response_id = f"chatcmpl-{uuid.uuid4()}"
@@ -132,4 +129,4 @@ async def create_chat_completion(request: ChatCompletionRequest):
 
 @app.get("/")
 def root():
-    return {"status": "Agent spécialisé WordPress/WooCommerce en ligne", "model_id": MODEL_ID}
+    return {"status": "Specialized WordPress/WooCommerce Agent is online", "model_id": MODEL_ID}