Update app.py

app.py

@@ -238,21 +238,21 @@ plt.close()
         print(outputs)
 
         # Nettoyage
-generated_code = codegen_tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
+        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()
+        # 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]
+        # 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)
+        # 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)