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

fraclab_sdk/workbench/pages/6_Algorithm_Edit.py

@@ -0,0 +1,116 @@
+"""Algorithm editor page."""
+
+import json
+import shutil
+
+import streamlit as st
+
+from fraclab_sdk.algorithm import AlgorithmLibrary
+from fraclab_sdk.config import SDKConfig
+from fraclab_sdk.workbench import ui_styles
+
+st.set_page_config(page_title="Algorithm Editor", page_icon="✏️", layout="wide", initial_sidebar_state="expanded")
+st.title("Algorithm Editor")
+
+ui_styles.apply_global_styles()
+
+# --- Page-Specific CSS: Editor Styling ---
+st.markdown("""
+<style>
+    /* Make Text Area look like a code editor */
+    textarea {
+        font-family: "Source Code Pro", "Consolas", "Courier New", monospace !important;
+        font-size: 14px !important;
+        line-height: 1.5 !important;
+        color: #333 !important;
+        background-color: #fcfcfc !important;
+    }
+</style>
+""", unsafe_allow_html=True)
+
+
+config = SDKConfig()
+algo_lib = AlgorithmLibrary(config)
+algos = algo_lib.list_algorithms()
+
+if not algos:
+    st.info("No algorithms imported. Use the Snapshots page to import one.")
+    st.stop()
+
+# --- 1. Selection Bar ---
+with st.container(border=True):
+    c1, c2 = st.columns([3, 1])
+    with c1:
+        algo_options = {f"{a.algorithm_id}:{a.version}": a for a in algos}
+        selected_key = st.selectbox(
+            "Select Algorithm",
+            options=list(algo_options.keys()),
+            format_func=lambda k: f"{algo_options[k].algorithm_id} (v{algo_options[k].version})",
+            label_visibility="collapsed"
+        )
+    with c2:
+        if selected_key:
+            selected = algo_options[selected_key]
+            st.caption(f"ID: `{selected.algorithm_id}`")
+
+if not selected_key:
+    st.stop()
+
+# Load Data
+handle = algo_lib.get_algorithm(selected.algorithm_id, selected.version)
+algo_dir = handle.directory
+algo_file = algo_dir / "main.py"
+manifest_file = algo_dir / "manifest.json"
+
+algo_text = algo_file.read_text(encoding="utf-8") if algo_file.exists() else ""
+manifest = json.loads(manifest_file.read_text(encoding="utf-8")) if manifest_file.exists() else {}
+current_version = manifest.get("codeVersion", selected.version)
+
+# --- 2. Action Bar (Specific to Algorithm Edit: Versioning) ---
+col_ver, col_spacer, col_save = st.columns([2, 4, 1])
+
+with col_ver:
+    new_version = st.text_input("Target Version", value=current_version, help="Change this to save as a new version")
+
+# --- 3. Editor Area ---
+st.caption(f"Editing: `{algo_file}`")
+edited_text = st.text_area(
+    "Code Editor",
+    value=algo_text,
+    height=600,
+    label_visibility="collapsed"
+)
+
+# --- Save Logic ---
+with col_save:
+    # Button aligned with the input box visually
+    st.write("") 
+    st.write("") 
+    if st.button("💾 Save Changes", type="primary", width="stretch"):
+        try:
+            # Write edits to current workspace
+            algo_dir.mkdir(parents=True, exist_ok=True)
+            algo_file.write_text(edited_text, encoding="utf-8")
+
+            # Update manifest version and write
+            if manifest:
+                manifest["codeVersion"] = new_version
+                manifest_file.write_text(json.dumps(manifest, indent=2), encoding="utf-8")
+
+            # If version changed, copy to new workspace folder
+            if new_version != selected.version:
+                new_dir = config.algorithms_dir / selected.algorithm_id / new_version
+                new_dir.mkdir(parents=True, exist_ok=True)
+                shutil.copytree(algo_dir, new_dir, dirs_exist_ok=True)
+                # Ensure manifest in new dir reflects version
+                new_manifest_path = new_dir / "manifest.json"
+                if new_manifest_path.exists():
+                    new_manifest = json.loads(new_manifest_path.read_text())
+                    new_manifest["codeVersion"] = new_version
+                    new_manifest_path.write_text(json.dumps(new_manifest, indent=2), encoding="utf-8")
+                st.toast(f"Saved as new version: {new_version}", icon="✅")
+                st.success(f"New workspace created at: `{new_dir}`")
+            else:
+                st.toast("File saved successfully", icon="✅")
+        except Exception as e:
+            st.error(f"Save failed: {e}")

