Update src/streamlit_app.py

src/streamlit_app.py

@@ -14,7 +14,7 @@ st.set_page_config(
 )
 
 # ---------------------------
-# Global CSS – crazy but clean
+# Global CSS (mac-neon + rev gauge)
 # ---------------------------
 st.markdown(
     """
@@ -25,8 +25,6 @@ st.markdown(
         font-family: system-ui, -apple-system, BlinkMacSystemFont, "SF Pro Text",
                      "Segoe UI", Roboto, "Helvetica Neue", Arial, sans-serif;
     }
-
-    /* Hide default Streamlit header/menu */
     [data-testid="stHeader"] { background: transparent; }
     [data-testid="stToolbar"] { display: none; }
 
@@ -41,7 +39,6 @@ st.markdown(
         position: relative;
         overflow: hidden;
     }
-
     .hero-title {
         font-size: 1.9rem;
         font-weight: 700;
@@ -51,12 +48,10 @@ st.markdown(
         color: transparent;
         animation: moveGradient 9s ease infinite;
     }
-
     .hero-sub {
         color: #9ca3af;
         font-size: 0.95rem;
     }
-
     .hero-pill {
         display: inline-flex;
         align-items: center;
@@ -69,7 +64,6 @@ st.markdown(
         color: #bbf7d0;
         margin-right: 8px;
     }
-
     @keyframes moveGradient {
         0% { background-position: 0% 50%; }
         50% { background-position: 100% 50%; }
@@ -101,7 +95,6 @@ st.markdown(
         color: #9ca3af;
         margin-bottom: 3px;
     }
-
     .muted {
         font-size: 0.8rem;
         color: #6b7280;
@@ -120,13 +113,83 @@ st.markdown(
         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
 )
 
 # ---------------------------
-# Data & model (same logic)
+# Data & model
 # ---------------------------
 def generate_dataset(n=400):
     np.random.seed(42)
@@ -173,18 +236,114 @@ model = KNeighborsRegressor(n_neighbors=5, weights="distance")
 model.fit(X_scaled, y)
 
 # ---------------------------
-# HERO SECTION
+# 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">⚙️ Powered by KNN + synthetic dyno data</div>
-                <div class="hero-title">Neural Tuner – Live Car Mod Performance Lab</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">
-                    Mix intakes, turbos, tunes & fuel in real time. Watch the HP jump, 
-                    the power-to-weight spike and your virtual build go absolutely feral. 🐺
+                    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="
@@ -200,9 +359,9 @@ st.markdown(
                     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;">Live Build</div>
-                        <div style="font-size:1.6rem;font-weight:700;color:#e5e7eb;">HP Lab</div>
-                        <div style="font-size:0.75rem;color:#22c55e;">Realtime estimator</div>
+                        <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>
@@ -215,24 +374,43 @@ st.markdown(
 st.markdown("")
 
 # ---------------------------
-# Layout: left (controls) / right (dashboard)
+# Layout
 # ---------------------------
 left, right = st.columns([1.15, 1])
 
-# ===== LEFT: TUNING CONTROLS =====
+# ========= LEFT SIDE =========
 with left:
     st.markdown('<div class="glass">', unsafe_allow_html=True)
     st.subheader("🎛️ Tune Your Build")
 
+    # 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)
+
+    st.markdown("")
+
     tabs = st.tabs(["Core Specs", "Bolt-Ons", "Boost & Cooling", "ECU & Fuel"])
 
-    # --- Core specs tab ---
+    # --- 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)", engine_disp_options, index=engine_disp_options.index(2.0))
-        engine_layout = col2.selectbox("Engine Layout", ["I3","I4","I6","V6","V8","V10","V12"], index=2)
+        engine_disp = col1.selectbox(
+            "Engine Displacement (L)",
+            engine_disp_options,
+            index=engine_disp_options.index(2.0),
+            key="engine_disp"
+        )
+        engine_layout = col2.selectbox(
+            "Engine Layout",
+            ["I3","I4","I6","V6","V8","V10","V12"],
+            index=2,
+            key="engine_layout"
+        )
 
         layout_to_cyl = {"I3":3,"I4":4,"I6":6,"V6":6,"V8":8,"V10":10,"V12":12}
         cyl = layout_to_cyl[engine_layout]
@@ -240,38 +418,80 @@ with left:
         base_hp = st.number_input(
             "Base Horsepower (stock dyno)",
             min_value=60, max_value=1400,
-            value=int(max(90, round(engine_disp * cyl * 20)))
+            value=int(max(90, round(engine_disp * cyl * 20))),
+            key="base_hp"
         )
 
         col3, col4 = st.columns(2)
-        weight_kg = col3.number_input("Vehicle Weight (kg)", min_value=700, max_value=4000, value=1500)
-        weight_reduction = col4.slider("Weight Reduction (%)", 0, 40, 0)
+        weight_kg = col3.number_input(
+            "Vehicle Weight (kg)",
+            min_value=700, max_value=4000, value=1500,
+            key="weight_kg"
+        )
+        weight_reduction = col4.slider(
+            "Weight Reduction (%)",
+            0, 40, 0,
+            key="weight_reduction"
+        )
 
-    # --- Bolt-ons tab ---
+    # --- Bolt-ons ---
     with tabs[1]:
         col1, col2, col3 = st.columns(3)
-        intake = col1.selectbox("Intake", ["Stock", "Cold Air", "Performance"])
-        headers = col2.selectbox("Headers", ["Stock", "Shorty", "Long Tube"])
-        exhaust = col3.selectbox("Exhaust", ["Stock", "Cat-back", "Straight Pipe"])
-        exhaust_dia = st.slider("Exhaust Diameter (mm)", 40, 120, 60)
-
-        cam = st.selectbox("Cam Profile", ["Stock", "Stage 1 – Road", "Stage 2 – Aggressive", "Stage 3 – Race"])
-        intake_manifold = st.selectbox("Intake Manifold", ["Stock", "High-flow", "Individual throttle bodies"])
+        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 tab ---
+    # --- Boost & cooling ---
     with tabs[2]:
-        induction = st.selectbox("Forced Induction Setup", ["None", "Turbo", "Twin-Turbo", "Supercharger", "Twincharged"])
-        boost_psi = st.slider("Target Boost (psi)", 0, 40, 10)
-        turbo_size = st.selectbox("Turbo Size", ["N/A", "Small 45-55mm", "Medium 56-65mm", "Big 66-75mm", "XL 76mm+"])
-        intercooler = st.selectbox("Intercooler Type", ["None", "Air-to-Air", "Air-to-Water", "Front-mount High-Flow"])
-        meth = st.checkbox("Methanol Injection Kit", value=False)
+        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"
+        )
+        intercooler = st.selectbox(
+            "Intercooler Type",
+            ["None", "Air-to-Air", "Air-to-Water", "Front-mount High-Flow"],
+            key="intercooler"
+        )
+        meth = st.checkbox("Methanol Injection Kit", value=False, key="meth")
 
-    # --- ECU & fuel tab ---
+    # --- ECU & fuel ---
     with tabs[3]:
-        tune = st.selectbox("ECU Tune Level", ["None", "Mild Street", "Stage 1", "Stage 2", "Kill Mode"])
-        fuel = st.selectbox("Fuel Type", ["87", "91", "93", "E85 / Race Blend"])
-        altitude = st.slider("Altitude (meters)", 0, 3500, 200)
-        traction_mode = st.radio("Traction Mode Vibe", ["Daily", "Spirited", "Track / Drag"], horizontal=True)
+        tune = st.selectbox(
+            "ECU Tune Level",
+            ["None", "Mild Street", "Stage 1", "Stage 2", "Kill Mode"],
+            key="tune"
+        )
+        fuel = st.selectbox(
+            "Fuel Type",
+            ["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"
+        )
 
     st.markdown(
         '<p class="muted" style="margin-top:8px;">Numbers are synthetic; this is a dyno-inspired playground, not a tuning bible. 🧪</p>',
@@ -279,9 +499,8 @@ with left:
     )
     st.markdown('</div>', unsafe_allow_html=True)
 
-# ===== MODEL INPUT & CALC =====
+# ========= MODEL & CALCS =========
 
-# maps for model
 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}
@@ -303,7 +522,6 @@ input_for_model = np.array([[
 input_scaled = scaler.transform(input_for_model)
 pred_base = float(model.predict(input_scaled)[0])
 
-# --- heuristic extras ---
 cam_gain_map = {
     "Stock":0.0,
     "Stage 1 – Road":6.0,
@@ -357,21 +575,18 @@ extra_gain = (
     exhaust_dia_gain
 )
 
-# traction / vibe adjusts "usable feel"
 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)  # clamp
+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)
 
-# normalized hype level 0–100
 hype_raw = np.clip(pred_total_gain / 120 * 100, 0, 100)
 hype_level = int(hype_raw)
 
-# verdict text
 if hype_level < 20:
     verdict = "Sleeper grocery getter 🚙"
 elif hype_level < 40:
@@ -381,7 +596,10 @@ elif hype_level < 70:
 else:
     verdict = "Full send, tyres cry for mercy 🏁"
 
-# ===== RIGHT: DASHBOARD =====
+# 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")
@@ -391,13 +609,26 @@ with right:
     m2.metric("New Output", f"{new_hp:.1f} HP")
     m3.metric("HP per Ton", f"{hp_per_ton:.1f}")
 
-    st.markdown("")
+    # 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
+    )
+
     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}")
 
-    # small bar chart: stock vs tuned
     st.markdown("")
     st.bar_chart(
         pd.DataFrame(
@@ -406,7 +637,6 @@ with right:
         )
     )
 
-    # chips summary
     st.markdown(
         f"""
         <div style="margin-top:6px;">
@@ -426,7 +656,6 @@ with right:
 
     st.markdown("---")
 
-    # Contribution breakdown (no Styler.hide_index)
     breakdown = pd.DataFrame({
         "Component": [
             "Model base (from dataset)",
@@ -456,16 +685,16 @@ with right:
     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 + extra heuristic math for advanced mods.</p>',
+        '<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 =====
+# ========= FOOTER =========
 st.markdown(
     """
     <div style="text-align:center;margin-top:10px;" class="muted">
-        Built for fun, learning & ridiculous builds – drop this in a Space and watch car nerds lose it. 🔧🔥
+        Presets + rev limiter + neon dyno… if this doesn’t make people go mad, we can push it even further. 🔧🔥
     </div>
     """,
     unsafe_allow_html=True