fraclab-sdk 0.1.0__py3-none-any.whl → 0.1.2__py3-none-any.whl

fraclab_sdk/workbench/pages/4_Run.py

@@ -0,0 +1,331 @@
+"""Run page: edit params for existing runs and execute."""
+
+import json
+from pathlib import Path
+from typing import Any, Dict
+
+import streamlit as st
+
+from fraclab_sdk.algorithm import AlgorithmLibrary
+from fraclab_sdk.config import SDKConfig
+from fraclab_sdk.run import RunManager, RunStatus
+from fraclab_sdk.workbench import ui_styles
+
+st.set_page_config(page_title="Run", page_icon="▶️", layout="wide", initial_sidebar_state="expanded")
+st.title("Run")
+
+ui_styles.apply_global_styles()
+
+# Guidance: if no params UI shows up, validate InputSpec via the editor page.
+st.info(
+    "看不到参数输入框？请在 Schema Editor 页面点击 Validate 检查 "
+    "`schema/inputspec.py`，确保生成的 schema 可用。",
+    icon="ℹ️",
+)
+# --- Page-Specific CSS ---
+st.markdown("""
+<style>
+    /* Compact form labels */
+    div[data-testid="stNumberInput"] label,
+    div[data-testid="stTextInput"] label,
+    div[data-testid="stCheckbox"] label {
+        margin-bottom: 0px !important;
+        font-size: 0.85rem !important;
+        color: #666 !important;
+    }
+
+    /* Divider spacing */
+    hr { margin-top: 1rem; margin-bottom: 1rem; }
+</style>
+""", unsafe_allow_html=True)
+
+config = SDKConfig()
+run_mgr = RunManager(config)
+algo_lib = AlgorithmLibrary(config)
+
+
+# --- Intelligent Layout Engine ---
+
+def _is_compact_field(schema: dict) -> bool:
+    """Determine if a field is small enough to fit in a grid column."""
+    ftype = schema.get("type")
+    # Numbers, Booleans, and short Strings (without enums/long defaults) are compact
+    if ftype in ["integer", "number", "boolean"]:
+        return True
+    if ftype == "string" and len(str(schema.get("default", ""))) < 50:
+        return True
+    return False
+
+def render_field_widget(key: str, schema: dict, value: Any, path: str) -> Any:
+    """Render a single widget based on schema type."""
+    ftype = schema.get("type")
+    title = schema.get("title") or key
+    # Simplify label: if title is camelCase, maybe split it? For now use title directly.
+    # description = schema.get("description") # Tooltip is enough, don't clutter UI text
+    
+    default_val = schema.get("default")
+    help_text = schema.get("description")
+
+    if ftype == "string":
+        val = value if value is not None else (default_val or "")
+        return st.text_input(title, value=val, help=help_text, key=path)
+    
+    if ftype == "number":
+        val = value if value is not None else default_val
+        return st.number_input(title, value=float(val or 0.0), help=help_text, key=path)
+    
+    if ftype == "integer":
+        val = value if value is not None else default_val
+        return int(st.number_input(title, value=int(val or 0), step=1, help=help_text, key=path))
+    
+    if ftype == "boolean":
+        val = value if value is not None else default_val
+        # Toggle looks better than checkbox in grid
+        return st.toggle(title, value=bool(val), help=help_text, key=path)
+    
+    if ftype == "array":
+        # Arrays are complex, stick to full width expansion
+        return _render_json_editor(title, value, default_val, help_text, path)
+            
+    if ftype == "object":
+        # Nested object -> Recursive layout
+        with st.container(border=True):
+            st.markdown(f"**{title}**")
+            props = schema.get("properties", {})
+            obj = value if isinstance(value, dict) else (default_val if isinstance(default_val, dict) else {})
+            
+            # Recursive call to grid layout
+            return render_schema_grid(props, obj, path)
+
+    # Fallback
+    return _render_json_editor(title, value, default_val, help_text, path)
+
+def _render_json_editor(title, value, default, help_text, path):
+    """Helper for raw JSON fields."""
+    st.markdown(f"<small>{title}</small>", unsafe_allow_html=True)
+    current = value if value is not None else (default if default is not None else [])
+    text = st.text_area(
+        title,
+        value=json.dumps(current, indent=2, ensure_ascii=False),
+        help=f"{help_text} (Edit as JSON)",
+        key=path,
+        label_visibility="collapsed",
+        height=100
+    )
+    try:
+        return json.loads(text) if text.strip() else current
+    except Exception:
+        return current
+
+def render_schema_grid(properties: Dict[str, dict], current_values: Dict[str, Any], prefix: str) -> Dict[str, Any]:
+    """
+    Renders fields in a smart grid layout:
+    - Compact fields (numbers, bools) get packed into columns (up to 4).
+    - Wide fields (objects, arrays) break the line and take full width.
+    """
+    result = {}
+    
+    # 1. Separate fields into groups to maintain partial order while grid-packing
+    # Strategy: Iterate and buffer compact fields. Flush buffer when a wide field hits.
+    
+    compact_buffer = [] # List of (key, schema)
+    
+    def flush_buffer():
+        if not compact_buffer:
+            return
+        
+        # Calculate optimal columns (max 4, min 2)
+        n_items = len(compact_buffer)
+        n_cols = 4 if n_items >= 4 else (n_items if n_items > 0 else 1)
+        
+        # Split into rows if > 4 items? Simple logic: Just wrap
+        # Actually st.columns handles wrapping poorly, better to batch by 4
+        
+        for i in range(0, n_items, 4):
+            batch = compact_buffer[i : i+4]
+            cols = st.columns(len(batch))
+            for col, (b_key, b_schema) in zip(cols, batch):
+                with col:
+                    val = current_values.get(b_key)
+                    result[b_key] = render_field_widget(b_key, b_schema, val, f"{prefix}.{b_key}")
+        
+        compact_buffer.clear()
+
+    for key, prop_schema in properties.items():
+        if _is_compact_field(prop_schema):
+            compact_buffer.append((key, prop_schema))
+        else:
+            # Wide field encountered: flush buffer first
+            flush_buffer()
+            # Render wide field
+            val = current_values.get(key)
+            result[key] = render_field_widget(key, prop_schema, val, f"{prefix}.{key}")
+    
+    # Final flush
+    flush_buffer()
+    
+    # Preserve extra keys in current_values that aren't in schema
+    for k, v in current_values.items():
+        if k not in result:
+            result[k] = v
+            
+    return result
+
+
+def load_params(run_dir: Path) -> dict:
+    path = run_dir / "input" / "params.json"
+    try:
+        return json.loads(path.read_text())
+    except Exception:
+        return {}
+
+
+# ==========================================
+# Main Logic
+# ==========================================
+
+runs = run_mgr.list_runs()
+
+if not runs:
+    st.info("No runs available. Create a run from the Selection page.")
+    st.stop()
+
+pending_runs = [r for r in runs if r.status == RunStatus.PENDING]
+other_runs = [r for r in runs if r.status != RunStatus.PENDING]
+
+# ------------------------------------------
+# 1. Pending Runs (Editor Workspace)
+# ------------------------------------------
+st.subheader("Pending Runs")
+
+if not pending_runs:
+    st.info("No pending runs waiting for execution.")
+else:
+    # Use tabs for context switching
+    tabs = st.tabs([f"⚙️ {run.run_id}" for run in pending_runs])
+    
+    for tab, run in zip(tabs, pending_runs):
+        with tab:
+            run_dir = run_mgr.get_run_dir(run.run_id)
+            algo_handle = algo_lib.get_algorithm(run.algorithm_id, run.algorithm_version)
+            schema = algo_handle.params_schema
+            current_params = load_params(run_dir)
+            
+            # --- Layout: Top Info Bar ---
+            with st.container(border=True):
+                c1, c2, c3, c4 = st.columns([2, 2, 2, 1])
+                with c1: st.caption(f"**Snapshot:** `{run.snapshot_id}`")
+                with c2: st.caption(f"**Algo:** `{run.algorithm_id}` v{run.algorithm_version}")
+                with c3: st.caption(f"**Created:** {run.created_at}")
+                with c4: 
+                    # Timeout setting tucked away here
+                    timeout = st.number_input("Timeout (s)", value=300, step=10, key=f"to_{run.run_id}", label_visibility="collapsed")
+
+            # --- Layout: Parameters Grid ---
+            st.markdown("##### Parameters")
+            with st.container(border=True):
+                if schema.get("type") == "object":
+                    props = schema.get("properties", {})
+                    # CALL THE GRID ENGINE
+                    new_params = render_schema_grid(props, current_params, prefix=f"run_{run.run_id}")
+                else:
+                    st.info("Schema is not an object, editing raw JSON.")
+                    new_params = _render_json_editor("Raw Params", current_params, {}, "", f"run_raw_{run.run_id}")
+
+            st.divider()
+            
+            # --- Layout: Action Footer ---
+            # Right-aligned actions
+            _, col_btns = st.columns([3, 4])
+            with col_btns:
+                b1, b2, b3 = st.columns([1, 1, 1.5], gap="small")
+                
+                with b1:
+                    if st.button("🚫 Cancel", key=f"cancel_{run.run_id}", width="stretch"):
+                        try:
+                            run_mgr.delete_run(run.run_id)
+                            st.success("Cancelled")
+                            st.rerun()
+                        except Exception as e:
+                            st.error(f"Error: {e}")
+                
+                with b2:
+                    if st.button("💾 Save", key=f"save_{run.run_id}", width="stretch"):
+                        try:
+                            (run_dir / "input").mkdir(parents=True, exist_ok=True)
+                            (run_dir / "input" / "params.json").write_text(
+                                json.dumps(new_params, indent=2, ensure_ascii=False),
+                                encoding="utf-8",
+                            )
+                            st.toast("Parameters saved successfully!", icon="💾")
+                        except Exception as e:
+                            st.error(f"Save failed: {e}")
+
+                with b3:
+                    if st.button("▶️ Run Algorithm", key=f"exec_{run.run_id}", type="primary", width="stretch"):
+                        try:
+                            # Auto-save before run
+                            (run_dir / "input").mkdir(parents=True, exist_ok=True)
+                            (run_dir / "input" / "params.json").write_text(
+                                json.dumps(new_params, indent=2, ensure_ascii=False),
+                                encoding="utf-8",
+                            )
+
+                            with st.spinner("Initializing execution..."):
+                                result = run_mgr.execute(run.run_id, timeout_s=int(timeout))
+
+                            if result.error:
+                                st.error(f"Run Finished: {result.status.value}\n{result.error}")
+                            else:
+                                # Navigate to Results page with executed run
+                                st.session_state.executed_run_id = run.run_id
+                                st.switch_page("pages/5_Results.py")
+                        except Exception as e:
+                            st.error(f"Execution Exception: {e}")
+
+
+# ------------------------------------------
+# 2. History (Other Runs)
+# ------------------------------------------
+st.subheader("Run History")
+
+if not other_runs:
+    st.caption("No historical runs.")
+
+other_runs_reversed = other_runs[::-1]
+
+for run in other_runs_reversed:
+    status_config = {
+        RunStatus.PENDING:   ("⏳", "Pending", "gray"),
+        RunStatus.RUNNING:   ("🔄", "Running", "blue"),
+        RunStatus.SUCCEEDED: ("✅", "Succeeded", "green"),
+        RunStatus.FAILED:    ("❌", "Failed", "red"),
+        RunStatus.TIMEOUT:   ("⏱️", "Timeout", "orange"),
+    }
+    icon, label, color = status_config.get(run.status, ("❓", "Unknown", "gray"))
+    
+    with st.expander(f"{icon} {run.run_id}", expanded=False):
+        with st.container(border=True):
+            # Info
+            c1, c2, c3 = st.columns([3, 2, 2])
+            with c1: 
+                st.caption("Context")
+                st.markdown(f"**{run.algorithm_id}** v{run.algorithm_version}")
+                st.text(f"Snap: {run.snapshot_id}")
+            with c2:
+                st.caption("Timing")
+                st.text(f"Start: {run.started_at or '--'}")
+                st.text(f"End:   {run.completed_at or '--'}")
+            with c3:
+                st.caption("Status")
+                st.markdown(f":{color}[**{label}**]")
+                if run.error:
+                    st.error(run.error)
+            
+            # Params Read-only
+            st.divider()
+            st.caption("Run Parameters")
+            run_dir = run_mgr.get_run_dir(run.run_id)
+            params_view = load_params(run_dir)
+            if params_view:
+                st.code(json.dumps(params_view, indent=2, ensure_ascii=False), language="json")

