m8flow 1.1.2 → 1.1.3

package/bundled/backend/services/pipeline_executor.py

@@ -67,6 +67,21 @@ def _serialize_value(value: Any) -> Any:
         return bool(value)
 
     # ── sklearn ──────────────────────────────────────────────────────────────
+    is_display = hasattr(value, "figure_") and hasattr(value.figure_, "savefig")
+    if is_display:
+        try:
+            import io
+            import base64
+            buf = io.BytesIO()
+            value.figure_.savefig(buf, format="png", bbox_inches="tight")
+            b64 = base64.b64encode(buf.getvalue()).decode("utf-8")
+            return {
+                "image_base64": b64,
+                "title": getattr(value, "estimator_name", type(value).__name__)
+            }
+        except Exception:
+            pass
+
     if isinstance(value, BaseEstimator):
         params = {}
         for k, v in value.get_params().items():
@@ -76,6 +91,32 @@ def _serialize_value(value: Any) -> Any:
                 params[k] = str(v)
         return {"_type": "Estimator", "class": type(value).__name__, "params": params}
 
+    # ── matplotlib / seaborn ──────────────────────────────────────────────────
+    is_figure = hasattr(value, "savefig") and type(value).__module__.startswith("matplotlib")
+    if is_figure:
+        try:
+            import io
+            import base64
+            buf = io.BytesIO()
+            value.savefig(buf, format="png", bbox_inches="tight")
+            b64 = base64.b64encode(buf.getvalue()).decode("utf-8")
+            return {
+                "image_base64": b64,
+                "title": "Matplotlib / Seaborn Plot"
+            }
+        except Exception:
+            pass
+
+    # ── plotly ───────────────────────────────────────────────────────────────
+    if hasattr(value, "to_html") and type(value).__module__.startswith("plotly"):
+        try:
+            return {
+                "plotly_html": value.to_html(full_html=False, include_plotlyjs="cdn"),
+                "title": "Plotly Visualization"
+            }
+        except Exception:
+            pass
+
     # ── plain Python scalars ─────────────────────────────────────────────────
     if isinstance(value, float):
         return _safe_float(value)

package/bundled/backend/templates.py

@@ -3,20 +3,15 @@
 # ── Data ──────────────────────────────────────────────────────────────────────
 
 CSV_LOADER = '''import pandas as pd
-import os
-from typing import Annotated
 
-def run(file_path: Annotated[str, "file"] = "data.csv") -> dict:
-    """Loads a CSV file. Looks in the storage directory if a relative path is provided."""
-    # Resolve path
-    if not os.path.isabs(file_path):
-        storage_dir = os.environ.get("M8FLOW_UPLOAD_DIR") or os.path.abspath("uploads")
-        potential_path = os.path.join(storage_dir, file_path)
-        if os.path.exists(potential_path):
-            file_path = potential_path
-            
-    df = pd.read_csv(file_path)
-    return {"data": df}
+def run(file_path=None):
+    if file_path is None:
+        raise ValueError("No file path provided")
+    try:
+        data = pd.read_csv(file_path)
+        return {"data": data}
+    except Exception as e:
+        raise ValueError(f"Failed to load CSV: {str(e)}")
 '''
 
 CSV_EXPORTER = '''import pandas as pd