flashstudio 0.0.1__py3-none-any.whl

flashstudio/__init__.py

@@ -0,0 +1,5 @@
+"""FlashStudio — Interactive Training & Inference UI for FlashDet."""
+
+__version__ = "0.1.2"
+
+from flashstudio.launcher import launch  # noqa: F401

flashstudio/app.py

@@ -0,0 +1,194 @@
+"""FlashStudio — Main Streamlit Application."""
+
+import streamlit as st
+
+st.set_page_config(
+    page_title="FlashStudio",
+    page_icon="FS",
+    layout="wide",
+    initial_sidebar_state="expanded",
+)
+
+from flashstudio.components.styles import inject_custom_css  # noqa: E402
+from flashstudio.components.sidebar import render_sidebar  # noqa: E402
+from flashstudio.components.wizard import render_step_indicator, render_navigation  # noqa: E402
+from flashstudio.components.project_manager import (  # noqa: E402
+    get_active_project, render_project_manager_page,
+    save_project_state, list_projects, create_project,
+    load_project_state,
+)
+from flashstudio.pages.dashboard import render_dashboard  # noqa: E402
+from flashstudio.pages.data import render_data_page  # noqa: E402
+from flashstudio.pages.model import render_model_page  # noqa: E402
+from flashstudio.pages.training import render_training_page  # noqa: E402
+from flashstudio.pages.export import render_export_page  # noqa: E402
+from flashstudio.pages.inference import render_inference_page  # noqa: E402
+
+STEPS = [
+    {"id": "dashboard", "label": "Dashboard", "icon": "1"},
+    {"id": "data", "label": "Data", "icon": "2"},
+    {"id": "model", "label": "Model", "icon": "3"},
+    {"id": "training", "label": "Training", "icon": "4"},
+    {"id": "export", "label": "Export", "icon": "5"},
+    {"id": "inference", "label": "Inference", "icon": "6"},
+]
+
+PAGE_RENDERERS = {
+    "dashboard": render_dashboard,
+    "data": render_data_page,
+    "model": render_model_page,
+    "training": render_training_page,
+    "export": render_export_page,
+    "inference": render_inference_page,
+}
+
+
+def _ensure_config_in_session(project_id: str):
+    """Re-seed any missing model config keys from saved project state.
+
+    Streamlit may clean up widget-owned session state keys when the widget is
+    not rendered (e.g. navigating away from the Model page). This ensures
+    critical config values like epochs, batch_size, lr etc. always remain
+    available in session state for other pages to read.
+    """
+    import os
+    import json
+    from flashstudio.components.project_manager import get_project_dir
+
+    config_keys = [
+        "epochs", "batch_size", "lr", "img_size", "weight_decay",
+        "warmup_epochs", "patience", "num_workers", "grad_accum",
+        "val_interval", "lr_final_ratio", "optimizer",
+        "model_arch", "arch_family", "pico_backbone",
+        "amp", "grad_clip", "multiscale",
+        "aug_mosaic", "aug_mixup", "aug_copypaste",
+        "finetune_strategy", "pretrain_option", "custom_weights",
+        "lora_variant", "lora_rank", "lora_alpha", "lora_dropout", "lora_targets",
+        "activation_checkpointing", "activation_offloading", "optimizer_in_bwd",
+        "compile_model", "use_8bit_optimizer", "ddp", "chunked_loss", "chunk_size",
+        "save_dir", "save_best", "resume_training",
+        "num_classes", "class_names",
+    ]
+
+    missing = [k for k in config_keys if k not in st.session_state]
+    if not missing:
+        return
+
+    proj_dir = get_project_dir(project_id)
+    state_file = os.path.join(proj_dir, "session_state.json")
+    if not os.path.isfile(state_file):
+        return
+
+    try:
+        with open(state_file, "r") as f:
+            saved = json.load(f)
+    except (json.JSONDecodeError, OSError):
+        return
+
+    for key in missing:
+        if key in saved:
+            st.session_state[key] = saved[key]
+
+
+def _init_config_mirror():
+    """Build initial _config_mirror from loaded session state."""
+    mirror_keys = [
+        "epochs", "batch_size", "lr", "img_size", "weight_decay",
+        "warmup_epochs", "patience", "num_workers", "grad_accum",
+        "val_interval", "lr_final_ratio", "optimizer",
+        "model_arch", "arch_family", "pico_backbone",
+        "amp", "grad_clip", "multiscale",
+        "aug_mosaic", "aug_mixup", "aug_copypaste",
+        "finetune_strategy", "pretrain_option", "custom_weights",
+        "lora_variant", "lora_rank", "lora_alpha", "lora_dropout", "lora_targets",
+        "activation_checkpointing", "activation_offloading", "optimizer_in_bwd",
+        "compile_model", "use_8bit_optimizer", "ddp", "chunked_loss", "chunk_size",
+        "save_dir", "save_best", "resume_training",
+    ]
+    mirror = {}
+    for k in mirror_keys:
+        if k in st.session_state:
+            mirror[k] = st.session_state[k]
+    if mirror:
+        st.session_state["_config_mirror"] = mirror
+
+
+def main():
+    import traceback as _tb
+
+    inject_custom_css()
+
+    if "current_step" not in st.session_state:
+        st.session_state["current_step"] = 0
+
+    # Project system — check if user has an active project
+    active_project = get_active_project()
+
+    if active_project is None and not list_projects():
+        # First time user — show project creation
+        _render_first_time_setup()
+        return
+
+    # Load project state once on initial page load
+    if active_project and not st.session_state.get("_project_state_loaded"):
+        load_project_state(active_project["id"])
+        st.session_state["_project_state_loaded"] = True
+        # Initialize config mirror from loaded project state
+        _init_config_mirror()
+
+    # Re-seed any config keys that Streamlit's widget lifecycle may have removed
+    if active_project:
+        _ensure_config_in_session(active_project["id"])
+
+    if st.session_state.get("show_project_manager"):
+        render_project_manager_page()
+        if st.button("← Back to Workspace", key="back_from_projects"):
+            st.session_state["show_project_manager"] = False
+            st.rerun()
+        return
+
+    render_sidebar()
+    render_step_indicator(STEPS, st.session_state["current_step"])
+
+    current = STEPS[st.session_state["current_step"]]
+
+    try:
+        PAGE_RENDERERS[current["id"]]()
+    except Exception as _e:
+        st.error(f"Error: {_e}")
+        st.code(_tb.format_exc())
+
+    render_navigation(STEPS, st.session_state["current_step"])
+
+    # Auto-save project state on each page render
+    if active_project:
+        save_project_state()
+
+
+def _render_first_time_setup():
+    """Compact first-time project creation."""
+    st.markdown(
+        "<div style='text-align:center; padding:2rem 0 0.5rem;'>"
+        "<b style='font-size:2.5rem;'>FS</b><br>"
+        "<b style='font-size:1.3rem;'>Welcome to FlashStudio</b><br>"
+        "<span style='color:#6B7280;font-size:0.9rem;'>Create your first project to get started</span>"
+        "</div>",
+        unsafe_allow_html=True,
+    )
+
+    with st.container(border=True):
+        col_form, col_info = st.columns([3, 2])
+        with col_form:
+            name = st.text_input("Project Name", placeholder="e.g. Traffic Detection", key="first_project_name")
+            desc = st.text_input("Description (optional)", placeholder="Detect vehicles in traffic footage", key="first_project_desc")
+            if st.button("Create and Start", type="primary", disabled=not name, use_container_width=True):
+                create_project(name, desc)
+                st.rerun()
+        with col_info:
+            st.markdown("#### What's a Project?")
+            for item in ["Dataset management", "Model configuration", "Training runs", "Model exports", "Inference pipeline"]:
+                st.markdown(f'<span style="font-size:0.88rem;color:#4B5563;">• {item}</span>', unsafe_allow_html=True)
+
+
+if __name__ == "__main__":
+    main()

flashstudio/cli.py

@@ -0,0 +1,18 @@
+"""FlashStudio CLI entrypoint."""
+
+import argparse
+
+
+def main():
+    parser = argparse.ArgumentParser(description="FlashStudio — Training & Inference UI for FlashDet")
+    parser.add_argument("--port", type=int, default=8501, help="Port to serve on")
+    parser.add_argument("--no-share", action="store_true", help="Disable ngrok sharing in Colab")
+    parser.add_argument("--ngrok-token", type=str, default=None, help="ngrok auth token")
+    args = parser.parse_args()
+
+    from flashstudio.launcher import launch
+    launch(port=args.port, share=not args.no_share, ngrok_token=args.ngrok_token)
+
+
+if __name__ == "__main__":
+    main()