add

app.py

@@ -237,15 +237,22 @@ plt.close()
         print("🟠 Raw output du modèle :")
         print(outputs)
 
-        generated_code = codegen_tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
-
         # Nettoyage
-        generated_code = re.sub(r"(import matplotlib.pyplot as plt\nimport seaborn as sns\n)+", "import matplotlib.pyplot as plt\nimport seaborn as sns\n", generated_code)
-        if generated_code.strip().endswith("sns."):
-            generated_code = generated_code.rsplit("\n", 1)[0]
+generated_code = codegen_tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
+
+# Si le modèle recopie le prompt, on garde seulement avant la répétition
+if "### Génère uniquement du code Python fonctionnel" in generated_code:
+    generated_code = generated_code.split("### Génère uniquement du code Python fonctionnel")[0].strip()
+
+# Protection si jamais il reste un morceau incomplet
+if generated_code.strip().endswith("import") or generated_code.strip().endswith("sns."):
+    generated_code = generated_code.rsplit("\n", 1)[0]
+
+# Nettoyage doublon imports matplotlib/seaborn
+generated_code = re.sub(r"(import matplotlib\.pyplot as plt\nimport seaborn as sns\n)+", "import matplotlib.pyplot as plt\nimport seaborn as sns\n", generated_code)
 
-        print("🔵 Code généré propre :")
-        print(generated_code)
+print("🔵 Code généré propre :")
+print(generated_code)
 
         if not generated_code.strip():
             return JSONResponse(content={"error": "Erreur : Code généré vide."}, status_code=500)