adde

app.py

@@ -192,15 +192,12 @@ VALID_PLOTS = {"histplot", "scatterplot", "barplot", "lineplot", "boxplot"}
 print("hello")
 @app.post("/generate_viz/")
 async def generate_viz(file: UploadFile = File(...), query: str = Form(...)):
-    print("hello")
+    print("🔵 Début /generate_viz")
     try:
-        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)
 
+        df = pd.read_excel(file.file)
         numeric_cols = df.select_dtypes(include=["number"]).columns
 
         if len(numeric_cols) < 1:
@@ -210,7 +207,7 @@ async def generate_viz(file: UploadFile = File(...), query: str = Form(...)):
         y_col = numeric_cols[1] if query != "histplot" and len(numeric_cols) > 1 else None
         prompt_y = f', y="{y_col}"' if y_col else ""
 
-        # Prompt d'entrée pour le modèle
+        # Prompt pour l'IA
         prompt = f"""
 ### Génère uniquement du code Python fonctionnel pour tracer un {query} avec Matplotlib et Seaborn ###
 # Contraintes :
@@ -226,32 +223,39 @@ plt.savefig("plot.png")
 plt.close()
 """
 
+        print("🟣 Prompt envoyé au modèle :")
+        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)
-        generated_code = codegen_tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
+        outputs = codegen_model.generate(
+            **inputs,
+            max_new_tokens=120,
+            pad_token_id=codegen_tokenizer.eos_token_id
+        )
 
-        # 🔥 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]
+        print("🟠 Raw output du modèle :")
+        print(outputs)
 
-        # 🔥 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)
+        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]
 
-        print("🔹 Code généré propre :\n", 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)
 
         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)
 
-        exec_env = {
-            "__builtins__": __builtins__,
-            "df": df,
-            }
-        print("🔹🔹🔹 Code réellement exécuté :\n", generated_code)
+        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"
@@ -262,9 +266,12 @@ plt.close()
 
         with open(img_path, "rb") as image_file:
             encoded_string = base64.b64encode(image_file.read()).decode('utf-8')
-            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)