Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,352 +1,702 @@
 import streamlit as st
-
-# -------------------------------------------------
-# PAGE CONFIG
-# -------------------------------------------------
+import pandas as pd
+import numpy as np
+from sklearn.preprocessing import StandardScaler
+from sklearn.neighbors import KNeighborsRegressor
+
+# ---------------------------
+# Page config
+# ---------------------------
 st.set_page_config(
-    page_title="Tune Your Build",
-    page_icon="🛠️",
-    layout="wide",
+    page_title="Neural Tuner – Car Mod Performance Estimator",
+    page_icon="🔥",
+    layout="wide"
 )
 
-# -------------------------------------------------
-# GLOBAL CONSTANTS
-# -------------------------------------------------
-BASE_HP_MIN = 50
-BASE_HP_MAX = 2000  # keep this high so no preset causes an error
-
-ENGINE_LAYOUTS = ["I4", "I6", "V6", "V8", "V10", "V12"]
-
-PRESETS = {
-    "daily": {
-        "label": "Daily Driver",
-        "emoji": "🚘",
-        "base_hp": 180,
-        "engine_l": 2.0,
-        "layout": "I4",
-        "turbo": False,
-        "intercooler": False,
-        "tune": False,
-        "intake": False,
-        "exhaust": True,
-        "fuel": "Petrol",
-    },
-    "street": {
-        "label": "Street Fun",
-        "emoji": "🏙️",
-        "base_hp": 320,
-        "engine_l": 3.0,
-        "layout": "V6",
-        "turbo": True,
-        "intercooler": False,
-        "tune": True,
-        "intake": True,
-        "exhaust": True,
-        "fuel": "Octane 97",
-    },
-    "drag": {
-        "label": "Drag Strip",
-        "emoji": "🏁",
-        "base_hp": 650,
-        "engine_l": 5.0,
-        "layout": "V8",
-        "turbo": True,
-        "intercooler": True,
-        "tune": True,
-        "intake": True,
-        "exhaust": True,
-        "fuel": "E85",
-    },
-    "max": {
-        "label": "Max Attack",
-        "emoji": "🔥",
-        "base_hp": 1100,
-        "engine_l": 6.0,
-        "layout": "V12",
-        "turbo": True,
-        "intercooler": True,
-        "tune": True,
-        "intake": True,
-        "exhaust": True,
-        "fuel": "E85",
-    },
-}
-
-# -------------------------------------------------
-# CUSTOM CSS
-# -------------------------------------------------
+# ---------------------------
+# Global CSS (mac-neon + rev gauge)
+# ---------------------------
 st.markdown(
     """
-<style>
-/* Page background */
-main.block-container {
-    padding-top: 1.5rem;
-}
-
-/* Top title bar */
-.tune-header {
-    background: radial-gradient(circle at top left, #4c6fff55, #070b20 60%);
-    border-radius: 999px;
-    padding: 0.9rem 1.6rem;
-    border: 1px solid rgba(255,255,255,0.06);
-    display: flex;
-    align-items: center;
-    gap: 0.8rem;
-}
+    <style>
+    html, body, [data-testid="stAppViewContainer"] {
+        background: radial-gradient(circle at top, #020617 0, #020617 35%, #020617 40%, #000000 100%) !important;
+        color: #e5e7eb;
+        font-family: system-ui, -apple-system, BlinkMacSystemFont, "SF Pro Text",
+                     "Segoe UI", Roboto, "Helvetica Neue", Arial, sans-serif;
+    }
+    [data-testid="stHeader"] { background: transparent; }
+    [data-testid="stToolbar"] { display: none; }
+
+    .hero {
+        border-radius: 24px;
+        padding: 18px 22px;
+        background: radial-gradient(circle at top left, rgba(96,165,250,0.35), transparent 55%),
+                    radial-gradient(circle at bottom right, rgba(236,72,153,0.35), transparent 55%),
+                    rgba(15,23,42,0.94);
+        box-shadow: 0 25px 60px rgba(15,23,42,0.9);
+        border: 1px solid rgba(148,163,184,0.35);
+        position: relative;
+        overflow: hidden;
+    }
+    .hero-title {
+        font-size: 1.9rem;
+        font-weight: 700;
+        background: linear-gradient(90deg, #f97316, #facc15, #22c55e, #38bdf8, #a855f7, #f97316);
+        background-size: 400% 100%;
+        -webkit-background-clip: text;
+        color: transparent;
+        animation: moveGradient 9s ease infinite;
+    }
+    .hero-sub {
+        color: #9ca3af;
+        font-size: 0.95rem;
+    }
+    .hero-pill {
+        display: inline-flex;
+        align-items: center;
+        gap: 6px;
+        padding: 3px 11px;
+        border-radius: 999px;
+        background: rgba(15,118,110,0.2);
+        border: 1px solid rgba(34,197,94,0.6);
+        font-size: 0.75rem;
+        color: #bbf7d0;
+        margin-right: 8px;
+    }
+    @keyframes moveGradient {
+        0% { background-position: 0% 50%; }
+        50% { background-position: 100% 50%; }
+        100% { background-position: 0% 50%; }
+    }
+
+    .glass {
+        background: radial-gradient(circle at top left, rgba(148,163,184,0.24), transparent 55%),
+                    rgba(15,23,42,0.96);
+        border-radius: 18px;
+        padding: 18px 18px 14px 18px;
+        border: 1px solid rgba(148,163,184,0.4);
+        box-shadow: 0 20px 40px rgba(15,23,42,0.6);
+    }
+
+    .chip {
+        display:inline-block;
+        padding:4px 10px;
+        margin:3px 4px 3px 0;
+        border-radius:999px;
+        font-size:0.78rem;
+        background:rgba(59,130,246,0.14);
+        border:1px solid rgba(59,130,246,0.45);
+        color:#bfdbfe;
+    }
+
+    .meter-label {
+        font-size: 0.85rem;
+        color: #9ca3af;
+        margin-bottom: 3px;
+    }
+    .muted {
+        font-size: 0.8rem;
+        color: #6b7280;
+    }
+
+    /* Tabs */
+    button[data-baseweb="tab"] {
+        background: transparent !important;
+        border-radius: 999px !important;
+        padding: 0.5rem 1rem !important;
+        margin-right: 0.4rem;
+        color: #9ca3af !important;
+    }
+    button[data-baseweb="tab"][aria-selected="true"] {
+        background: rgba(59,130,246,0.2) !important;
+        color: #e5e7eb !important;
+        box-shadow: 0 0 0 1px rgba(59,130,246,0.7);
+    }
+
+    /* Rev limiter gauge */
+    .rev-wrap {
+        display:flex;
+        justify-content:center;
+        align-items:center;
+        margin-top:4px;
+        margin-bottom:4px;
+    }
+    .rev-gauge {
+        position:relative;
+        width:170px;
+        height:90px;
+        border-radius:170px 170px 14px 14px;
+        background: radial-gradient(circle at 50% 120%, #991b1b, transparent 60%),
+                    radial-gradient(circle at 0% 0%, rgba(59,130,246,0.4), transparent 60%),
+                    #020617;
+        overflow:hidden;
+        border:1px solid rgba(148,163,184,0.5);
+        box-shadow: 0 12px 30px rgba(0,0,0,0.8);
+    }
+    .rev-arc {
+        position:absolute;
+        inset:10px 10px auto 10px;
+        height:62px;
+        border-radius:999px;
+        border-top:6px solid rgba(249,250,251,0.16);
+        border-left:6px solid rgba(34,197,94,0.75);
+        border-right:6px solid rgba(220,38,38,0.9);
+        border-bottom:none;
+    }
+    .rev-needle {
+        --base-angle: -40deg;
+        position:absolute;
+        left:50%;
+        bottom:8px;
+        width:3px;
+        height:68px;
+        background: linear-gradient(to top, #fecaca, #f97316);
+        border-radius:999px;
+        transform-origin:50% 90%;
+        animation: revBounce 0.6s ease-in-out infinite alternate;
+    }
+    .rev-center {
+        position:absolute;
+        left:50%;
+        bottom:4px;
+        transform:translateX(-50%);
+        width:22px;
+        height:22px;
+        border-radius:999px;
+        background:#020617;
+        border:2px solid #9ca3af;
+        display:flex;
+        align-items:center;
+        justify-content:center;
+        font-size:0.55rem;
+        color:#9ca3af;
+    }
+    .rev-label {
+        position:absolute;
+        right:8px;
+        bottom:6px;
+        font-size:0.7rem;
+        color:#fecaca;
+    }
+    @keyframes revBounce {
+        from { transform: rotate(var(--base-angle)); }
+        to   { transform: rotate(calc(var(--base-angle) + 14deg)); }
+    }
+    </style>
+    """,
+    unsafe_allow_html=True
+)
 
