flashstudio 0.1.3__tar.gz → 0.1.4__tar.gz

{flashstudio-0.1.3 → flashstudio-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flashstudio
-Version: 0.1.3
+Version: 0.1.4
 Summary: Interactive Training & Inference UI for FlashDet — runs on Google Colab
 Author: Gaurav14cs17
 License: Apache-2.0
@@ -12,7 +12,7 @@ Classifier: Intended Audience :: Science/Research
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Programming Language :: Python :: 3
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
-Requires-Python: >=3.9
+Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 Requires-Dist: streamlit>=1.28.0
 Requires-Dist: plotly>=5.0
@@ -33,6 +33,7 @@ Requires-Dist: pyngrok>=6.0; extra == "colab"
 Provides-Extra: dev
 Requires-Dist: ruff; extra == "dev"
 Requires-Dist: pytest; extra == "dev"
+Requires-Dist: pytest-cov; extra == "dev"
 
 <p align="center">
   <img src="https://raw.githubusercontent.com/FlashVision/FlashStudio/main/assets/logo.png" width="200" alt="FlashStudio Logo">
@@ -237,26 +238,93 @@ Click the **Public URL** to open FlashStudio in a new tab.
 ```
 FlashStudio/
 ├── flashstudio/
-│   ├── __init__.py              # Package init + launch() export
-│   ├── app.py                   # Main Streamlit app (wizard flow)
-│   ├── launcher.py              # Colab/local launcher with ngrok
-│   ├── cli.py                   # CLI entrypoint
-│   ├── pages/
-│   │   ├── dashboard.py         # Overview + recent training runs
-│   │   ├── data.py              # Dataset upload/download
-│   │   ├── model.py             # Architecture & hyperparameter config
-│   │   ├── training.py          # Training monitor (reads real workspace)
-│   │   ├── export.py            # ONNX export
-│   │   └── inference.py         # 4-step inference pipeline
+│   ├── __init__.py                  # Package init + launch() export
+│   ├── app.py                       # Main Streamlit app (wizard flow)
+│   ├── launcher.py                  # Colab/local launcher with ngrok
+│   ├── cli.py                       # CLI entrypoint
+│   ├── constants.py                 # Centralized constants (paths, defaults, models)
+│   │
+│   ├── pages/                       # Each page is a sub-package
+│   │   ├── dashboard/
+│   │   │   ├── page.py              # Main dashboard render
+│   │   │   ├── pipeline_status.py   # Pipeline status cards
+│   │   │   └── recent_runs.py       # Recent training runs table
+│   │   ├── data/
+│   │   │   ├── page.py              # Main data page render
+│   │   │   ├── upload.py            # Upload tab (ZIP/TAR + class config)
+│   │   │   ├── download.py          # Download tab (quick start + registries)
+│   │   │   ├── preview.py           # Preview tab (image grid + annotations)
+│   │   │   ├── verify.py            # Verify tab (dataset validation)
+│   │   │   └── helpers.py           # Shared helpers (extract, detect, convert)
+│   │   ├── model/
+│   │   │   ├── page.py              # Main model page render
+│   │   │   ├── architecture.py      # Architecture tab (FlashDet/YOLO selection)
+│   │   │   ├── hyperparams.py       # Hyperparameters tab
+│   │   │   ├── augmentation.py      # Augmentation tab
+│   │   │   ├── advanced.py          # Advanced tab (memory, distributed)
+│   │   │   └── summary.py           # Config summary bar + YAML save/load
+│   │   ├── training/
+│   │   │   ├── page.py              # Main training page render
+│   │   │   ├── _common.py           # Shared utilities (_get_save_dir)
+│   │   │   ├── launch/              # Launch sub-package
+│   │   │   │   ├── tab.py           # Launch tab entry point
+│   │   │   │   ├── preflight.py     # Pre-flight checks
+│   │   │   │   ├── runner.py        # FlashDet Trainer subprocess
+│   │   │   │   ├── controls.py      # Start/stop/pause/resume buttons
+│   │   │   │   └── dialogs.py       # Clean/resume/config dialogs
+│   │   │   └── monitor/             # Monitor sub-package
+│   │   │       ├── tab.py           # Monitor tab entry point
+│   │   │       ├── run_meta.py      # Run metadata extraction
+│   │   │       ├── parsers.py       # CSV/log parsing
+│   │   │       ├── dashboard.py     # Run dashboard + metrics
+│   │   │       ├── curves.py        # Plotly training curves
+│   │   │       ├── visualizations.py # Epoch visualizations
+│   │   │       ├── gt_verification.py # Ground truth verification
+│   │   │       ├── log_viewer.py    # Full log viewer
+│   │   │       └── checkpoints.py   # File browser + checkpoints
+│   │   ├── export/
+│   │   │   └── page.py              # Export page (ONNX/TorchScript)
+│   │   └── inference/
+│   │       ├── page.py              # Main inference page render
+│   │       ├── model_tab.py         # Model selection tab
+│   │       ├── data_tab.py          # Data input tab (images/video/RTSP)
+│   │       ├── solution_tab.py      # Solution selection + zone drawing
+│   │       ├── run_tab.py           # Run tab + results display
+│   │       ├── detection.py         # Detection utilities (real + demo)
+│   │       └── video.py             # Video/image inference runners
+│   │
 │   ├── components/
-│   │   ├── sidebar.py           # Navigation sidebar
-│   │   ├── styles.py            # Custom CSS
-│   │   └── wizard.py            # Step indicator & navigation
+│   │   ├── sidebar.py               # Navigation sidebar
+│   │   ├── styles.py                # Custom CSS + UI helpers
+│   │   ├── wizard.py                # Step indicator + navigation
+│   │   ├── project_manager.py       # Project CRUD + state persistence
+│   │   └── zone_drawer/             # Interactive canvas zone drawing
+│   │
 │   └── utils/
-│       └── device.py            # GPU/environment detection
+│       ├── __init__.py              # Shared helpers (defaults, state, flash)
+│       ├── device.py                # GPU/environment detection
+│       ├── jobs.py                  # Background process tracking
+│       ├── filesystem.py            # Directory size/listing utilities
+│       ├── config.py                # Training config build/save/load
+│       └── training_hooks.py        # FlashDet training callbacks
+│
+├── tests/                           # Pytest test suite
+│   ├── conftest.py                  # Shared fixtures
+│   ├── test_constants.py            # Constants validation
+│   ├── test_cli.py                  # CLI tests
+│   ├── test_utils/                  # Utils sub-package tests
+│   │   ├── test_init.py
+│   │   ├── test_device.py
+│   │   ├── test_filesystem.py
+│   │   ├── test_config.py
+│   │   └── test_jobs.py
+│   ├── test_pages/                  # Page logic tests
+│   │   └── test_training_parsers.py
+│   └── test_components/
+│       └── test_project_manager.py
+│
+├── .github/workflows/ci.yml        # GitHub Actions CI
 ├── notebooks/
-│   ├── FlashStudio_Train.ipynb
-│   └── FlashStudio_Inference.ipynb
 ├── .streamlit/config.toml
 ├── pyproject.toml
 └── README.md

{flashstudio-0.1.3 → flashstudio-0.1.4}/README.md

@@ -201,26 +201,93 @@ Click the **Public URL** to open FlashStudio in a new tab.
 ```
 FlashStudio/
 ├── flashstudio/
-│   ├── __init__.py              # Package init + launch() export
-│   ├── app.py                   # Main Streamlit app (wizard flow)
-│   ├── launcher.py              # Colab/local launcher with ngrok
-│   ├── cli.py                   # CLI entrypoint
-│   ├── pages/
-│   │   ├── dashboard.py         # Overview + recent training runs
-│   │   ├── data.py              # Dataset upload/download
-│   │   ├── model.py             # Architecture & hyperparameter config
-│   │   ├── training.py          # Training monitor (reads real workspace)
-│   │   ├── export.py            # ONNX export
-│   │   └── inference.py         # 4-step inference pipeline
+│   ├── __init__.py                  # Package init + launch() export
+│   ├── app.py                       # Main Streamlit app (wizard flow)
+│   ├── launcher.py                  # Colab/local launcher with ngrok
+│   ├── cli.py                       # CLI entrypoint
+│   ├── constants.py                 # Centralized constants (paths, defaults, models)
+│   │
+│   ├── pages/                       # Each page is a sub-package
+│   │   ├── dashboard/
+│   │   │   ├── page.py              # Main dashboard render
+│   │   │   ├── pipeline_status.py   # Pipeline status cards
+│   │   │   └── recent_runs.py       # Recent training runs table
+│   │   ├── data/
+│   │   │   ├── page.py              # Main data page render
+│   │   │   ├── upload.py            # Upload tab (ZIP/TAR + class config)
+│   │   │   ├── download.py          # Download tab (quick start + registries)
+│   │   │   ├── preview.py           # Preview tab (image grid + annotations)
+│   │   │   ├── verify.py            # Verify tab (dataset validation)
+│   │   │   └── helpers.py           # Shared helpers (extract, detect, convert)
+│   │   ├── model/
+│   │   │   ├── page.py              # Main model page render
+│   │   │   ├── architecture.py      # Architecture tab (FlashDet/YOLO selection)
+│   │   │   ├── hyperparams.py       # Hyperparameters tab
+│   │   │   ├── augmentation.py      # Augmentation tab
+│   │   │   ├── advanced.py          # Advanced tab (memory, distributed)
+│   │   │   └── summary.py           # Config summary bar + YAML save/load
+│   │   ├── training/
+│   │   │   ├── page.py              # Main training page render
+│   │   │   ├── _common.py           # Shared utilities (_get_save_dir)
+│   │   │   ├── launch/              # Launch sub-package
+│   │   │   │   ├── tab.py           # Launch tab entry point
+│   │   │   │   ├── preflight.py     # Pre-flight checks
+│   │   │   │   ├── runner.py        # FlashDet Trainer subprocess
+│   │   │   │   ├── controls.py      # Start/stop/pause/resume buttons
+│   │   │   │   └── dialogs.py       # Clean/resume/config dialogs
+│   │   │   └── monitor/             # Monitor sub-package
+│   │   │       ├── tab.py           # Monitor tab entry point
+│   │   │       ├── run_meta.py      # Run metadata extraction
+│   │   │       ├── parsers.py       # CSV/log parsing
+│   │   │       ├── dashboard.py     # Run dashboard + metrics
+│   │   │       ├── curves.py        # Plotly training curves
+│   │   │       ├── visualizations.py # Epoch visualizations
+│   │   │       ├── gt_verification.py # Ground truth verification
+│   │   │       ├── log_viewer.py    # Full log viewer
+│   │   │       └── checkpoints.py   # File browser + checkpoints
+│   │   ├── export/
+│   │   │   └── page.py              # Export page (ONNX/TorchScript)
+│   │   └── inference/
+│   │       ├── page.py              # Main inference page render
+│   │       ├── model_tab.py         # Model selection tab
+│   │       ├── data_tab.py          # Data input tab (images/video/RTSP)
+│   │       ├── solution_tab.py      # Solution selection + zone drawing
+│   │       ├── run_tab.py           # Run tab + results display
+│   │       ├── detection.py         # Detection utilities (real + demo)
+│   │       └── video.py             # Video/image inference runners
+│   │
 │   ├── components/
-│   │   ├── sidebar.py           # Navigation sidebar
-│   │   ├── styles.py            # Custom CSS
-│   │   └── wizard.py            # Step indicator & navigation
+│   │   ├── sidebar.py               # Navigation sidebar
+│   │   ├── styles.py                # Custom CSS + UI helpers
+│   │   ├── wizard.py                # Step indicator + navigation
+│   │   ├── project_manager.py       # Project CRUD + state persistence
+│   │   └── zone_drawer/             # Interactive canvas zone drawing
+│   │
 │   └── utils/
-│       └── device.py            # GPU/environment detection
+│       ├── __init__.py              # Shared helpers (defaults, state, flash)
+│       ├── device.py                # GPU/environment detection
+│       ├── jobs.py                  # Background process tracking
+│       ├── filesystem.py            # Directory size/listing utilities
+│       ├── config.py                # Training config build/save/load
+│       └── training_hooks.py        # FlashDet training callbacks
+│
+├── tests/                           # Pytest test suite
+│   ├── conftest.py                  # Shared fixtures
+│   ├── test_constants.py            # Constants validation
+│   ├── test_cli.py                  # CLI tests
+│   ├── test_utils/                  # Utils sub-package tests
+│   │   ├── test_init.py
+│   │   ├── test_device.py
+│   │   ├── test_filesystem.py
+│   │   ├── test_config.py
+│   │   └── test_jobs.py
+│   ├── test_pages/                  # Page logic tests
+│   │   └── test_training_parsers.py
+│   └── test_components/
+│       └── test_project_manager.py
+│
+├── .github/workflows/ci.yml        # GitHub Actions CI
 ├── notebooks/
-│   ├── FlashStudio_Train.ipynb
-│   └── FlashStudio_Inference.ipynb
 ├── .streamlit/config.toml
 ├── pyproject.toml
 └── README.md

{flashstudio-0.1.3 → flashstudio-0.1.4}/flashstudio/app.py

@@ -43,6 +43,76 @@ PAGE_RENDERERS = {
 }
 
 
+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
 
@@ -51,11 +121,6 @@ def main():
     if "current_step" not in st.session_state:
         st.session_state["current_step"] = 0
 
-    # Ensure critical config values are always initialized
-    from flashstudio.utils import DEFAULTS
-    if not st.session_state.get("save_dir"):
-        st.session_state["save_dir"] = DEFAULTS["save_dir"]
-
     # Project system — check if user has an active project
     active_project = get_active_project()
 
@@ -68,6 +133,12 @@ def main():
     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()

{flashstudio-0.1.3 → flashstudio-0.1.4}/flashstudio/components/project_manager.py

@@ -105,10 +105,12 @@ def create_project(name: str, description: str = "") -> dict:
 
 def switch_project(project_id: str):
     """Switch active project and load its state."""
+    _clear_session_for_new_project()
     index = _load_index()
     index["active"] = project_id
     _save_index(index)
     load_project_state(project_id)
+    st.session_state["_project_state_loaded"] = True
 
 
 def delete_project(project_id: str):
@@ -203,6 +205,8 @@ def load_project_state(project_id: str):
 
 def duplicate_project(project_id: str, new_name: str) -> dict:
     """Duplicate an existing project with a new name."""
+    index = _load_index()
+    prev_active = index.get("active")
     src_dir = get_project_dir(project_id)
     new_project = create_project(new_name)
     dst_dir = get_project_dir(new_project["id"])
@@ -220,12 +224,20 @@ def duplicate_project(project_id: str, new_name: str) -> dict:
     if os.path.isfile(src_state):
         shutil.copy2(src_state, os.path.join(dst_dir, "session_state.json"))
 
+    # Restore previously active project so duplication doesn't switch context
+    if prev_active and prev_active != new_project["id"]:
+        idx = _load_index()
+        idx["active"] = prev_active
+        _save_index(idx)
+
     return new_project
 
 
 def get_project_stats(project_id: str) -> dict:
     """Get statistics for a project."""
-    import re, csv, glob
+    import re
+    import csv
+    import glob
     proj_dir = get_project_dir(project_id)
     stats = {"runs": 0, "best_map": None, "total_size": "0 KB", "has_export": False}
 
@@ -257,7 +269,8 @@ def get_project_stats(project_id: str) -> dict:
                 except Exception:
                     pass
             if best_map == 0:
-                logs = glob.glob(os.path.join(rd, "train_*.log"))
+                from flashstudio.constants import TRAINING_LOG_GLOB
+                logs = glob.glob(os.path.join(rd, TRAINING_LOG_GLOB))
                 for lf in logs:
                     try:
                         with open(lf, "r", encoding="utf-8", errors="replace") as f:
@@ -298,7 +311,6 @@ def render_project_selector():
         return False  # No projects exist — show creation UI
 
     # Quick selector
-    project_names = {p["id"]: p["name"] for p in projects}
     active_idx = 0
     for i, p in enumerate(projects):
         if p["id"] == active_id:
@@ -324,8 +336,8 @@ def render_project_selector():
 def render_project_manager_page():
     """Full project management page — create, switch, delete, manage projects."""
     st.markdown(
-        "<div style='text-align:center; padding:1rem 0;'>"
-        "<b style='font-size:1.4rem;'>Project Manager</b>"
+        "<div style='text-align:center; padding:0.4rem 0;'>"
+        "<b style='font-size:1.2rem;'>Project Manager</b>"
         "</div>",
         unsafe_allow_html=True,
     )
@@ -336,7 +348,7 @@ def render_project_manager_page():
 
     # ─── Create New Project ───
     with st.container(border=True):
-        st.markdown("### Create New Project")
+        st.markdown("#### Create New Project")
         col_name, col_desc, col_btn = st.columns([2, 3, 1])
         with col_name:
             new_name = st.text_input("Project Name", placeholder="e.g. PPE Detection v2",
@@ -349,7 +361,7 @@ def render_project_manager_page():
             st.markdown("<div style='height:28px'></div>", unsafe_allow_html=True)
             if st.button("Create", type="primary", key="create_project_btn",
                          use_container_width=True, disabled=not new_name):
-                proj = create_project(new_name, new_desc)
+                create_project(new_name, new_desc)
                 st.success(f"Project **{new_name}** created!")
                 _clear_session_for_new_project()
                 st.rerun()
@@ -359,14 +371,12 @@ def render_project_manager_page():
         st.info("No projects yet. Create your first project above to get started.")
         return
 
-    st.markdown(f"### Your Projects ({len(projects)})")
+    st.markdown(f"#### Your Projects ({len(projects)})")
 
     for p in sorted(projects, key=lambda x: x.get("last_modified", ""), reverse=True):
         is_active = p["id"] == active_id
-        proj_dir = get_project_dir(p["id"])
         stats = get_project_stats(p["id"])
 
-        border_style = "border-left: 3px solid #7C3AED;" if is_active else ""
         with st.container(border=True):
             col_info, col_stats, col_actions = st.columns([3, 2, 2])
 

{flashstudio-0.1.3 → flashstudio-0.1.4}/flashstudio/components/sidebar.py

@@ -39,17 +39,17 @@ def render_sidebar():
         logo_b64 = _get_logo_b64()
         if logo_b64:
             st.markdown(
-                f'<div style="display:flex;align-items:center;gap:0.5rem;padding:0.5rem 0.2rem 0.4rem;">'
-                f'<img src="data:image/png;base64,{logo_b64}" style="width:32px;height:32px;">'
+                f'<div style="display:flex;flex-direction:column;align-items:center;gap:0.3rem;padding:0.6rem 0.2rem 0.5rem;">'
+                f'<img src="data:image/png;base64,{logo_b64}" style="width:56px;height:56px;">'
                 f'<b style="font-size:1.1rem;color:{COLOR_TEXT_PRIMARY};">FlashStudio</b>'
                 f'</div>',
                 unsafe_allow_html=True,
             )
         else:
             st.markdown(
-                f'<div style="text-align:center;padding:0.4rem 0 0.3rem;">'
+                f'<div style="display:flex;flex-direction:column;align-items:center;gap:0.2rem;padding:0.5rem 0 0.3rem;">'
                 f'<b style="font-size:1.5rem;">FS</b>'
-                f'<b style="font-size:1rem;color:{COLOR_TEXT_PRIMARY};margin-left:0.3rem;">FlashStudio</b>'
+                f'<b style="font-size:1rem;color:{COLOR_TEXT_PRIMARY};">FlashStudio</b>'
                 f'</div>',
                 unsafe_allow_html=True,
             )
@@ -71,14 +71,15 @@ def render_sidebar():
         st.divider()
 
         gpu = get_gpu_info()
+        from flashstudio.constants import GPU_NAME_TRUNCATE, DATASET_NAME_TRUNCATE
         train_status = st.session_state.get("training_status", "Idle")
         status_color = COLOR_SUCCESS if train_status == "Complete" else (
             COLOR_WARNING if train_status == "Running" else (
             COLOR_ERROR if train_status == "Failed" else COLOR_TEXT_SECONDARY))
         st.markdown(
             f'<div style="font-size:0.75rem;color:{COLOR_TEXT_BODY};padding:0.1rem 0.5rem;line-height:1.9;">'
-            f'GPU: {gpu["name"][:15] if gpu["available"] else "CPU"}<br>'
-            f'Data: {(st.session_state.get("dataset_name") or "—")[:15]}<br>'
+            f'GPU: {gpu["name"][:GPU_NAME_TRUNCATE] if gpu["available"] else "CPU"}<br>'
+            f'Data: {(st.session_state.get("dataset_name") or "—")[:DATASET_NAME_TRUNCATE]}<br>'
             f'Model: {st.session_state.get("model_arch", "Pico").replace("FlashDet-", "")}<br>'
             f'Status: <span style="color:{status_color};font-weight:600;">{train_status[:12]}</span>'
             f'</div>',

{flashstudio-0.1.3 → flashstudio-0.1.4}/flashstudio/components/styles.py

@@ -9,13 +9,33 @@ def inject_custom_css():
     <style>
     @import url('https://fonts.googleapis.com/css2?family=Inter:wght@400;500;600;700&display=swap');
 
-    /* ═══════ GLOBAL ═══════ */
-    .stApp { font-family: 'Inter', -apple-system, sans-serif; }
+    /* ═══════ GLOBAL — NO SCROLL ═══════ */
+    .stApp {
+        font-family: 'Inter', -apple-system, sans-serif;
+        overflow: hidden !important;
+        height: 100vh !important;
+    }
+
+    .main {
+        overflow: hidden !important;
+        height: 100vh !important;
+    }
 
     .main .block-container {
         padding: 0.8rem 1.5rem 0.5rem !important;
         max-width: 1500px;
+        height: calc(100vh - 2rem) !important;
+        overflow: hidden !important;
+    }
+
+    /* Allow internal scroll on specific containers */
+    .scrollable-panel {
+        overflow-y: auto !important;
+        scrollbar-width: thin;
+        scrollbar-color: #D1D5DB transparent;
     }
+    .scrollable-panel::-webkit-scrollbar { width: 4px; }
+    .scrollable-panel::-webkit-scrollbar-thumb { background: #D1D5DB; border-radius: 2px; }
 
     .main .stMarkdown { margin-bottom: 0 !important; }
     .main .element-container { margin-bottom: 0.15rem !important; }
@@ -152,6 +172,17 @@ def inject_custom_css():
         border-bottom: 2px solid #7C3AED !important;
     }
 
+    /* Tab content area: constrain height to prevent page scroll */
+    .stTabs [data-baseweb="tab-panel"] {
+        max-height: calc(100vh - 12rem) !important;
+        overflow-y: auto !important;
+        overflow-x: hidden !important;
+        scrollbar-width: thin;
+        scrollbar-color: #D1D5DB transparent;
+    }
+    .stTabs [data-baseweb="tab-panel"]::-webkit-scrollbar { width: 4px; }
+    .stTabs [data-baseweb="tab-panel"]::-webkit-scrollbar-thumb { background: #D1D5DB; border-radius: 2px; }
+
     /* ═══════ FORM INPUTS ═══════ */
     .main .stSelectbox > div > div,
     .main .stTextInput > div > div > input,
@@ -296,9 +327,15 @@ def render_info_bar(items: dict):
 
 
 def render_page_header(icon: str, title: str, subtitle: str = ""):
+    icon_html = f'<span style="margin-right:0.4rem;">{icon}</span>' if icon else ""
+    subtitle_html = (
+        f'<div style="font-size:0.82rem;color:#6B7280;margin-top:0.1rem;">{subtitle}</div>'
+        if subtitle else ""
+    )
     st.markdown(
         f'<div style="margin-bottom:0.4rem;">'
-        f'<span style="font-size:1.2rem;font-weight:700;color:#1A1A2E;">{title}</span>'
+        f'<span style="font-size:1.2rem;font-weight:700;color:#1A1A2E;">{icon_html}{title}</span>'
+        f'{subtitle_html}'
         f'</div>',
         unsafe_allow_html=True,
     )

{flashstudio-0.1.3 → flashstudio-0.1.4}/flashstudio/constants.py

@@ -124,6 +124,13 @@ VIS_QUEUE_SIZE = 3
 VIS_DIR_NAMES = ("visualizations", "plots", "vis")
 VIS_SKIP_FILE = "latest_visualization.jpg"
 GT_VERIFICATION_DIR = "gt_verification"
+GT_REPORT_FILE = "verification_report.json"
+GT_SUMMARY_FILE = "verification_summary.txt"
+TRAINING_LOG_GLOB = "train_*.log"
+ONNX_MODEL_FILE = "model.onnx"
+ONNX_DATA_FILE = "model.onnx.data"
+CONFIG_YAML_FILE = "config.yaml"
+RESULTS_JSON_FILE = "results.json"
 
 # ════════════════════════════════════════
 # MODEL ARCHITECTURE
@@ -135,7 +142,16 @@ MODEL_SIZE_DEFAULT = "n"
 
 ARCH_FAMILIES = ["FlashDet (recommended)", "YOLOv8", "YOLOv9", "YOLOv10", "YOLOv11", "YOLOX"]
 OPTIMIZERS = ["AdamW", "SGD", "MuSGD"]
+DEFAULT_OPTIMIZER = "AdamW"
 LORA_VARIANTS = ["standard", "dora", "lora_plus", "adalora", "ortho", "lora_fa"]
+LORA_RANK_DEFAULT = 8
+LORA_ALPHA_DEFAULT = 16.0
+LORA_DROPOUT_DEFAULT = 0.05
+LORA_TARGETS_DEFAULT = ["backbone", "fpn"]
+QLORA_DTYPE_DEFAULT = "int8"
+CHUNKED_LOSS_CHUNK_SIZE = 1024
+DEFAULT_FINETUNE_STRATEGY = "Full fine-tune"
+DEFAULT_PRETRAIN_OPTION = "COCO pretrained"
 
 FLASHDET_MODELS = {
     "FlashDet-Pico": {"size": "p", "params": "~298K", "speed": "Ultra-fast", "backbone": "LiteBackbone(0.5x)", "neck": "PicoNeck(64ch)", "head": "E2EDualHead", "for": "Edge/Mobile"},
@@ -159,7 +175,25 @@ ZONE_DEFAULT_POLYGON = "100,100\n500,100\n500,400\n100,400"
 # FONT
 # ════════════════════════════════════════
 
-FONT_PATH_LINUX = "/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf"
+def _resolve_font_path():
+    """Find a usable TrueType font across platforms, with env override."""
+    env = os.environ.get("FLASHSTUDIO_FONT_PATH", "")
+    if env and os.path.isfile(env):
+        return env
+    candidates = [
+        "/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf",       # Debian/Ubuntu
+        "/usr/share/fonts/dejavu-sans-fonts/DejaVuSans-Bold.ttf",      # Fedora/RHEL
+        "/usr/share/fonts/TTF/DejaVuSans-Bold.ttf",                    # Arch
+        "/System/Library/Fonts/Helvetica.ttc",                         # macOS
+        "C:\\Windows\\Fonts\\arial.ttf",                               # Windows
+    ]
+    for c in candidates:
+        if os.path.isfile(c):
+            return c
+    return ""
+
+FONT_PATH = _resolve_font_path()
+FONT_PATH_LINUX = FONT_PATH  # backward compat alias
 FONT_SIZE_DEFAULT = 14
 
 # ════════════════════════════════════════

flashstudio-0.1.4/flashstudio/core/__init__.py

@@ -0,0 +1,3 @@
+"""FlashStudio core package."""
+
+from flashstudio.core.constants import *  # noqa: F401,F403

flashstudio-0.1.4/flashstudio/core/constants.py

@@ -0,0 +1,3 @@
+"""Re-export all constants from the canonical location."""
+
+from flashstudio.constants import *  # noqa: F401,F403

flashstudio-0.1.4/flashstudio/pages/__init__.py

@@ -0,0 +1,8 @@
+"""FlashStudio pages package — each page is a sub-package."""
+
+from flashstudio.pages.dashboard.page import render_dashboard  # noqa: F401
+from flashstudio.pages.data.page import render_data_page  # noqa: F401
+from flashstudio.pages.model.page import render_model_page  # noqa: F401
+from flashstudio.pages.training.page import render_training_page  # noqa: F401
+from flashstudio.pages.export.page import render_export_page  # noqa: F401
+from flashstudio.pages.inference.page import render_inference_page  # noqa: F401

flashstudio-0.1.4/flashstudio/pages/dashboard/__init__.py

@@ -0,0 +1,3 @@
+"""FlashStudio — Dashboard page package."""
+
+from flashstudio.pages.dashboard.page import render_dashboard  # noqa: F401