fix: Display warning when trainer_state.json is missing for model metrics

app.py

@@ -78,6 +78,7 @@ with gr.Blocks(theme=gr.themes.Monochrome(), title="Multi-Class Classification (
             choices=hf_models,
             value="darklorddad/Model-Swin-Transformer-88"
         )
+        metrics_status = gr.Markdown(visible=False)
         with gr.Column(visible=False) as inf_plots_container:
             with gr.Row():
                 inf_plot_loss = gr.Plot(label="Loss")
@@ -105,12 +106,12 @@ with gr.Blocks(theme=gr.themes.Monochrome(), title="Multi-Class Classification (
         metrics_model_path.change(
             fn=show_model_charts,
             inputs=[metrics_model_path],
-            outputs=inf_plots + [inf_plots_container]
+            outputs=inf_plots + [inf_plots_container, metrics_status]
         )
         demo.load(
             fn=show_model_charts,
             inputs=[metrics_model_path],
-            outputs=inf_plots + [inf_plots_container]
+            outputs=inf_plots + [inf_plots_container, metrics_status]
         )
     with gr.Tab("Data Preparation"):
         with gr.Accordion("Organise Raw Dataset", open=False):

gradio_wrapper.py

@@ -268,7 +268,8 @@ def show_model_charts(model_path):
     and returns metric plots.
     """
     if not model_path:
-        return (None,) * 11 + (gr.update(visible=False),)
+        no_model_msg = "### Please select a model."
+        return (None,) * 11 + (gr.update(visible=False), gr.update(value=no_model_msg, visible=True))
 
     search_path = model_path
     if not os.path.isdir(model_path):
@@ -277,11 +278,13 @@ def show_model_charts(model_path):
             search_path = snapshot_download(repo_id=model_path)
             print(f"Model '{model_path}' downloaded to: {search_path}")
         except RepositoryNotFoundError:
+            msg = f"### ⚠️ Error\n\nHugging Face model repository not found: `{model_path}`"
             print(f"Error: Hugging Face model repository not found: {model_path}")
-            return (None,) * 11 + (gr.update(visible=False),)
+            return (None,) * 11 + (gr.update(visible=False), gr.update(value=msg, visible=True))
         except Exception as e:
+            msg = f"### ⚠️ Error\n\nAn error occurred while downloading the model `{model_path}`: {e}"
             print(f"An error occurred while downloading the model '{model_path}': {e}")
-            return (None,) * 11 + (gr.update(visible=False),)
+            return (None,) * 11 + (gr.update(visible=False), gr.update(value=msg, visible=True))
 
     json_path = None
     for root, _, files in os.walk(search_path):
@@ -290,8 +293,9 @@ def show_model_charts(model_path):
             break
 
     if not json_path:
+        msg = f"### ⚠️ Warning\n\n`trainer_state.json` not found for model `{model_path}`. Cannot display training metrics."
         print(f"trainer_state.json not found in '{search_path}'")
-        return (None,) * 11 + (gr.update(visible=False),)
+        return (None,) * 11 + (gr.update(visible=False), gr.update(value=msg, visible=True))
 
     print(f"Found trainer_state.json at: {json_path}")
     try:
@@ -301,11 +305,12 @@ def show_model_charts(model_path):
             figures.get('Gradient Norm'), figures.get('F1 Scores'), figures.get('Precision'),
             figures.get('Recall'), figures.get('Epoch'), figures.get('Eval Runtime'),
             figures.get('Eval Samples/sec'), figures.get('Eval Steps/sec'),
-            gr.update(visible=True),
+            gr.update(visible=True), gr.update(visible=False)
         )
     except Exception as e:
+        msg = f"### ⚠️ Error\n\nAn error occurred while generating plots for `{model_path}`: {e}"
         print(f"Error generating plots for {json_path}: {e}")
-        return (None,) * 11 + (gr.update(visible=False),)
+        return (None,) * 11 + (gr.update(visible=False), gr.update(value=msg, visible=True))
 
 
 def run_plot_metrics(json_path):