fraclab_sdk/workbench/pages/5_Results.py

@@ -0,0 +1,298 @@
+"""Results viewing page."""
+
+import json
+
+import streamlit as st
+
+from fraclab_sdk.config import SDKConfig
+from fraclab_sdk.algorithm import AlgorithmLibrary
+from fraclab_sdk.results import (
+    ResultReader,
+    get_artifact_preview_type,
+    preview_image,
+    preview_json_raw,
+    preview_json_table,
+    preview_scalar,
+)
+from fraclab_sdk.run import RunManager, RunStatus
+from fraclab_sdk.workbench import ui_styles
+
+st.set_page_config(page_title="Results", page_icon="📊", layout="wide", initial_sidebar_state="expanded")
+st.title("Results")
+
+ui_styles.apply_global_styles()
+
+# Guidance: if artifacts preview looks empty, validate OutputContract via the editor page.
+st.info(
+    "看不到期望的输出？请在 Output Editor 页面点击 Validate 检查 "
+    "`schema/output_contract.py` 的 datasets/items/artifacts 是否正确。",
+    icon="ℹ️",
+)
+
+def get_manager():
+    """Get run manager."""
+    config = SDKConfig()
+    return RunManager(config)
+
+
+run_manager = get_manager()
+algo_lib = AlgorithmLibrary(run_manager._config)
+runs = run_manager.list_runs()
+
+if not runs:
+    st.info("No runs available.")
+    st.stop()
+
+# ==========================================
+# 1. Run Selection & Status
+# ==========================================
+
+# Prepare options
+run_options = {r.run_id: r for r in runs}
+# Sort by latest first usually makes sense
+run_ids = list(reversed(list(run_options.keys()))) 
+
+# Check for navigation context
+default_run_id = st.session_state.pop("executed_run_id", None) or st.session_state.pop("created_run_id", None)
+default_index = run_ids.index(default_run_id) if default_run_id in run_ids else 0
+
+with st.container(border=True):
+    col_sel, col_stat = st.columns([4, 1])
+    
+    with col_sel:
+        selected_run_id = st.selectbox(
+            "Select Run",
+            options=run_ids,
+            index=default_index,
+            format_func=lambda x: f"{x} — {run_options[x].algorithm_id} (v{run_options[x].algorithm_version})",
+            label_visibility="collapsed"
+        )
+    
+    with col_stat:
+        if selected_run_id:
+            status = run_options[selected_run_id].status
+            status_color = {
+                RunStatus.SUCCEEDED: "green",
+                RunStatus.FAILED: "red",
+                RunStatus.PENDING: "gray",
+                RunStatus.RUNNING: "blue"
+            }.get(status, "gray")
+            st.markdown(f"<div style='text-align:center; padding: 8px; border: 1px solid #ddd; border-radius: 6px;'>Status: <b style='color:{status_color}'>{status.value}</b></div>", unsafe_allow_html=True)
+
+if not selected_run_id:
+    st.stop()
+
+run = run_options[selected_run_id]
+run_dir = run_manager.get_run_dir(selected_run_id)
+reader = ResultReader(run_dir)
+
+# ==========================================
+# 2. Run Context
+# ==========================================
+
+def _load_output_contract(algo_id: str, algo_version: str):
+    """Load output_contract.json from algorithm directory."""
+    try:
+        handle = algo_lib.get_algorithm(algo_id, algo_version)
+        manifest_path = handle.directory / "manifest.json"
+        manifest_data = {}
+        if manifest_path.exists():
+            manifest_data = json.loads(manifest_path.read_text())
+        files = manifest_data.get("files") or {}
+        rel = files.get("outputContractPath", "dist/output_contract.json")
+        contract_path = (handle.directory / rel).resolve()
+        if not contract_path.exists():
+            return None
+        return json.loads(contract_path.read_text())
+    except Exception:
+        return None
+
+output_contract = _load_output_contract(run.algorithm_id, run.algorithm_version)
+
+with st.expander("ℹ️ Run Metadata", expanded=False):
+    c1, c2, c3 = st.columns(3)
+    with c1:
+        st.caption("Snapshot ID")
+        st.code(run.snapshot_id, language="text")
+    with c2:
+        st.caption("Algorithm ID")
+        st.code(run.algorithm_id, language="text")
+    with c3:
+        st.caption("Timestamps")
+        st.text(f"Start: {run.started_at}\nEnd:   {run.completed_at}")
+
+if run.error:
+    st.error(f"Run Error: {run.error}")
+elif reader.has_manifest() and reader.get_error():
+    st.error(f"Manifest Error: {reader.get_error()}")
+
+st.divider()
+
+# ==========================================
+# 3. Artifacts (Default Expanded)
+# ==========================================
+st.subheader("Artifacts")
+
+if not reader.has_manifest():
+    st.warning("⚠️ Output manifest not found (Run may have failed or produced no output)")
+else:
+    manifest = reader.read_manifest()
+
+    if not manifest.datasets:
+        st.info("No artifacts produced.")
+    else:
+        for ds in manifest.datasets:
+            # Match with contract
+            contract_ds = None
+            if output_contract:
+                contract_ds = next((d for d in output_contract.get("datasets", []) if d.get("key") == ds.datasetKey), None)
+
+            header = f"📂 {ds.datasetKey}"
+            if contract_ds and contract_ds.get("role"):
+                header += f" ({contract_ds.get('role')})"
+            
+            # --- DATASET LEVEL: EXPANDED BY DEFAULT ---
+            with st.expander(header, expanded=True):
+                # Contract Info Bar
+                if contract_ds:
+                    st.caption(
+                        f"**Schema Definition:** Kind=`{contract_ds.get('kind')}` • "
+                        f"Owner=`{contract_ds.get('owner')}` • "
+                        f"Card=`{contract_ds.get('cardinality')}`"
+                    )
+                
+                # --- ITEMS LEVEL: CARDS (Always Visible) ---
+                for item in ds.items:
+                    art = item.artifact
+                    preview_type = get_artifact_preview_type(art)
+                    
+                    with st.container(border=True):
+                        # Item Header & Metadata
+                        m1, m2, m3 = st.columns([2, 2, 3])
+                        with m1:
+                            st.markdown(f"**Item:** `{item.itemKey or art.artifactKey}`")
+                        with m2:
+                            st.caption(f"Type: `{art.artifactType}`")
+                        with m3:
+                            if art.mimeType: st.caption(f"MIME: `{art.mimeType}`")
+                        
+                        # [Modified] 删除了 Owner 和 Dims 的显示
+                        st.markdown("---")
+                        
+                        # Content Preview
+                        if preview_type == "scalar":
+                            value = preview_scalar(art)
+                            st.metric(label="Value", value=value)
+
+                        elif preview_type == "image":
+                            image_path = preview_image(art)
+                            if image_path and image_path.exists():
+                                # [Modified] use_column_width=True -> width="stretch"
+                                st.image(str(image_path), caption=art.artifactKey, width="stretch")
+                            else:
+                                st.warning("Image file missing")
+
+                        elif preview_type == "json_table":
+                            table_data = preview_json_table(art)
+                            if table_data:
+                                # Use static table for cleaner look
+                                st.table([dict(zip(table_data["columns"], row)) for row in table_data["rows"]])
+                            else:
+                                st.warning("Invalid table data")
+
+                        elif preview_type == "json_raw":
+                            json_content = preview_json_raw(art)
+                            if json_content:
+                                st.code(json_content, language="json")
+                            else:
+                                st.warning("Empty JSON")
+
+                        elif preview_type == "file":
+                            path = reader.get_artifact_path(art.artifactKey)
+                            if path:
+                                f_col1, f_col2 = st.columns([4, 1])
+                                with f_col1:
+                                    st.code(str(path), language="text")
+                                with f_col2:
+                                    if path.exists():
+                                        st.download_button(
+                                            "⬇️ Download",
+                                            data=path.read_bytes(),
+                                            file_name=path.name,
+                                            mime=art.mimeType or "application/octet-stream",
+                                            use_container_width=True # Button still uses old kwarg? No, replaced below if needed in logic, but standard download_button uses use_container_width in modern versions. If your version deprecated it for buttons too, this should be width="stretch". Let's stick to consistent modern API.
+                                        )
+                            else:
+                                st.warning("File path resolution failed")
+
+                        else:
+                            st.info("No preview available for this artifact type.")
+
+st.divider()
+
+# ==========================================
+# 4. Logs & Debug
+# ==========================================
+st.subheader("Logs & System Info")
+
+tab1, tab2, tab3, tab4 = st.tabs(["📜 Algorithm Log", "📤 Stdout", "⚠️ Stderr", "🔍 Manifest JSON"])
+
+with tab1:
+    log = reader.read_algorithm_log()
+    if log:
+        st.code(log, language="text")
+    else:
+        st.caption("No algorithm log available.")
+
+with tab2:
+    stdout = reader.read_stdout()
+    if stdout:
+        st.code(stdout, language="text")
+    else:
+        st.caption("No stdout recorded.")
+
+with tab3:
+    stderr = reader.read_stderr()
+    if stderr:
+        st.code(stderr, language="text")
+    else:
+        st.caption("No stderr recorded.")
+
+with tab4:
+    if reader.has_manifest():
+        manifest = reader.read_manifest()
+        st.code(json.dumps(manifest.model_dump(exclude_none=True), indent=2), language="json")
+    else:
+        st.caption("No manifest file.")
+
+st.divider()
+
+# ==========================================
+# 5. Danger Zone
+# ==========================================
+with st.expander("🗑️ Danger Zone", expanded=False):
+    st.markdown("Deleting a run is irreversible. It will remove all artifacts and logs.")
+    
+    confirm_key = f"confirm_del_run_{run.run_id}"
+    
+    if st.button("Delete This Run", key=f"del_btn_{run.run_id}", type="secondary"):
+        st.session_state[confirm_key] = True
+
+    if st.session_state.get(confirm_key):
+        st.warning(f"Are you sure you want to delete Run `{run.run_id}`?")
+        d_c1, d_c2 = st.columns([1, 1])
+        with d_c1:
+            # [Modified] use_container_width -> width="stretch"
+            if st.button("Yes, Delete", key=f"yes_del_{run.run_id}", type="primary", width="stretch"):
+                try:
+                    run_manager.delete_run(run.run_id)
+                    st.success(f"Deleted run {run.run_id}")
+                    st.session_state.pop(confirm_key, None)
+                    st.rerun()
+                except Exception as e:
+                    st.error(f"Delete failed: {e}")
+        with d_c2:
+            # [Modified] use_container_width -> width="stretch"
+            if st.button("Cancel", key=f"no_del_{run.run_id}", width="stretch"):
+                st.session_state.pop(confirm_key, None)
+                st.rerun()