spinner transformer

streamlit_simulation/app.py

@@ -148,7 +148,8 @@ def init_session_state():
         "true_timestamps": [],
         "pred_timestamps": [],
         "last_fig": None,
-        "valid_pos": 0
+        "valid_pos": 0,
+        "first_plot_shown": False
     }
     for key, value in defaults.items():
         if key not in st.session_state:
@@ -379,6 +380,7 @@ if reset_button:
     st.session_state.last_fig = None
     st.session_state.is_running = False
     st.session_state.valid_pos = 0
+    st.session_state.first_plot_shown = False
     st.rerun()
 
 # Auto-reset on critical parameter change while running
@@ -394,6 +396,7 @@ if st.session_state.is_running and (
     st.session_state.true_timestamps = []
     st.session_state.last_fig = None
     st.session_state.valid_pos = 0
+    st.session_state.first_plot_shown = False
     st.rerun()
 
 # Track current selections for change detection
@@ -478,11 +481,16 @@ if model_choice == "LightGBM" and st.session_state.is_running:
 
 # ============================== Transformer Simulation ==============================
 
+spinner_placeholder = st.empty()
+
 if model_choice == "Transformer Model (moments)":
     if st.session_state.is_running:
         st.write("Simulation started (Transformer)...")
         st.markdown('<div id="simulation"></div>', unsafe_allow_html=True)
 
+        if not st.session_state.first_plot_shown:
+            spinner_placeholder.markdown("Running first prediction – please wait...")
+
         plot_title, plot_container, x_axis_label, info_container = init_simulation_layout()
 
         # Zugriff auf Modell, Dataset, Device
@@ -558,6 +566,10 @@ if model_choice == "Transformer Model (moments)":
             )
             if len(pred_vals) >= 2 and len(true_vals) >= 1:
                 render_simulation_view(current_time, current_pred, actual_val if i >= 1 else None, st.session_state.valid_pos / total_steps, fig)
+                if not st.session_state.first_plot_shown:
+                    spinner_placeholder.empty()
+                    st.session_state.first_plot_shown = True
+
 
             plt.close(fig)  # Speicher freigeben