aaaa

app.py

@@ -220,51 +220,51 @@ 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 = 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 : 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()
+        # 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.strip():
-        return JSONResponse(content={"error": "Erreur : Code généré vide."}, status_code=500)
+        if not generated_code.strip():
+            return JSONResponse(content={"error": "Erreur : Code généré vide."}, status_code=500)
 
-    try:
+        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)
+        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 = {"df": df, "plt": plt, "sns": sns, "pd": pd}
-    print("🔹🔹🔹 Code réellement exécuté :")
-    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)