Skip to content

[WIP] ML model controls w/ Logos & Progress#433

Closed
ax3l wants to merge 2 commits into
BLAST-AI-ML:mainfrom
ax3l:topic-mlflow-download-progress
Closed

[WIP] ML model controls w/ Logos & Progress#433
ax3l wants to merge 2 commits into
BLAST-AI-ML:mainfrom
ax3l:topic-mlflow-download-progress

Conversation

@ax3l
Copy link
Copy Markdown
Member

@ax3l ax3l commented May 1, 2026

synapse_dl_progress.webm

Summary

This updates the dashboard model controls with model/vendor branding, MLflow download feedback, and safer handling for missing registered models.

Key changes:

  • Show the Genesis logo for Neural Network (single) in the model type dropdown.
  • Show the AmSC logo next to the model type selector.
  • Use the Genesis logo as the app favicon.
  • Show NN download status/progress while loading from AmSC MLflow.
  • Keep missing MLflow models from leaving stale errors, stale links, or stale plotted
    model traces.
  • Make the AmSC logo link to the selected model’s MLflow page only when that model is
    available.

Review Guide

Recommended review order:

  1. model_manager.py

    • New logo constants and MLflow model URL expression.
    • New helpers:
      • clear_model_load_errors
      • is_missing_mlflow_model
      • load_model_from_mlflow
    • ModelManager(..., loaded_model=...) support.
    • Model panel layout and conditional AmSC link.
  2. app.py

    • update_plot_figure() uses a stable Plotly state key to avoid stale model traces.
    • NN model download wrapper with progress reporting.
    • Async reset() path and guarded state-change handling.
    • Genesis favicon assignment.
  3. state_manager.py

    • New model download/availability state.
  4. logos/

    • Adds only the two assets used by the UI.

Non-Obvious Details

  • MLflow artifact progress is mirrored by temporarily wrapping ArtifactProgressBar in
    the relevant MLflow modules. The original classes are restored in finally.
  • The Plotly figure is replaced through a stable state key because figure.update()
    could retain stale ML model traces when switching between available and missing models.
  • Missing registered models are treated as “model unavailable” rather than fatal UI
    errors.

Manual Test Plan that was Performed

  • Start the app and confirm the Genesis favicon is shown.
  • Confirm Neural Network (single) has the Genesis logo in the dropdown.
  • Confirm the AmSC logo appears beside the model type selector.
  • Select a valid NN model:
    • download status says Downloading from American Science Cloud...
    • progress bar updates
    • AmSC logo opens the selected model page in a new tab
    • ML model line appears in the plot
  • Select a missing model, e.g. missing GP:
    • app does not hard-fail
    • stale error is cleared when switching back to a valid model
    • AmSC logo is not clickable
    • previous ML model line disappears
  • Switch valid -> invalid -> valid repeatedly:
    • plot and AmSC link match the current selection every time

@ax3l ax3l requested review from EZoni and RemiLehe May 1, 2026 23:29
Add model/vendor logos, download progress, and safer MLflow
model switching. Missing models no longer leave stale links,
errors, or plotted model traces behind.
@ax3l ax3l force-pushed the topic-mlflow-download-progress branch from 5daa6d3 to c00fcdf Compare May 1, 2026 23:33
@RemiLehe RemiLehe changed the title ML model controls w/ Logos & Progress [WIP] ML model controls w/ Logos & Progress May 4, 2026
@ax3l ax3l mentioned this pull request May 5, 2026
2 tasks
ax3l added a commit to ax3l/synapse that referenced this pull request May 5, 2026
Simpler version of BLAST-AI-ML#433:

- Show download progress from MLFlow
- Add the Genesis AMSC Logo and make it link to MLFlow model catalogue
ax3l added a commit to ax3l/synapse that referenced this pull request May 5, 2026
Simpler version of BLAST-AI-ML#433:

- Show download progress from MLFlow
- Add the Genesis AMSC Logo and make it link to MLFlow model catalogue
@ax3l
Copy link
Copy Markdown
Member Author

ax3l commented May 5, 2026

Thinning out a bit, moved to #434

@ax3l ax3l closed this May 5, 2026
ax3l added a commit to ax3l/synapse that referenced this pull request May 6, 2026
Simpler version of BLAST-AI-ML#433:

- Show download progress from MLFlow
- Add the Genesis AMSC Logo and make it link to MLFlow model catalogue
ax3l added a commit to ax3l/synapse that referenced this pull request May 6, 2026
Simpler version of BLAST-AI-ML#433:

- Show download progress from MLFlow
- Add the Genesis AMSC Logo and make it link to MLFlow model catalogue
ax3l added a commit to ax3l/synapse that referenced this pull request May 6, 2026
Simpler version of BLAST-AI-ML#433:

- Show download progress from MLFlow
- Add the Genesis AMSC Logo and make it link to MLFlow model catalogue
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant