add

app.py

@@ -194,9 +194,9 @@ print("hello")
 async def generate_viz(file: UploadFile = File(...), query: str = Form(...)):
     print("hello")
     try:
-        contents = await file.read()  
-        excel_file = io.BytesIO(contents)  
-        df = pd.read_excel(excel_file)  
+        contents = await file.read()
+        excel_file = io.BytesIO(contents)
+        df = pd.read_excel(excel_file)
 
         if query not in VALID_PLOTS:
             return JSONResponse(content={"error": f"Type de graphique invalide. Choisissez parmi : {', '.join(VALID_PLOTS)}"}, status_code=400)
@@ -230,11 +230,17 @@ plt.close()
         outputs = codegen_model.generate(**inputs, max_new_tokens=120, pad_token_id=codegen_tokenizer.eos_token_id)
         generated_code = codegen_tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
 
+        # 🔥 Nettoyage : couper si répétition du prompt
+        if "### Génère uniquement du code Python fonctionnel" in generated_code[1:]:
+            generated_code = generated_code.split("### Génère uniquement du code Python fonctionnel")[0]
+
+        # 🔥 Nettoyage : éviter doublon imports
         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]
 
-        print("🔹 Code généré par l'IA :\n", generated_code)
+        print("🔹 Code généré propre :\n", generated_code)
 
         try:
             compile(generated_code, "<string>", "exec")