-/* Preset buttons row */
-.preset-row {
-    margin-top: 1.0rem;
-    margin-bottom: 1.0rem;
-    display: flex;
-    flex-wrap: wrap;
-    gap: 0.6rem;
-}
+# ---------------------------
+# Data & model
+# ---------------------------
+def generate_dataset(n=400):
+    np.random.seed(42)
+    data = []
+    for _ in range(n):
+        engine = np.random.choice([1.6, 2.0, 2.5, 3.0, 3.5, 5.0])
+        cyl = np.random.choice([4, 6, 8])
+        base_hp = int(engine * cyl * np.random.uniform(18, 22))
+
+        intake = np.random.choice([0, 1, 2])
+        exhaust = np.random.choice([0, 1, 2])
+        induction = np.random.choice([0, 1, 2])
+        fuel = np.random.choice([0, 1, 2, 3])
+        tune = np.random.choice([0, 1, 2])
+        altitude = np.random.uniform(0, 2000)
+
+        hp_gain = (
+            intake * np.random.uniform(3, 10) +
+            exhaust * np.random.uniform(5, 20) +
+            induction * np.random.uniform(25, 100) +
+            tune * np.random.uniform(10, 35) +
+            fuel * np.random.uniform(2, 8) -
+            altitude * 0.01 +
+            np.random.uniform(-3, 3)
+        )
 