fraclab_sdk/workbench/pages/7_Schema_Edit.py

@@ -0,0 +1,160 @@
+"""Schema editor page for editing algorithm InputSpec."""
+
+from __future__ import annotations
+
+import json
+from pathlib import Path
+
+import streamlit as st
+
+from fraclab_sdk.algorithm import AlgorithmLibrary
+from fraclab_sdk.config import SDKConfig
+from fraclab_sdk.devkit.validate import validate_inputspec
+from fraclab_sdk.workbench import ui_styles
+from fraclab_sdk.workbench.utils import run_workspace_script
+
+st.set_page_config(page_title="Schema Editor", page_icon="🧩", layout="wide", initial_sidebar_state="expanded")
+st.title("Schema Editor (InputSpec)")
+
+ui_styles.apply_global_styles()
+
+# --- Page-Specific CSS: Editor Styling ---
+st.markdown("""
+<style>
+    textarea {
+        font-family: "Source Code Pro", monospace !important;
+        font-size: 14px !important;
+        background-color: #fcfcfc !important;
+    }
+</style>
+""", unsafe_allow_html=True)
+
+
+def write_params_schema(ws_dir: Path) -> None:
+    """Generate dist/params.schema.json via subprocess."""
+    script = '''
+import json
+from schema.inputspec import INPUT_SPEC
+
+if hasattr(INPUT_SPEC, "model_json_schema"):
+    schema = INPUT_SPEC.model_json_schema()
+elif hasattr(INPUT_SPEC, "schema"):
+    schema = INPUT_SPEC.schema()
+else:
+    raise SystemExit("INPUT_SPEC missing schema generator")
+print(json.dumps(schema))
+'''
+    result = run_workspace_script(ws_dir, script)
+    if result.returncode != 0:
+        raise RuntimeError(result.stderr or "Failed to generate params.schema.json")
+
+    try:
+        data = json.loads(result.stdout)
+    except json.JSONDecodeError as exc:
+        raise RuntimeError("Failed to parse generated params schema") from exc
+
+    dist_dir = ws_dir / "dist"
+    dist_dir.mkdir(parents=True, exist_ok=True)
+    (dist_dir / "params.schema.json").write_text(json.dumps(data, indent=2), encoding="utf-8")
+
+
+DOC_SUMMARY = """
+**InputSpec Cheatsheet:**
+- **Types**: `str`, `int`, `float`, `bool`, `datetime`, `Optional[T]`, `Literal["A", "B"]`.
+- **Field**: `Field(..., title="Title", description="Desc")`.
+- **UI Metadata**: `json_schema_extra=schema_extra(group="Basic", order=1, ui_type="range")`.
+- **Visibility**: `show_when=show_when_condition("mode", "equals", "advanced")`.
+- **Validation**: `@field_validator("field")` or `@model_validator(mode="after")`.
+"""
+
+config = SDKConfig()
+algo_lib = AlgorithmLibrary(config)
+algos = algo_lib.list_algorithms()
+
+if not algos:
+    st.info("No algorithms imported.")
+    st.stop()
+
+# --- 1. Selection ---
+with st.container(border=True):
+    c1, c2 = st.columns([3, 1])
+    with c1:
+        algo_options = {f"{a.algorithm_id}:{a.version}": a for a in algos}
+        selected_key = st.selectbox(
+            "Select Algorithm",
+            options=list(algo_options.keys()),
+            format_func=lambda k: f"{algo_options[k].algorithm_id} (v{algo_options[k].version})",
+            label_visibility="collapsed"
+        )
+
+if not selected_key:
+    st.stop()
+
+selected = algo_options[selected_key]
+handle = algo_lib.get_algorithm(selected.algorithm_id, selected.version)
+algo_dir = handle.directory
+
+st.caption(f"Algorithm dir: `{algo_dir}`")
+schema_dir = algo_dir / "schema"
+schema_dir.mkdir(parents=True, exist_ok=True)
+inputspec_path = schema_dir / "inputspec.py"
+
+DEFAULT_INPUTSPEC = '''from __future__ import annotations
+from pydantic import BaseModel, Field
+from .base import schema_extra, show_when_condition, show_when_and, show_when_or
+
+class INPUT_SPEC(BaseModel):
+    """Algorithm parameters."""
+    # datasetKey: str = Field(..., title="Dataset Key")
+'''
+
+if not inputspec_path.exists():
+    inputspec_path.write_text(DEFAULT_INPUTSPEC, encoding="utf-8")
+
+# --- 2. Documentation ---
+with st.expander("📚 Documentation & Tips", expanded=True):
+    st.markdown(DOC_SUMMARY)
+
+# --- 3. Editor ---
+content = inputspec_path.read_text(encoding="utf-8")
+edited = st.text_area("inputspec.py", value=content, height=600, label_visibility="collapsed")
+
+# --- 4. Actions ---
+col_save, col_valid, col_spacer = st.columns([1, 1, 4])
+
+with col_save:
+    if st.button("💾 Save & Generate", type="primary", width="stretch"):
+        try:
+            inputspec_path.write_text(edited, encoding="utf-8")
+            write_params_schema(algo_dir)
+            st.toast("Schema saved and JSON generated!", icon="✅")
+        except Exception as e:
+            st.error(f"Save failed: {e}")
+
+with col_valid:
+    if st.button("🔍 Validate", type="secondary", width="stretch"):
+        try:
+            # Auto-save before validate
+            inputspec_path.write_text(edited, encoding="utf-8")
+            write_params_schema(algo_dir)
+
+            result = validate_inputspec(algo_dir)
+            if result.valid:
+                if result.warnings:
+                    st.warning(f"Validation Passed with {len(result.warnings)} warning(s)", icon="⚠️")
+                else:
+                    st.success("Validation Passed!", icon="✅")
+            else:
+                st.error(f"Validation Failed ({len(result.errors)} error(s))", icon="🚫")
+
+            # Show all issues (errors and warnings)
+            for issue in result.issues:
+                icon = "🔴" if issue.severity.value == "error" else "🟡"
+                path_str = f" at `{issue.path}`" if issue.path else ""
+                details_str = ""
+                if issue.details:
+                    if "suggested" in issue.details:
+                        details_str = f" → Suggested: `{issue.details['suggested']}`"
+                st.markdown(f"{icon} **{issue.code}**{path_str}: {issue.message}{details_str}")
+        except Exception as e:
+            st.error(f"Validation error: {e}")

