fix: final prediction display and label JSON compatibility

webapp/main.py

@@ -222,7 +222,7 @@ async def benchmark(
             "name": best_name,
             "task": task,
             "features": list(X.columns),
-            "labels": list(le.classes_) if le else None,
+            "labels": [str(c) for c in le.classes_] if le else None,
             "encoders": feat_encoders  # Store these for the /predict endpoint!
         }
             

webapp/static/app.js

@@ -502,8 +502,8 @@ function renderPlayground(datasetInfo, bestOverall, task) {
   if (!form || !bestOverall) return;
   form.innerHTML = "";
   
-  const targetCol = datasetInfo.target_col;
-  const features = (datasetInfo.columns || []).filter(c => c !== targetCol);
+  const targetCol = String(datasetInfo.target_col || "").trim().toLowerCase();
+  const features = (datasetInfo.columns || []).filter(c => String(c).trim().toLowerCase() !== targetCol);
   const preview = datasetInfo.preview ? datasetInfo.preview[0] : {};
 
   features.forEach(f => {
@@ -553,7 +553,7 @@ function renderPlayground(datasetInfo, bestOverall, task) {
       }
 
       if (task === "classification") {
-        valueEl.textContent = res.prediction || "—";
+        valueEl.textContent = (res.prediction !== undefined && res.prediction !== null) ? res.prediction : "—";
         subEl.textContent = `Most likely class (via ${bestOverall.model})`;
         
         if (res.probabilities && res.labels) {