-/* Tabs container rounded card */
-div[data-testid="stTabs"] {
-    background: rgba(5, 8, 26, 0.96);
-    padding: 1.15rem 1.15rem 0.85rem 1.15rem;
-    border-radius: 22px;
-    border: 1px solid rgba(255,255,255,0.06);
-}
+        data.append([engine, cyl, base_hp, intake, exhaust,
+                     induction, fuel, tune, altitude, hp_gain])
+
+    columns = [
+        "engine", "cyl", "base_hp", "intake", "exhaust",
+        "induction", "fuel", "tune", "altitude", "hp_gain"
+    ]
+    return pd.DataFrame(data, columns=columns)
+
+df = generate_dataset()
+X = df.drop("hp_gain", axis=1)
+y = df["hp_gain"]
+
+scaler = StandardScaler()
+X_scaled = scaler.fit_transform(X)
+
+model = KNeighborsRegressor(n_neighbors=5, weights="distance")
+model.fit(X_scaled, y)
+
+# ---------------------------
+# PRESET CALLBACKS (use session_state)
+# ---------------------------
+def preset_daily():
+    s = st.session_state
+    s.engine_disp = 1.6
+    s.engine_layout = "I4"
+    s.base_hp = 120
+    s.weight_kg = 1350
+    s.weight_reduction = 0
+    s.intake = "Stock"
+    s.headers = "Stock"
+    s.exhaust = "Stock"
+    s.exhaust_dia = 50
+    s.cam = "Stock"
+    s.intake_manifold = "Stock"
+    s.induction = "None"
+    s.boost_psi = 0
+    s.turbo_size = "N/A"
+    s.intercooler = "None"
+    s.meth = False
+    s.tune = "None"
+    s.fuel = "87"
+    s.altitude = 200
+    s.traction_mode = "Daily"
+
+def preset_street():
+    s = st.session_state
+    s.engine_disp = 2.0
+    s.engine_layout = "I4"
+    s.base_hp = 190
+    s.weight_kg = 1400
+    s.weight_reduction = 5
+    s.intake = "Cold Air"
+    s.headers = "Shorty"
+    s.exhaust = "Cat-back"
+    s.exhaust_dia = 60
+    s.cam = "Stage 1 – Road"
+    s.intake_manifold = "High-flow"
+    s.induction = "Turbo"
+    s.boost_psi = 12
+    s.turbo_size = "Small 45-55mm"
+    s.intercooler = "Air-to-Air"
+    s.meth = False
+    s.tune = "Stage 1"
+    s.fuel = "93"
+    s.altitude = 150
+    s.traction_mode = "Spirited"
+
+def preset_drag():
+    s = st.session_state
+    s.engine_disp = 5.0
+    s.engine_layout = "V8"
+    s.base_hp = 420
+    s.weight_kg = 1500
+    s.weight_reduction = 18
+    s.intake = "Performance"
+    s.headers = "Long Tube"
+    s.exhaust = "Straight Pipe"
+    s.exhaust_dia = 90
+    s.cam = "Stage 2 – Aggressive"
+    s.intake_manifold = "High-flow"
+    s.induction = "Twin-Turbo"
+    s.boost_psi = 24
+    s.turbo_size = "Big 66-75mm"
+    s.intercooler = "Front-mount High-Flow"
+    s.meth = True
+    s.tune = "Kill Mode"
+    s.fuel = "E85 / Race Blend"
+    s.altitude = 100
+    s.traction_mode = "Track / Drag"
+
+def preset_max():
+    s = st.session_state
+    s.engine_disp = 6.0
+    s.engine_layout = "V12"
+    s.base_hp = 650
+    s.weight_kg = 1400
+    s.weight_reduction = 25
+    s.intake = "Performance"
+    s.headers = "Long Tube"
+    s.exhaust = "Straight Pipe"
+    s.exhaust_dia = 100
+    s.cam = "Stage 3 – Race"
+    s.intake_manifold = "Individual throttle bodies"
+    s.induction = "Twincharged"
+    s.boost_psi = 30
+    s.turbo_size = "XL 76mm+"
+    s.intercooler = "Front-mount High-Flow"
+    s.meth = True
+    s.tune = "Kill Mode"
+    s.fuel = "E85 / Race Blend"
+    s.altitude = 0
+    s.traction_mode = "Track / Drag"
+
+# ---------------------------
+# HERO
+# ---------------------------
+st.markdown(
+    """
+    <div class="hero">
+        <div style="display:flex;justify-content:space-between;align-items:center;gap:14px;">
+            <div>
+                <div class="hero-pill">⚙️ KNN model + synthetic dyno data</div>
+                <div class="hero-pill">🎛️ Live tuning playground</div>
+                <div class="hero-title">Neural Tuner – Car Mod Performance Lab</div>
+                <p class="hero-sub">
+                    Mash intakes, boost, tunes and fuel in real time. 
+                    Watch horsepower climb and the rev limiter dance. Built to make car nerds go feral. 🐺
+                </p>
+            </div>
+            <div style="
+                width:170px;height:110px;
+                border-radius:22px;
+                background:conic-gradient(from 220deg,
+                    #22c55e, #38bdf8, #a855f7, #f97316, #facc15, #22c55e);
+                padding:2px;
+            ">
+                <div style="
+                    width:100%;height:100%;
+                    border-radius:19px;
+                    background:radial-gradient(circle at 30% 0%, rgba(248,250,252,0.2), transparent 55%),
+                               #020617;">
+                    <div style="display:flex;flex-direction:column;justify-content:center;align-items:center;height:100%;">
+                        <div style="font-size:0.78rem;color:#9ca3af;">Virtual</div>
+                        <div style="font-size:1.6rem;font-weight:700;color:#e5e7eb;">Dyno</div>
+                        <div style="font-size:0.75rem;color:#22c55e;">No hardware, all chaos</div>
+                    </div>
+                </div>
+            </div>
+        </div>
+    </div>
+    """,
+    unsafe_allow_html=True
+)
 