fraclab_sdk/workbench/pages/8_Output_Edit.py

@@ -0,0 +1,155 @@
+"""OutputSpec editor page for editing algorithm output_contract."""
+
+from __future__ import annotations
+
+import json
+import shutil
+from pathlib import Path
+
+import streamlit as st
+
+from fraclab_sdk.algorithm import AlgorithmLibrary
+from fraclab_sdk.config import SDKConfig
+from fraclab_sdk.devkit.validate import validate_output_contract
+from fraclab_sdk.workbench import ui_styles
+from fraclab_sdk.workbench.utils import run_workspace_script
+
+st.set_page_config(page_title="OutputSpec Editor", page_icon="📤", layout="wide", initial_sidebar_state="expanded")
+st.title("OutputSpec Editor")
+
+ui_styles.apply_global_styles()
+
+# --- Page-Specific CSS: Editor Styling ---
+st.markdown("""
+<style>
+    textarea {
+        font-family: "Source Code Pro", monospace !important;
+        font-size: 14px !important;
+        background-color: #fcfcfc !important;
+    }
+</style>
+""", unsafe_allow_html=True)
+def write_dist_from_contract(ws_dir: Path, algo_id: str, version: str) -> None:
+    """Import OUTPUT_CONTRACT and dump to dist/output_contract.json."""
+    script = '''
+import json
+from schema.output_contract import OUTPUT_CONTRACT
+if hasattr(OUTPUT_CONTRACT, "model_dump"):
+    print(json.dumps(OUTPUT_CONTRACT.model_dump(mode="json")))
+else:
+    print(json.dumps(OUTPUT_CONTRACT.dict()))
+'''
+    result = run_workspace_script(ws_dir, script)
+    if result.returncode != 0:
+        raise RuntimeError(result.stderr or "Failed to load OUTPUT_CONTRACT")
+
+    try:
+        data = json.loads(result.stdout)
+    except json.JSONDecodeError as exc:
+        raise RuntimeError("Failed to parse OUTPUT_CONTRACT output") from exc
+
+    dist_dir = ws_dir / "dist"
+    dist_dir.mkdir(parents=True, exist_ok=True)
+    (dist_dir / "output_contract.json").write_text(json.dumps(data, indent=2), encoding="utf-8")
+
+
+DOC_SUMMARY = """
+**OutputSpec Cheatsheet:**
+- **Datasets**: List of `OutputDatasetContract` inside `OutputContract`.
+- **Props**: `key` (unique), `kind` (frame/object/blob/scalar), `owner`, `cardinality` (one/many), `required`.
+- **Schema**: Must match `kind` (e.g., `ScalarSchema` for kind='scalar').
+- **Dimensions**: List of string keys used in artifact dims.
+"""
+
+config = SDKConfig()
+algo_lib = AlgorithmLibrary(config)
+algos = algo_lib.list_algorithms()
+
+if not algos:
+    st.info("No algorithms imported.")
+    st.stop()
+
+# --- 1. Selection ---
+with st.container(border=True):
+    c1, c2 = st.columns([3, 1])
+    with c1:
+        algo_options = {f"{a.algorithm_id}:{a.version}": a for a in algos}
+        selected_key = st.selectbox(
+            "Select Algorithm",
+            options=list(algo_options.keys()),
+            format_func=lambda k: f"{algo_options[k].algorithm_id} (v{algo_options[k].version})",
+            label_visibility="collapsed"
+        )
+
+if not selected_key:
+    st.stop()
+
+selected = algo_options[selected_key]
+handle = algo_lib.get_algorithm(selected.algorithm_id, selected.version)
+workspace_dir = handle.directory
+
+st.caption(f"Algorithm dir: `{workspace_dir}`")
+schema_dir = workspace_dir / "schema"
+schema_dir.mkdir(parents=True, exist_ok=True)
+output_spec_path = schema_dir / "output_contract.py"
+
+DEFAULT_OUTPUTSPEC = '''from __future__ import annotations
+from fraclab_sdk.specs.output import BlobSchema, OutputContract, OutputDatasetContract, ScalarSchema
+
+OUTPUT_CONTRACT = OutputContract(
+    datasets=[
+        # OutputDatasetContract(key="metrics", kind="scalar", cardinality="many", ...)
+    ]
+)
+'''
+
+if not output_spec_path.exists():
+    output_spec_path.write_text(DEFAULT_OUTPUTSPEC, encoding="utf-8")
+
+# --- 2. Documentation ---
+with st.expander("📚 Documentation & Tips", expanded=True):
+    st.markdown(DOC_SUMMARY)
+
+# --- 3. Editor ---
+content = output_spec_path.read_text(encoding="utf-8")
+edited = st.text_area("output_contract.py", value=content, height=600, label_visibility="collapsed")
+
+# --- 4. Actions ---
+col_save, col_valid, col_spacer = st.columns([1, 1, 4])
+
+with col_save:
+    if st.button("💾 Save & Generate", type="primary", width="stretch"):
+        try:
+            output_spec_path.write_text(edited, encoding="utf-8")
+            write_dist_from_contract(workspace_dir, selected.algorithm_id, selected.version)
+            st.toast("Output spec saved and JSON generated!", icon="✅")
+        except Exception as e:
+            st.error(f"Save failed: {e}")
+
+with col_valid:
+    if st.button("🔍 Validate", type="secondary", width="stretch"):
+        try:
+            # Auto-save before validate
+            output_spec_path.write_text(edited, encoding="utf-8")
+            write_dist_from_contract(workspace_dir, selected.algorithm_id, selected.version)
+
+            result = validate_output_contract(workspace_dir)
+            if result.valid:
+                if result.warnings:
+                    st.warning(f"Validation Passed with {len(result.warnings)} warning(s)", icon="⚠️")
+                else:
+                    st.success("Validation Passed!", icon="✅")
+            else:
+                st.error(f"Validation Failed ({len(result.errors)} error(s))", icon="🚫")
+
+            # Show all issues (errors and warnings)
+            for issue in result.issues:
+                icon = "🔴" if issue.severity.value == "error" else "🟡"
+                path_str = f" at `{issue.path}`" if issue.path else ""
+                details_str = ""
+                if issue.details:
+                    if "suggested" in issue.details:
+                        details_str = f" → Suggested: `{issue.details['suggested']}`"
+                st.markdown(f"{icon} **{issue.code}**{path_str}: {issue.message}{details_str}")
+        except Exception as e:
+            st.error(f"Validation error: {e}")