UI

app.py

@@ -70,15 +70,15 @@ def load_model(workload: str, target: str):
         raise RuntimeError(f"Model not preloaded: {workload}, {target}")
 
 def physical_sanity_check(ipc, miss_rate):
-    warnings_out = []
+    out = []
     if ipc < 0 or ipc > 3.5:
-        warnings_out.append(f"IPC={ipc:.3f} out of physical range")
+        out.append(f"IPC={ipc:.3f} out of physical range")
     if miss_rate < 0 or miss_rate > 1:
-        warnings_out.append(f"L2 miss rate={miss_rate:.3f} out of [0,1]")
-    return warnings_out
+        out.append(f"L2 miss rate={miss_rate:.3f} out of [0,1]")
+    return out
 
 # -------------------------------------------------
-# Preload all models at startup
+# Preload models (runs once at app start)
 # -------------------------------------------------
 def preload_models():
     ensure_models()
@@ -87,16 +87,16 @@ def preload_models():
 
     for workload in workloads:
         for target in TARGETS:
-            model_path = os.path.join(
+            path = os.path.join(
                 MODEL_DIR, f"model_{workload}_{target}.pkl"
             )
-            payload = joblib.load(model_path)
+            payload = joblib.load(path)
             MODEL_CACHE[(workload, target)] = (
                 payload["model"],
                 payload["log_target"],
             )
 
-    return "✅ Models loaded successfully."
+    return "ready"
 
 # -------------------------------------------------
 # Inference Core
@@ -106,7 +106,6 @@ def run_inference(df: pd.DataFrame) -> pd.DataFrame:
     if missing:
         raise ValueError(f"Missing required columns: {missing}")
 
-    # Feature engineering
     for col in [
         "l1d_size",
         "l1i_size",
@@ -173,44 +172,49 @@ def infer_from_csv(file):
 # UI
 # -------------------------------------------------
 with gr.Blocks(title="AIDE Chip Surrogate Inference") as demo:
-    loading_md = gr.Markdown("## ⏳ Loading surrogate models…", visible=True)
-    ready_md = gr.Markdown("## ✅ Models ready", visible=False)
-
-    gr.Markdown(
-        """
-        # AIDE Chip Surrogate Inference
-
-        Upload a CSV describing cache configurations and workloads.
-        The app will run surrogate models to predict:
-        - IPC
-        - L2 Miss Rate
-        """
+    # Loading screen ONLY
+    loading_md = gr.Markdown(
+        "## Downloading surrogate models…\n\nThis may take a while.",
+        visible=True,
     )
 
-    csv_input = gr.File(label="Input CSV", file_types=[".csv"])
-    run_btn = gr.Button("Run Inference", interactive=False)
+    # Main app (hidden initially)
+    with gr.Column(visible=False) as app_ui:
+        gr.Markdown(
+            """
+            # AIDE Chip Surrogate Inference
+
+            Upload a CSV describing cache configurations and workloads.
+            The app will run surrogate models to predict:
+            - IPC
+            - L2 Miss Rate
+            """
+        )
+
+        csv_input = gr.File(label="Input CSV", file_types=[".csv"])
+        run_btn = gr.Button("Run Inference")
+
+        preview = gr.Dataframe(label="Preview (first 20 rows)")
+        output_csv = gr.File(label="Download Full Output CSV")
+        warnings_box = gr.Textbox(label="Sanity Check Summary")
 
-    preview = gr.Dataframe(label="Preview (first 20 rows)")
-    output_csv = gr.File(label="Download Full Output CSV")
-    warnings_box = gr.Textbox(label="Sanity Check Summary")
+        run_btn.click(
+            infer_from_csv,
+            inputs=csv_input,
+            outputs=[preview, output_csv, warnings_box],
+        )
 
+    # Startup load hook
     demo.load(
         preload_models,
         inputs=None,
-        outputs=ready_md,
+        outputs=None,
     ).then(
         lambda: (
             gr.update(visible=False),
             gr.update(visible=True),
-            gr.update(interactive=True),
         ),
-        outputs=[loading_md, ready_md, run_btn],
-    )
-
-    run_btn.click(
-        infer_from_csv,
-        inputs=csv_input,
-        outputs=[preview, output_csv, warnings_box],
+        outputs=[loading_md, app_ui],
     )
 
 if __name__ == "__main__":