-/* Tab buttons = pills */
-div[data-testid="stTabs"] button[role="tab"] {
-    border-radius: 999px !important;
-    padding: 0.35rem 1.0rem;
-    border: 1px solid rgba(255,255,255,0.14);
-    background-color: rgba(255,255,255,0.02);
-    font-size: 0.85rem;
-}
+st.markdown("")
 
-/* Active tab */
-div[data-testid="stTabs"] button[role="tab"][aria-selected="true"] {
-    background: linear-gradient(135deg, #ff8a00cc, #ff3b6acc);
-    border-color: rgba(255,255,255,0.8);
-}
+# ---------------------------
+# Layout
+# ---------------------------
+left, right = st.columns([1.15, 1])
 
-/* Metric card tweaks */
-[data-testid="stMetricValue"] {
-    font-size: 1.4rem;
-}
-</style>
-""",
-    unsafe_allow_html=True,
-)
+# ========= LEFT SIDE =========
+with left:
+    st.markdown('<div class="glass">', unsafe_allow_html=True)
+    st.subheader("🎛️ Tune Your Build")
 
-# -------------------------------------------------
-# PRESET HANDLING
-# -------------------------------------------------
-if "active_preset" not in st.session_state:
-    st.session_state.active_preset = "daily"
-
-
-def apply_preset(name: str):
-    """Load preset values into session_state."""
-    p = PRESETS[name]
-    st.session_state.active_preset = name
-    st.session_state.engine_l = p["engine_l"]
-    st.session_state.engine_layout = p["layout"]
-    st.session_state.base_hp = min(max(p["base_hp"], BASE_HP_MIN), BASE_HP_MAX)
-    st.session_state.intake = p["intake"]
-    st.session_state.exhaust = p["exhaust"]
-    st.session_state.turbo = p["turbo"]
-    st.session_state.intercooler = p["intercooler"]
-    st.session_state.tune = p["tune"]
-    st.session_state.fuel = p["fuel"]
-
-
-# initialise first time
-if "base_hp" not in st.session_state:
-    apply_preset("daily")
-
-# -------------------------------------------------
-# HEADER
-# -------------------------------------------------
-st.markdown(
-    """
-<div class="tune-header">
-  <span style="font-size:1.4rem;">🧪</span>
-  <div>
-    <div style="font-size:1.1rem; font-weight:600;">Tune Your Build</div>
-    <div style="font-size:0.85rem; opacity:0.8;">Estimate power gains from your mods in real-time.</div>
-  </div>
-</div>
-""",
-    unsafe_allow_html=True,
-)
+    # Preset buttons
+    bcol1, bcol2, bcol3, bcol4 = st.columns(4)
+    bcol1.button("🚙 Daily Driver", on_click=preset_daily)
+    bcol2.button("🌃 Street Fun", on_click=preset_street)
+    bcol3.button("🏁 Drag Strip", on_click=preset_drag)
+    bcol4.button("🔥 Max Attack", on_click=preset_max)
 
-# -------------------------------------------------
-# PRESET BUTTONS
-# -------------------------------------------------
-st.markdown('<div class="preset-row">', unsafe_allow_html=True)
-cols = st.columns(len(PRESETS))
-
-for i, (key, preset) in enumerate(PRESETS.items()):
-    with cols[i]:
-        if st.button(
-            f"{preset['emoji']} {preset['label']}",
-            key=f"preset_{key}",
-            use_container_width=True,
-        ):
-            apply_preset(key)
-            st.toast(f"Loaded {preset['label']} preset {preset['emoji']}")
-st.markdown("</div>", unsafe_allow_html=True)
-
-active_preset_name = PRESETS[st.session_state.active_preset]["label"]
-st.caption(f"Active preset: **{active_preset_name}**")
-
-# -------------------------------------------------
-# TABS (CORE SPECS, BOLT-ONS, BOOST, ECU)
-# -------------------------------------------------
-tab1, tab2, tab3, tab4 = st.tabs(
-    ["⚙️ Core Specs", "🔩 Bolt-Ons", "💨 Boost & Cooling", "🧠 ECU & Fuel"]
-)
+    st.markdown("")
 
-# ---------------- CORE SPECS ----------------
-with tab1:
-    c1, c2, c3 = st.columns([1, 1, 1])
+    tabs = st.tabs(["Core Specs", "Bolt-Ons", "Boost & Cooling", "ECU & Fuel"])
 
-    with c1:
-        st.session_state.engine_l = st.number_input(
+    # --- Core specs ---
+    with tabs[0]:
+        col1, col2 = st.columns(2)
+        engine_disp_options = [0.8,1.0,1.2,1.4,1.6,1.8,2.0,2.2,2.4,2.5,
+                               2.8,3.0,3.2,3.5,4.0,4.4,5.0,6.0,7.0,8.0]
+        engine_disp = col1.selectbox(
             "Engine Displacement (L)",
-            min_value=0.8,
-            max_value=8.0,
-            value=float(st.session_state.get("engine_l", 2.0)),
-            step=0.1,
+            engine_disp_options,
+            index=engine_disp_options.index(2.0),
+            key="engine_disp"
         )
-
-    with c2:
-        st.session_state.engine_layout = st.selectbox(
+        engine_layout = col2.selectbox(
             "Engine Layout",
-            ENGINE_LAYOUTS,
-            index=ENGINE_LAYOUTS.index(
-                st.session_state.get("engine_layout", "I4")
-            ),
+            ["I3","I4","I6","V6","V8","V10","V12"],
+            index=2,
+            key="engine_layout"
         )
 
-    with c3:
-        # clamp base hp to safe range
-        default_hp = int(st.session_state.get("base_hp", 180))
-        default_hp = min(max(default_hp, BASE_HP_MIN), BASE_HP_MAX)
-        st.session_state.base_hp = st.number_input(
-            "Base Horsepower (Stock Dyno)",
-            min_value=BASE_HP_MIN,
-            max_value=BASE_HP_MAX,
-            value=default_hp,
-            step=10,
+        layout_to_cyl = {"I3":3,"I4":4,"I6":6,"V6":6,"V8":8,"V10":10,"V12":12}
+        cyl = layout_to_cyl[engine_layout]
+
+        # FIX: Increased max_value to 3000 to prevent crash on high displacement/cylinder combinations
+        base_hp = st.number_input(
+            "Base Horsepower (stock dyno)",
+            min_value=60, max_value=3000, 
+            value=int(max(90, round(engine_disp * cyl * 20))),
+            key="base_hp"
         )
 
-# ---------------- BOLT-ONS ----------------
-with tab2:
-    c1, c2 = st.columns(2)
-    with c1:
-        st.session_state.intake = st.checkbox(
-            "High-Flow Intake", value=st.session_state.get("intake", False)
+        col3, col4 = st.columns(2)
+        weight_kg = col3.number_input(
+            "Vehicle Weight (kg)",
+            min_value=700, max_value=4000, value=1500,
+            key="weight_kg"
         )
-        st.session_state.exhaust = st.checkbox(
-            "Performance Exhaust", value=st.session_state.get("exhaust", False)
+        weight_reduction = col4.slider(
+            "Weight Reduction (%)",
+            0, 40, 0,
+            key="weight_reduction"
         )
-    with c2:
-        st.write("Typical Gains:")
-        st.write("• Intake: +10–20 hp")
-        st.write("• Exhaust: +20–40 hp")
-
-# ---------------- BOOST & COOLING ----------------
-with tab3:
-    c1, c2 = st.columns(2)
-    with c1:
-        st.session_state.turbo = st.checkbox(
-            "Turbo / Supercharger",
-            value=st.session_state.get("turbo", False),
+
+    # --- Bolt-ons ---
+    with tabs[1]:
+        col1, col2, col3 = st.columns(3)
+        intake = col1.selectbox("Intake", ["Stock", "Cold Air", "Performance"], key="intake")
+        headers = col2.selectbox("Headers", ["Stock", "Shorty", "Long Tube"], key="headers")
+        exhaust = col3.selectbox("Exhaust", ["Stock", "Cat-back", "Straight Pipe"], key="exhaust")
+        exhaust_dia = st.slider("Exhaust Diameter (mm)", 40, 120, 60, key="exhaust_dia")
+
+        cam = st.selectbox(
+            "Cam Profile",
+            ["Stock", "Stage 1 – Road", "Stage 2 – Aggressive", "Stage 3 – Race"],
+            key="cam"
+        )
+        intake_manifold = st.selectbox(
+            "Intake Manifold",
+            ["Stock", "High-flow", "Individual throttle bodies"],
+            key="intake_manifold"
+        )
+
+    # --- Boost & cooling ---
+    with tabs[2]:
+        induction = st.selectbox(
+            "Forced Induction Setup",
+            ["None", "Turbo", "Twin-Turbo", "Supercharger", "Twincharged"],
+            key="induction"
+        )
+        boost_psi = st.slider("Target Boost (psi)", 0, 40, 10, key="boost_psi")
+        turbo_size = st.selectbox(
+            "Turbo Size",
+            ["N/A", "Small 45-55mm", "Medium 56-65mm", "Big 66-75mm", "XL 76mm+"],
+            key="turbo_size"
         )
-        st.session_state.intercooler = st.checkbox(
-            "Upgraded Intercooler",
-            value=st.session_state.get("intercooler", False),
+        intercooler = st.selectbox(
+            "Intercooler Type",
+            ["None", "Air-to-Air", "Air-to-Water", "Front-mount High-Flow"],
+            key="intercooler"
         )
-    with c2:
-        st.write("Boost Tips:")
-        st.write("• Forced induction gives the biggest jump in power.")
-        st.write("• Intercoolers help keep power consistent on long pulls.")
-
-# ---------------- ECU & FUEL ----------------
-with tab4:
-    c1, c2 = st.columns(2)
-    with c1:
-        st.session_state.tune = st.checkbox(
-            "ECU / Remap Tune",
-            value=st.session_state.get("tune", False),
+        meth = st.checkbox("Methanol Injection Kit", value=False, key="meth")
+
+    # --- ECU & fuel ---
+    with tabs[3]:
+        tune = st.selectbox(
+            "ECU Tune Level",
+            ["None", "Mild Street", "Stage 1", "Stage 2", "Kill Mode"],
+            key="tune"
         )
-    with c2:
-        st.session_state.fuel = st.selectbox(
+        fuel = st.selectbox(
             "Fuel Type",
-            ["Petrol", "Octane 97", "E85"],
-            index=["Petrol", "Octane 97", "E85"].index(
-                st.session_state.get("fuel", "Petrol")
-            ),
+            ["87", "91", "93", "E85 / Race Blend"],
+            key="fuel"
+        )
+        altitude = st.slider("Altitude (meters)", 0, 3500, 200, key="altitude")
+        traction_mode = st.radio(
+            "Traction Mode Vibe",
+            ["Daily", "Spirited", "Track / Drag"],
+            horizontal=True,
+            key="traction_mode"
         )
 
-# -------------------------------------------------
-# PERFORMANCE CALCULATION
-# -------------------------------------------------
-hp = float(st.session_state.base_hp)
-
-# bolt-ons
-if st.session_state.intake:
-    hp += 15
-if st.session_state.exhaust:
-    hp += 30
-
-# boost
-if st.session_state.turbo:
-    hp *= 1.40
-if st.session_state.intercooler:
-    hp *= 1.08
-
-# tune
-if st.session_state.tune:
-    hp *= 1.12
-
-# fuel
-fuel = st.session_state.fuel
-if fuel == "Octane 97":
-    hp *= 1.04
-elif fuel == "E85":
-    hp *= 1.16
-
-final_hp = int(hp)
-
-# 0–100 km/h (simple heuristic)
-zero_to_hundred = round(11.5 - (final_hp / 190), 2)
-zero_to_hundred = max(zero_to_hundred, 2.2)
-
-# top speed estimate
-top_speed = int(170 + final_hp / 4.2)
-
-# -------------------------------------------------
-# RESULTS
-# -------------------------------------------------
-st.markdown("---")
-st.subheader("📊 Estimated Performance")
-
-mc1, mc2, mc3 = st.columns(3)
-
-with mc1:
-    st.metric(
-        "Final Horsepower",
-        f"{final_hp} hp",
-        f"+{final_hp - int(st.session_state.base_hp)} hp vs stock",
+    st.markdown(
+        '<p class="muted" style="margin-top:8px;">Numbers are synthetic; this is a dyno-inspired playground, not a tuning bible. 🧪</p>',
+        unsafe_allow_html=True
+    )
+    st.markdown('</div>', unsafe_allow_html=True)
+
+# ========= MODEL & CALCS =========
+
+intake_map = {"Stock":0, "Cold Air":1, "Performance":2}
+exhaust_map = {"Stock":0, "Cat-back":1, "Straight Pipe":2}
+induction_model_map = {"None":0, "Turbo":1, "Twin-Turbo":1, "Supercharger":2, "Twincharged":2}
+fuel_map = {"87":0, "91":1, "93":2, "E85 / Race Blend":3}
+tune_base_map = {"None":0, "Mild Street":1, "Stage 1":1, "Stage 2":2, "Kill Mode":2}
+
+input_for_model = np.array([[
+    engine_disp,
+    cyl,
+    base_hp,
+    intake_map.get(intake,0),
+    exhaust_map.get(exhaust,0),
+    induction_model_map.get(induction,0),
+    fuel_map.get(fuel,0),
+    tune_base_map.get(tune,0),
+    altitude
+]])
+
+input_scaled = scaler.transform(input_for_model)
+pred_base = float(model.predict(input_scaled)[0])
+
+cam_gain_map = {
+    "Stock":0.0,
+    "Stage 1 – Road":6.0,
+    "Stage 2 – Aggressive":12.0,
+    "Stage 3 – Race":20.0
+}
+headers_gain_map = {"Stock":0.0, "Shorty":5.0, "Long Tube":9.0}
+intake_man_gain_map = {
+    "Stock":0.0,
+    "High-flow":5.0,
+    "Individual throttle bodies":10.0
+}
+intercooler_gain_map = {
+    "None":0.0,
+    "Air-to-Air":4.0,
+    "Air-to-Water":6.5,
+    "Front-mount High-Flow":9.0
+}
+turbo_size_map = {
+    "N/A":0.0,
+    "Small 45-55mm":5.0,
+    "Medium 56-65mm":12.0,
+    "Big 66-75mm":20.0,
+    "XL 76mm+":28.0
+}
+
+cam_gain = cam_gain_map.get(cam, 0.0)
+headers_gain = headers_gain_map.get(headers, 0.0)
+intake_man_gain = intake_man_gain_map.get(intake_manifold, 0.0)
+intercooler_gain = intercooler_gain_map.get(intercooler, 0.0)
+turbo_size_gain = turbo_size_map.get(turbo_size, 0.0)
+
+if induction in ["Turbo", "Twin-Turbo"]:
+    boost_gain = boost_psi * 1.8
+elif induction in ["Supercharger", "Twincharged"]:
+    boost_gain = boost_psi * 1.4
+else:
+    boost_gain = 0.0
+
+meth_gain = 12.0 if meth else 0.0
+exhaust_dia_gain = max(0.0, (exhaust_dia - 55) * 0.1)
+
+extra_gain = (
+    cam_gain +
+    headers_gain +
+    intake_man_gain +
+    intercooler_gain +
+    turbo_size_gain +
+    boost_gain * 0.9 +
+    meth_gain +
+    exhaust_dia_gain
+)
+
+traction_multiplier = {"Daily":0.9, "Spirited":1.0, "Track / Drag":1.05}[traction_mode]
+
+pred_total_gain = (pred_base + extra_gain) * traction_multiplier
+pred_total_gain = max(pred_total_gain, -5.0)
+new_hp = max(base_hp + pred_total_gain, 40.0)
+
+effective_weight = weight_kg * (1 - weight_reduction / 100.0)
+hp_per_ton = new_hp / (effective_weight / 1000.0)
+
+hype_raw = np.clip(pred_total_gain / 120 * 100, 0, 100)
+hype_level = int(hype_raw)
+
+if hype_level < 20:
+    verdict = "Sleeper grocery getter 🚙"
+elif hype_level < 40:
+    verdict = "Respectable street build 🌃"
+elif hype_level < 70:
+    verdict = "Serious weekend weapon ⚔️"
+else:
+    verdict = "Full send, tyres cry for mercy 🏁"
+
+# compute base angle for rev gauge (-90deg to +90deg)
+angle_deg = -90 + (hype_level / 100.0) * 180
+
+# ========= RIGHT SIDE =========
+with right:
+    st.markdown('<div class="glass">', unsafe_allow_html=True)
+    st.subheader("💥 Build Outcome")
+
+    m1, m2, m3 = st.columns(3)
+    m1.metric("Estimated HP Gain", f"{pred_total_gain:.1f} HP")
+    m2.metric("New Output", f"{new_hp:.1f} HP")
+    m3.metric("HP per Ton", f"{hp_per_ton:.1f}")
+
+    # Animated rev limiter gauge
+    st.markdown('<div class="meter-label">Rev Limiter Vibes</div>', unsafe_allow_html=True)
+    st.markdown(
+        f"""
+        <div class="rev-wrap">
+          <div class="rev-gauge">
+            <div class="rev-arc"></div>
+            <div class="rev-needle" style="--base-angle:{angle_deg}deg;"></div>
+            <div class="rev-center">RPM</div>
+            <div class="rev-label">REDLINE</div>
+          </div>
+        </div>
+        """,
+        unsafe_allow_html=True
     )
 
-with mc2:
-    st.metric("0–100 km/h", f"{zero_to_hundred} s")
+    st.markdown('<div class="meter-label">Hype Meter (relative craziness of this build)</div>', unsafe_allow_html=True)
+    st.progress(hype_level)
+    st.markdown(f"**Verdict:** {verdict}")
 
-with mc3:
-    st.metric("Estimated Top Speed", f"{top_speed} km/h")
+    st.markdown("")
+    st.bar_chart(
+        pd.DataFrame(
+            {"Horsepower": [base_hp, new_hp]},
+            index=["Stock", "Tuned"]
+        )
+    )
 
-# Small hint
-st.info(
-    "💡 Hint: For big gains, combine a turbo/supercharger with a good intercooler and a proper ECU tune."
-)
+    st.markdown(
+        f"""
+        <div style="margin-top:6px;">
+            <span class="chip">{engine_disp}L {engine_layout}</span>
+            <span class="chip">Weight: {weight_kg} kg ▸ −{weight_reduction}%</span>
+            <span class="chip">Intake: {intake}</span>
+            <span class="chip">Headers: {headers}</span>
+            <span class="chip">Exhaust: {exhaust} ({exhaust_dia}mm)</span><br/>
+            <span class="chip">Induction: {induction} @ {boost_psi} psi</span>
+            <span class="chip">IC: {intercooler}</span>
+            <span class="chip">Tune: {tune}</span>
+            <span class="chip">Fuel: {fuel}</span>
+        </div>
+        """,
+        unsafe_allow_html=True
+    )
+
+    st.markdown("---")
+
+    breakdown = pd.DataFrame({
+        "Component": [
+            "Model base (from dataset)",
+            "Cam profile",
+            "Headers",
+            "Intake manifold",
+            "Intercooler",
+            "Turbo size",
+            "Boost (net)",
+            "Meth kit",
+            "Exhaust diameter tweak"
+        ],
+        "Approx HP": [
+            pred_base,
+            cam_gain,
+            headers_gain,
+            intake_man_gain,
+            intercooler_gain,
+            turbo_size_gain,
+            boost_gain * 0.9,
+            meth_gain,
+            exhaust_dia_gain
+        ]
+    })
+
+    st.caption("🔬 Where is that extra power coming from?")
+    st.dataframe(breakdown, use_container_width=True, height=260)
+
+    st.markdown(
+        '<p class="muted" style="margin-top:8px;">Model: distance-weighted KNN on synthetic dyno-style data + heuristic math for advanced mods.</p>',
+        unsafe_allow_html=True
+    )
+    st.markdown('</div>', unsafe_allow_html=True)
 
+# ========= FOOTER =========
+st.markdown(
+    """
+    <div style="text-align:center;margin-top:10px;" class="muted">
+        Presets + rev limiter + neon dyno… if this doesn’t make people go mad, we can push it even further. 🔧🔥
+    </div>
+    """,
+    unsafe_allow_html=True
+)