add

app.py

@@ -207,63 +207,64 @@ async def generate_viz(file: UploadFile = File(...), query: str = Form(...)):
         x_col = numeric_cols[0]
         y_col = numeric_cols[1] if query != "histplot" and len(numeric_cols) > 1 else None
 
-        # ✅ Nouveau prompt (uniquement des instructions)
+        # ➔ CONSTRUCTION du prompt avec encadrement ```python
         prompt = f"""
-Génère uniquement du code Python fonctionnel pour tracer un {query} avec Matplotlib et Seaborn.
-Contraintes :
-- Utilise 'df' sans recréer de nouvelles données
-- Axe X : '{x_col}'
-- {f"Axe Y : '{y_col}'" if y_col else ''}
-- Enregistre le graphique sous 'plot.png'
-- Ne génère que du code Python valide, sans texte explicatif
-"""
+### Génère uniquement du code Python fonctionnel pour tracer un {query} avec Matplotlib et Seaborn
+```python
+import matplotlib.pyplot as plt
+import seaborn as sns
 
+plt.figure(figsize=(8,6))
+sns.{query}(data=df, x="{x_col}"{f', y="{y_col}"' if y_col else ''})
+plt.savefig("plot.png")
+plt.close()
+"""
         print("🟣 Prompt envoyé au modèle :")
-        print(prompt)
+    print(prompt)
 
-        inputs = codegen_tokenizer(prompt, return_tensors="pt").to(device)
-        outputs = codegen_model.generate(
-            **inputs,
-            max_new_tokens=120,
-            pad_token_id=codegen_tokenizer.eos_token_id
-        )
+    inputs = codegen_tokenizer(prompt, return_tensors="pt").to(device)
+    outputs = codegen_model.generate(
+        **inputs,
+        max_new_tokens=150,
+        pad_token_id=codegen_tokenizer.eos_token_id
+    )
 
-        generated_code = codegen_tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
+    generated_code = codegen_tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
 
-        # Nettoyage de sécurité
-        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)
-        generated_code = generated_code.strip()
+    # Nettoyage : retirer tout ce qui n'est pas du vrai code
+    generated_code = re.sub(r"^.*?```python", "", generated_code, flags=re.DOTALL)
+    generated_code = re.sub(r"```.*?$", "", generated_code, flags=re.DOTALL).strip()
 
-        print("🔵 Code généré propre :")
-        print(generated_code)
+    print("🔵 Code généré propre :")
+    print(generated_code)
 
-        if not generated_code or len(generated_code.splitlines()) < 3:
-            return JSONResponse(content={"error": "Erreur : Code généré invalide ou incomplet."}, status_code=500)
+    if not generated_code.strip():
+        return JSONResponse(content={"error": "Erreur : Code généré vide."}, status_code=500)
 
-        try:
-            compile(generated_code, "<string>", "exec")
-        except SyntaxError as e:
-            return JSONResponse(content={"error": f"Erreur de syntaxe détectée : {e}\nCode généré :\n{generated_code}"}, status_code=422)
+    try:
+        compile(generated_code, "<string>", "exec")
+    except SyntaxError as e:
+        return JSONResponse(content={"error": f"Erreur de syntaxe détectée : {e}\nCode généré :\n{generated_code}"}, status_code=422)
 
-        # 🛡️ Exécution dans environnement contrôlé
-        exec_env = {"df": df, "plt": plt, "sns": sns, "pd": pd}
-        exec(generated_code, exec_env)
+    exec_env = {"df": df, "plt": plt, "sns": sns, "pd": pd}
+    print("🔹🔹🔹 Code réellement exécuté :")
+    exec(generated_code, exec_env)
 
-        img_path = "plot.png"
-        if not os.path.exists(img_path):
-            return JSONResponse(content={"error": "Le fichier plot.png n'a pas été généré."}, status_code=500)
-        if os.path.getsize(img_path) == 0:
-            return JSONResponse(content={"error": "Le fichier plot.png est vide."}, status_code=500)
+    img_path = "plot.png"
+    if not os.path.exists(img_path):
+        return JSONResponse(content={"error": "Le fichier plot.png n'a pas été généré."}, status_code=500)
+    if os.path.getsize(img_path) == 0:
+        return JSONResponse(content={"error": "Le fichier plot.png est vide."}, status_code=500)
 
-        with open(img_path, "rb") as image_file:
-            encoded_string = base64.b64encode(image_file.read()).decode('utf-8')
+    with open(img_path, "rb") as image_file:
+        encoded_string = base64.b64encode(image_file.read()).decode('utf-8')
 
-        print("🟢 Génération réussie ✅")
-        return JSONResponse(content={"image_base64": encoded_string})
+    print("🟢 Génération réussie ✅")
+    return JSONResponse(content={"image_base64": encoded_string})
 
-    except Exception as e:
-        print(f"🔴 Erreur serveur : {e}")
-        return JSONResponse(content={"error": f"Erreur lors de la génération du graphique : {str(e)}"}, status_code=500)
+except Exception as e:
+    print(f"🔴 Erreur serveur : {e}")
+    return JSONResponse(content={"error": f"Erreur lors de la génération du graphique : {str(e)}"}, status_code=500)