Docker + FastAPI backend + v11 HUD for HF Spaces deploy

.dockerignore

@@ -0,0 +1,33 @@
+.git/
+__pycache__/
+*.pyc
+.ipynb_checkpoints/
+notebooks/
+
+# Project notes and personal vaults
+ATAS_Notes/
+
+# Local python environment
+env/
+
+# Large data folder (keep metadata CSV tracked)
+data/
+!data/aircraft_metadata.csv
+
+# Logs, artifacts and model binaries
+logs/
+models/
+release_models/
+!release_models/aircraft_classifier/
+!release_models/aircraft_classifier/atas_final_fine_tuned_aircraft_classifier_model.keras
+!release_models/eta/
+!release_models/eta/atas_final_eta_regressor_model.joblib
+!release_models/hit/
+!release_models/hit/atas_final_hit_classifier_model.joblib
+
+# Prediction outputs and training history
+predictions/
+model_history/
+
+# VS Code workspace config
+.vscode/

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.keras filter=lfs diff=lfs merge=lfs -text

Dockerfile

@@ -0,0 +1,11 @@
+FROM python:3.11-slim
+
+WORKDIR /app
+
+COPY . .
+
+RUN pip install --no-cache-dir -r requirements.txt
+
+EXPOSE 7860
+
+CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]

app/main.py

@@ -0,0 +1,160 @@
+"""
+main.py
+-----------------------------------------------------
+# WHAT: FastAPI application entry point for the ATAS backend.
+#       Exposes a single endpoint - POST /analyze - that accepts an aircraft
+#       image and engagement parameters, runs the full pipeline, and returns
+#       a JSON response.
+#
+# WHY:  All five src/ modules are isolated units. main.py is the glue layer
+#       that wires them together in the correct order and exposes the result
+#       to the outside world via HTTP.
+#
+# USE:  Called by the HUD frontend via fetch(). Also used directly during
+#       testing via curl or a Python test script to verify end-to-end pipeline
+#       behavior before any UI work begins.
+
+# User controls / HUD inputs: + 1 Image
+# your_speed, your_altitude, enemy_altitude, countermeasure_deployed,
+# launch_distance, remaining_distance, azimuth, elevation
+
+# Comes from aircraft_metadata.csv:
+# missile_speed, missile_range, enemy_generation, maneuverability
+
+# Derived by build_feature_array() internally:
+# closure_rate, missile_phase
+"""
+
+
+
+# Importing the necessary libraries
+from fastapi import FastAPI
+from fastapi import UploadFile, File, Form
+from fastapi.middleware.cors import CORSMiddleware
+import tempfile
+import os
+from src.classifier import predict_aircraft
+from src.metadata import get_aircraft_metadata
+from src.physics_generator import build_feature_array
+from src.models import make_predictions
+from src.decision import get_recommendation
+
+
+# Creating a server
+app = FastAPI()
+
+
+# Allow the browser to send requests to this server
+# Without this, the HUD's fetch() call gets blocked by the browser's security rules
+# CORS (Cross-Origin Resource Sharing) is the mechanism where your server tells the browser: "yes, I allow requests from other origins."
+# allow_origins=["*"]  → accept requests from any origin (local file, any URL)
+# allow_methods=["*"]  → accept any HTTP method (POST, GET, etc.)
+# allow_headers=["*"]  → accept any headers the browser sends with the request
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+
+# Function to take the inputs
+# image -> as temp file
+# 8 features -> form object
+@app.post("/analyze")
+async def analyze(
+    image: UploadFile = File(...),             # The ... just means required. No default value.
+    your_speed: float = Form(...),             # name : type = where_it_comes_from(required_or_default) 
+    your_altitude: float = Form(...),
+    enemy_altitude: float = Form(...),
+    countermeasure_deployed: int = Form(...),
+    launch_distance: float = Form(...),
+    remaining_distance: float = Form(...),
+    azimuth: float = Form(...),
+    elevation: float = Form(...),
+    friendly_aircraft: str = Form(...)
+):
+    
+    # Browser sends image as raw bytes, not a file path
+    # predict_aircraft() needs a real file path on disk
+    # So we write the bytes to a temp file and grab its path
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as tmp:
+        tmp.write(await image.read())  # dump bytes from memory to disk
+        tmp_path = tmp.name            # grab the path e.g. /tmp/abc123.jpg
+        
+        
+    # 1. Calling the function to predict the aircraft
+    aircraft_name = predict_aircraft(image_path=tmp_path)
+    
+    # 2. Get the metadata of predicted aircraft
+    metadata_dict = get_aircraft_metadata(aircraft_name=aircraft_name)
+    friendly_metadata = get_aircraft_metadata(aircraft_name=friendly_aircraft)
+    
+    # 3. Build the features to feed the models
+    feature_dict = build_feature_array(
+        # From HUD sliders
+        launch_distance=launch_distance,
+        remaining_distance=remaining_distance,
+        azimuth=azimuth,
+        elevation=elevation,
+        your_altitude=your_altitude,
+        your_speed=your_speed,
+        enemy_altitude=enemy_altitude,
+        countermeasure_deployed=countermeasure_deployed,
+        # From metadata lookup (enemy aircraft)
+        missile_speed=metadata_dict['missile_speed'],
+        missile_range=metadata_dict['missile_range'],
+        enemy_generation=metadata_dict['enemy_generation'],
+        # From metadata lookup (friendly aircraft)
+        your_maneuverability=friendly_metadata['maneuverability']
+    )
+    
+    # 4. Make predictions using models
+    predictions_dict = make_predictions(feature_dict=feature_dict)
+    
+    # 5. Providing recommendations to the pilot
+    suggestion = get_recommendation(
+        eta_seconds=predictions_dict['eta_seconds'],
+        hit_probability=predictions_dict['hit_probability'],
+        no_aa_capability=metadata_dict['no_aa_capability'],
+        enemy_generation=metadata_dict['enemy_generation'],
+        friendly_generation=friendly_metadata['enemy_generation']
+    )
+    
+    # 6. Delete the temparary file since work is done
+    os.remove(tmp_path)
+    
+    # 7. Return the value predicted and derived
+    return {
+        'aircraft_name' : aircraft_name,
+        'missile_speed' : metadata_dict['missile_speed'],
+        'missile_range' : metadata_dict['missile_range'],
+        'enemy_generation' : metadata_dict['enemy_generation'],
+        'maneuverability'  : metadata_dict['maneuverability'],
+        'no_aa_capability' : metadata_dict['no_aa_capability'],
+        'eta_seconds'     : predictions_dict['eta_seconds'],
+        'hit_probability' : predictions_dict['hit_probability'],
+        'recommendation'  : suggestion
+    }
+    
+    
+    
+"""
+Request comes in
+      ↓
+Image saved to temp file
+      ↓
+predict_aircraft()     → aircraft_name
+      ↓
+get_aircraft_metadata() x 2  → enemy + friendly metadata
+      ↓
+build_feature_array()  → 14 feature dict
+      ↓
+make_predictions()     → eta + hit probability
+      ↓
+get_recommendation()   → tactical suggestion
+      ↓
+Temp file deleted
+      ↓
+JSON returned to HUD
+"""

app/test_pipeline.py

@@ -0,0 +1,94 @@
+"""
+test_pipeline.py -- End-to-End Pipeline Verification Script
+
+WHAT:
+    Sends a real POST request to the running FastAPI server at /analyze.
+    Uses a real aircraft image and hardcoded engagement parameters.
+
+WHY:
+    Confirms the full pipeline runs without errors before any UI work begins.
+    If this script returns clean JSON with all expected keys, the backend is ready.
+    If it fails, we fix the backend here, not inside the HUD.
+
+HOW TO USE:
+    1. Start the FastAPI server in one terminal:
+       uvicorn app.main:app --reload
+
+    2. Run this script in a second terminal:
+       python app/test_pipeline.py
+
+EXPECTED OUTPUT:
+    A JSON response containing:
+    aircraft_name, missile_speed, missile_range, enemy_generation,
+    maneuverability, no_aa_capability, eta_seconds, hit_probability, recommendation
+"""
+
+
+# Import the necessary libraries
+import requests
+from pathlib import Path
+
+# Define image test path
+IMAGE_PATH = str(Path(__file__).parent.parent / 'assets'/ 'testing_images' / 'Su57.jpg')
+print(IMAGE_PATH)
+
+# Define Testing parameters
+params = {
+    "your_speed": 500.0,
+    "your_altitude": 10000.0,
+    "enemy_altitude": 8000.0,
+    "countermeasure_deployed": 0,
+    "launch_distance": 15000.0,
+    "remaining_distance": 10000.0,
+    "azimuth": 45.0,
+    "elevation": 5.0,
+    "friendly_aircraft": "Tejas"
+}
+
+
+# Send the data to the server
+# and 
+# store what the server sends back
+response = requests.post(
+    url='http://localhost:8000/analyze',   # Address of the running FastAPI server
+    data=params,                           # Form fields - speed, altitude, etc.
+    files={
+        "image": (                         # "image" matches the field name in main.py
+            "Su57.jpg",                    # Filename the server will see
+            open(IMAGE_PATH, "rb"),        # Open the image in read-binary mode
+            "image/jpeg"                   # Tell the server what type of file this is
+        )
+    }
+)
+
+print(response.status_code)   # should be 200 if everything worked
+print(response.json())        # the actual pipeline output
+
+
+
+
+"""
+VERIFIED OUTPUT - June 8, 2026
+End-to-end pipeline confirmed working. Status 200.
+
+Test image : assets/Su57.jpg
+Server     : uvicorn app.main:app (no --reload)
+Run with   : python -m app.test_pipeline
+
+Response:
+{
+    "aircraft_name"   : "Su57",
+    "missile_speed"   : 2058,
+    "missile_range"   : 400000,
+    "enemy_generation": 5.0,
+    "maneuverability" : 2,
+    "no_aa_capability": 0,
+    "eta_seconds"     : 3.77,
+    "hit_probability" : 0.969,
+    "recommendation"  : "BREAK HARD + DEPLOY CM + DISENGAGE IMMEDIATELY"
+}
+
+All 6 src/ modules executed without error.
+Aircraft correctly identified from image.
+Recommendation matches expected threat profile for Gen-5 enemy at 10km.
+"""

data/aircraft_metadata.csv

@@ -0,0 +1,103 @@
+aircraft,aircraft_max_speed,missile_speed,missile_range,enemy_generation,maneuverability,no_aa_capability
+A10,222,857,35000,4,1,0
+A400M,255,-1,-1,4,0,1
+AG600,155,-1,-1,4,0,1
+AH64,101,750,8000,4,1,0
+AKINCI,100,1372,65000,4,1,0
+AV8B,300,1372,160000,4,1,0
+An124,240,-1,-1,4,0,1
+An22,205,-1,-1,3.5,0,1
+An225,236,-1,-1,4,0,1
+An72,196,-1,-1,4,0,1
+B1,370,-1,-1,4,0,1
+B2,280,-1,-1,4.5,0,1
+B21,277,-1,-1,5,0,1
+B52,290,-1,-1,3.5,0,1
+Be200,194,-1,-1,4,0,1
+C1,224,-1,-1,3.5,0,1
+C130,186,-1,-1,3.5,0,1
+C17,264,-1,-1,4,0,1
+C2,255,-1,-1,4,0,1
+C390,274,-1,-1,4.5,0,1
+C5,237,-1,-1,3.5,0,1
+CH47,87,-1,-1,3.5,0,1
+CH53,87,-1,-1,3.5,0,1
+CL415,100,-1,-1,4,0,1
+E2,180,-1,-1,4,0,1
+E7,265,-1,-1,4.5,0,1
+EF2000,590,1372,200000,4.5,2,0
+EMB314,164,857,35000,4,1,0
+F117,289,-1,-1,4,0,1
+F14,690,1475,190000,4,2,0
+F15,737,1372,160000,4,2,0
+F16,590,1372,160000,4,2,0
+F18,532,1372,160000,4.5,2,0
+F2,590,1544,105000,4.5,2,0
+F22,669,1372,160000,5,2,0
+F35,544,1372,160000,5,2,0
+F4,658,1372,50000,3.5,1,0
+FCK1,532,2058,100000,4,2,0
+H6,292,-1,-1,3.5,0,1
+Il76,250,-1,-1,3.5,0,1
+J10,650,1715,300000,4.5,2,0
+J20,590,1715,300000,5,2,0
+J35,590,1715,300000,5,2,0
+J36,590,1715,400000,5,2,0
+J50,590,1715,400000,5,2,0
+JAS39,590,1372,200000,4.5,2,0
+JF17,544,1372,100000,4.5,2,0
+JH7,502,1372,100000,4,1,0
+KAAN,532,1372,65000,5,2,0
+KC135,259,-1,-1,3.5,0,1
+KF21,532,1372,200000,4.5,2,0
+KIZILELMA,305,1372,65000,4.5,2,0
+KJ600,166,-1,-1,4.5,0,1
+Ka27,75,-1,-1,4,1,1
+Ka52,87,857,40000,4,1,0
+MQ20,205,-1,-1,4.5,1,1
+MQ25,172,-1,-1,5,0,1
+MQ28,277,1372,160000,5,1,0
+MQ9,134,857,35000,4,1,0
+Mi24,93,847,8000,3.5,1,0
+Mi26,82,-1,-1,4,0,1
+Mi28,89,857,40000,4,1,0
+Mi8,78,-1,-1,3.5,0,1
+Mig29,667,1372,110000,4,2,0
+Mig31,833,2058,400000,4,1,0
+Mirage2000,649,1372,80000,4,2,0
+NH90,83,-1,-1,4,1,1
+P3,208,-1,-1,3.5,0,1
+RQ4,174,-1,-1,4,0,1
+Rafale,531,1372,200000,4.5,2,0
+SR71,983,-1,-1,3.5,0,1
+Su24,444,847,8000,3.5,1,0
+Su25,271,857,40000,3.5,1,0
+Su34,527,1372,110000,4,1,0
+Su47,611,1372,110000,4.5,2,0
+Su57,590,2058,400000,5,2,0
+T50,510,857,18000,4,1,0
+TB001,78,-1,-1,4,1,1
+TB2,61,-1,-1,4,1,1
+Tejas,550,1544,110000,4.5,2,0
+Tornado,667,1372,160000,4,1,0
+Tu160,616,-1,-1,4,0,1
+Tu22M,638,-1,-1,3.5,0,1
+Tu95,255,-1,-1,3.5,0,1
+U2,224,-1,-1,3.5,0,1
+UH60,99,-1,-1,4,1,1
+US2,155,-1,-1,4,0,1
+V22,157,-1,-1,4,0,1
+V280,154,-1,-1,5,1,1
+Vulcan,288,-1,-1,3.5,0,1
+WZ10,83,686,8000,4,1,0
+WZ7,208,-1,-1,4,0,1
+WZ9,85,686,8000,4,1,0
+X29,536,857,35000,4,2,0
+X32,536,-1,-1,5,2,1
+XB70,919,-1,-1,3.5,0,1
+XQ58,292,1372,160000,5,1,0
+Y20,255,-1,-1,4,0,1
+YF23,648,1372,160000,5,2,0
+Z10,83,686,8000,4,1,0
+Z19,78,686,8000,4,1,0
+Z21,83,686,8000,4,1,0

frontend/atas_hud_v11.html

@@ -0,0 +1,776 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+<meta charset="UTF-8">
+<meta name="viewport" content="width=device-width, initial-scale=1.0">
+<title>ATAS v2.5 Tactical MFD</title>
+<style>
+@import url('https://fonts.googleapis.com/css2?family=IBM+Plex+Mono:wght@400;600;700&display=swap');
+
+*, *::before, *::after { box-sizing: border-box; }
+body { background: #000; margin: 0; padding: 20px; display: flex; justify-content: center; align-items: flex-start; min-height: 100vh; font-family: 'IBM Plex Mono', monospace; }
+
+/* ── Animations ── */
+.hb { animation: hblink 1.6s ease-in-out infinite; }
+@keyframes hblink { 0%,100%{opacity:1} 50%{opacity:.3} }
+@keyframes scan { 0%{top:0} 100%{top:100%} }
+@keyframes spin { 100%{transform:rotate(360deg)} }
+@keyframes critPulse { 0%,100%{box-shadow:0 0 0 0 rgba(232,68,68,0)} 50%{box-shadow:0 0 18px 6px rgba(232,68,68,.25)} }
+
+.scan-line {
+  width:100%; height:2px; background:rgba(0,212,176,.9); position:absolute;
+  box-shadow:0 0 14px #00d4b0; animation:scan 1.4s linear infinite; z-index:5; display:none;
+}
+.rotor-spin { transform-origin:60px 45px; animation:spin .15s linear infinite; }
+.threat-critical { animation:critPulse 1s ease-in-out infinite; }
+
+/* ── Reusable classes ── */
+.bc { position:absolute; width:15px; height:15px; }
+.bc-tl { top:-1px; left:-1px; border-top:2px solid #00d4b0; border-left:2px solid #00d4b0; }
+.bc-tr { top:-1px; right:-1px; border-top:2px solid #00d4b0; border-right:2px solid #00d4b0; }
+.bc-bl { bottom:-1px; left:-1px; border-bottom:2px solid #00d4b0; border-left:2px solid #00d4b0; }
+.bc-br { bottom:-1px; right:-1px; border-bottom:2px solid #00d4b0; border-right:2px solid #00d4b0; }
+
+.panel {
+  border:1px solid rgba(0,212,176,.2); background:rgba(0,212,176,.025);
+  padding:11px; position:relative;
+}
+.panel-label {
+  font-size:9px; color:rgba(0,212,176,.4); letter-spacing:.2em;
+  margin-bottom:8px; text-transform:uppercase;
+}
+.data-row {
+  display:flex; justify-content:space-between; align-items:center;
+  border-bottom:1px solid rgba(0,212,176,.08); padding-bottom:4px; margin-bottom:4px;
+  font-size:9px;
+}
+.data-row:last-child { border-bottom:none; margin-bottom:0; }
+.data-key { color:rgba(0,212,176,.45); letter-spacing:.05em; }
+.data-val { color:#00d4b0; font-size:10px; font-weight:600; }
+
+.tac-slider {
+  -webkit-appearance:none; width:100%; height:2px;
+  background:rgba(0,212,176,.2); outline:none; margin:8px 0; display:block;
+}
+.tac-slider::-webkit-slider-thumb {
+  -webkit-appearance:none; width:8px; height:14px; background:#00d4b0;
+  cursor:pointer; border-radius:1px; box-shadow:0 0 5px rgba(0,212,176,.8);
+}
+.tac-select {
+  background:rgba(0,212,176,.05); border:1px solid rgba(0,212,176,.3);
+  color:#00d4b0; font-family:'IBM Plex Mono',monospace; font-size:10px;
+  padding:4px; outline:none; width:100%; cursor:pointer;
+}
+.tac-select option { background:#06070b; color:#00d4b0; }
+
+.tactical-feed {
+  width:100%; height:100%; object-fit:cover;
+  filter:grayscale(100%) sepia(100%) hue-rotate(130deg) saturate(300%) contrast(1.2);
+  opacity:.7; display:none; position:absolute; top:0; left:0; z-index:1;
+}
+
+/* Countermeasure toggle and Action Buttons */
+.cm-btn {
+  cursor:pointer; border:1px solid rgba(0,212,176,.3);
+  background:rgba(0,212,176,.04); color:rgba(0,212,176,.6);
+  font-family:'IBM Plex Mono',monospace; font-size:9px; letter-spacing:.1em;
+  padding:8px 8px; width:100%; text-align:center; transition:all .2s;
+}
+.cm-btn:hover { background:rgba(0,212,176,.1); border-color:#00d4b0; color:#00d4b0; }
+.cm-btn.deployed { border-color:#f0a030; background:rgba(240,160,48,.1); color:#f0a030; }
+.cm-btn.deployed:hover { background:rgba(240,160,48,.2); }
+
+.clear-btn {
+  margin-top:6px; border:1px solid rgba(0,212,176,.15); padding:4px 8px;
+  text-align:center; font-size:8px; color:rgba(0,212,176,.4); letter-spacing:.1em;
+  cursor:pointer; display:none; transition:all .2s;
+}
+.clear-btn:hover { border-color:rgba(0,212,176,.4); color:rgba(0,212,176,.7); }
+</style>
+</head>
+<body>
+
+<div style="background:#020203;padding:10px;border-radius:12px;box-shadow:inset 0 0 40px rgba(0,0,0,.8);width:880px;">
+<div id="main-hud" style="position:relative;background:#06070b;border:1px solid rgba(0,212,176,.1);padding:20px;">
+
+  <div class="bc" style="top:5px;left:5px;border-top:3px solid #00d4b0;border-left:3px solid #00d4b0;"></div>
+  <div class="bc" style="top:5px;right:5px;border-top:3px solid #00d4b0;border-right:3px solid #00d4b0;"></div>
+  <div class="bc" style="bottom:5px;left:5px;border-bottom:3px solid #00d4b0;border-left:3px solid #00d4b0;"></div>
+  <div class="bc" style="bottom:5px;right:5px;border-bottom:3px solid #00d4b0;border-right:3px solid #00d4b0;"></div>
+
+  <div style="display:flex;justify-content:space-between;align-items:center;border-bottom:1px solid rgba(0,212,176,.12);padding-bottom:12px;margin-bottom:14px;">
+    <div style="display:flex;align-items:baseline;gap:8px;">
+      <span style="font-size:18px;font-weight:700;letter-spacing:.3em;color:#00d4b0;">ATAS</span>
+      <span style="font-size:9px;color:rgba(0,212,176,.35);letter-spacing:.12em;">v2.5 // TACTICAL MFD PIPELINE</span>
+    </div>
+    <span id="header-alert" style="font-size:10px;color:rgba(0,212,176,.3);letter-spacing:.2em;font-weight:700;">SCANNING AIRSPACE</span>
+    <div style="font-size:9px;display:flex;gap:10px;">
+      <span style="color:#2dd4a0;">● UPLINK SECURE</span>
+      <span style="color:rgba(0,212,176,.4);">SYS.TIME: <span id="sys-time">00:00:00 IST</span></span>
+    </div>
+  </div>
+
+  <div style="display:grid;grid-template-columns:1fr 1.55fr 1fr;gap:14px;margin-bottom:12px;">
+
+    <div style="display:flex;flex-direction:column;gap:12px;">
+
+      <div class="panel">
+        <div class="bc bc-tl"></div><div class="bc bc-tr"></div>
+        <div class="bc bc-bl"></div><div class="bc bc-br"></div>
+        <div class="panel-label" style="display:flex;justify-content:space-between;">
+          <span>LIVE CAMERA FEED</span>
+          <span id="sensor-status" style="color:rgba(0,212,176,.3);">STANDBY</span>
+        </div>
+        <div id="sensor-feed"
+          style="border:1px dashed rgba(0,212,176,.28);background:rgba(0,0,0,.6);height:108px;display:flex;align-items:center;justify-content:center;position:relative;overflow:hidden;cursor:pointer;"
+          onclick="document.getElementById('file-upload').click()">
+          <input type="file" id="file-upload" accept="image/*" style="display:none;" onchange="handleImageUpload(event)">
+          <div class="scan-line" id="scan-line"></div>
+          <img id="sensor-img" class="tactical-feed" alt="sensor feed">
+          <div id="sensor-text" style="font-size:9px;color:rgba(0,212,176,.38);letter-spacing:.1em;text-align:center;z-index:2;position:relative;">
+            <svg width="22" height="22" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" style="margin-bottom:5px;opacity:.55;display:block;margin-left:auto;margin-right:auto;">
+              <rect x="3" y="3" width="18" height="18" rx="2"/><circle cx="8.5" cy="8.5" r="1.5"/><polyline points="21 15 16 10 5 21"/>
+            </svg>
+            UPLOAD TARGET PHOTO
+          </div>
+          <div style="position:absolute;top:50%;left:50%;width:22px;height:1px;background:rgba(0,212,176,.25);transform:translate(-50%,-50%);z-index:3;"></div>
+          <div style="position:absolute;top:50%;left:50%;width:1px;height:22px;background:rgba(0,212,176,.25);transform:translate(-50%,-50%);z-index:3;"></div>
+        </div>
+        <div id="clear-btn" class="clear-btn" onclick="clearTarget()">↺ CLEAR TARGET LOCK</div>
+      </div>
+
+      <div class="panel" style="border-color:rgba(0,212,176,.28);background:rgba(0,212,176,.04);">
+        <div class="bc bc-tl"></div><div class="bc bc-tr"></div>
+        <div class="bc bc-bl"></div><div class="bc bc-br"></div>
+        <div style="font-size:9px;color:#00d4b0;letter-spacing:.15em;font-weight:700;margin-bottom:10px;">FRIENDLY PLATFORM</div>
+        <div style="display:flex;gap:10px;align-items:center;margin-bottom:10px;">
+          <svg width="36" height="36" viewBox="0 0 100 100" style="opacity:.8;flex-shrink:0;">
+            <polygon points="50,10 55,30 90,65 90,75 60,70 60,90 70,100 30,100 40,90 40,70 10,75 10,65 45,30" fill="none" stroke="#00d4b0" stroke-width="2"/>
+            <line x1="50" y1="10" x2="50" y2="90" stroke="rgba(0,212,176,.3)" stroke-width="1"/>
+          </svg>
+          <select id="own-platform-select" class="tac-select" style="flex:1;" onchange="updateOwnPlatform()"></select>
+        </div>
+        <div style="font-size:9px;color:rgba(0,212,176,.6);">
+          <div style="display:flex;justify-content:space-between;margin-bottom:2px;">
+            <span>AIRSPEED (MACH)</span>
+            <span>MAX <span id="max-spd-label">1.8</span> | <span id="val-spd" style="color:#00d4b0;font-weight:700;">1.20</span></span>
+          </div>
+          <input type="range" class="tac-slider" id="slide-spd" min="0.5" max="1.8" step="0.01" value="1.20" oninput="updateOwnSliders()">
+          <div style="display:flex;justify-content:space-between;margin-bottom:2px;">
+            <span>ALTITUDE (FT)</span>
+            <span>Max <span id="max-alt-label">60k</span> | <span id="val-alt" style="color:#00d4b0;font-weight:700;">32,000</span></span>
+          </div>
+          <input type="range" class="tac-slider" id="slide-alt" min="5000" max="60000" step="500" value="32000" oninput="updateOwnSliders()">
+        </div>
+        <div style="display:grid;grid-template-columns:1fr 1fr;gap:5px;font-size:8px;margin-top:4px;">
+          <div style="background:rgba(0,0,0,.3);padding:4px;border-left:1px solid rgba(0,212,176,.35);">
+            <div style="color:rgba(0,212,176,.4);">GENERATION</div>
+            <div id="own-gen-label" style="color:#2dd4a0;">4.5</div>
+          </div>
+          <div style="background:rgba(0,0,0,.3);padding:4px;border-left:1px solid rgba(0,212,176,.35);">
+            <div style="color:rgba(0,212,176,.4);">MANEUVER</div>
+            <div id="own-man-label" style="color:#2dd4a0;">HIGH (2)</div>
+          </div>
+        </div>
+
+        <div id="start-btn" class="cm-btn" style="margin-top:12px; border-color:#00d4b0; color:#00d4b0; font-weight:bold; letter-spacing:0.15em;" onclick="startSimulation()">▶ START SIMULATION</div>
+      </div>
+    </div>
+
+    <div class="panel" style="display:flex;flex-direction:column;">
+      <div class="bc bc-tl"></div><div class="bc bc-tr"></div>
+      <div class="bc bc-bl"></div><div class="bc bc-br"></div>
+
+      <div style="display:flex;justify-content:space-between;align-items:center;margin-bottom:6px;">
+        <div style="font-size:10px;color:#00d4b0;letter-spacing:.2em;font-weight:700;">TARGET CLASSIFIER</div>
+        <span style="font-size:7px;color:rgba(0,212,176,.35);background:rgba(0,212,176,.05);padding:2px 6px;border:1px solid rgba(0,212,176,.18);letter-spacing:.07em;">EFFICIENTNETV2-L · 78.08%</span>
+      </div>
+
+      <div style="flex:1;display:flex;flex-direction:column;align-items:center;justify-content:center;border:1px solid rgba(0,212,176,.05);background:rgba(0,0,0,.3);padding:10px;position:relative;overflow:hidden;">
+        <svg width="100%" height="100%" style="position:absolute;top:0;left:0;opacity:.07;pointer-events:none;">
+          <line x1="50%" y1="0" x2="50%" y2="100%" stroke="#00d4b0" stroke-width="1"/>
+          <line x1="0" y1="50%" x2="100%" y2="50%" stroke="#00d4b0" stroke-width="1"/>
+          <circle cx="50%" cy="50%" r="45" fill="none" stroke="#00d4b0" stroke-width="1"/>
+          <circle cx="50%" cy="50%" r="85" fill="none" stroke="#00d4b0" stroke-width="1" stroke-dasharray="2 4"/>
+        </svg>
+
+        <svg id="target-svg" width="158" height="158" viewBox="0 0 120 120"
+          style="margin-bottom:8px;z-index:1;position:relative;filter:drop-shadow(0 0 6px currentColor);flex-shrink:0;"></svg>
+        <div id="target-info" style="text-align:center;width:100%;z-index:1;position:relative;"></div>
+
+        <div id="true-specs-box" style="display:none;width:100%;margin-top:8px;border-top:1px dashed rgba(255,255,255,.1);padding-top:8px;z-index:1;position:relative;">
+          <div style="font-size:8px;color:rgba(255,255,255,.3);margin-bottom:5px;letter-spacing:.1em;">THREAT METADATA // LOOKUP TABLE</div>
+          <div style="display:grid;grid-template-columns:repeat(3,1fr);gap:3px;font-size:9px;">
+            <div style="padding:4px;background:rgba(0,0,0,.45);border-left:1px solid #444;">
+              <span style="color:rgba(255,255,255,.28);font-size:7px;display:block;">MANEUVER</span>
+              <span id="e-man" style="color:#ccc;">---</span>
+            </div>
+            <div style="padding:4px;background:rgba(0,0,0,.45);border-left:1px solid #444;">
+              <span style="color:rgba(255,255,255,.28);font-size:7px;display:block;">ALTITUDE</span>
+              <span id="e-alt" style="color:#ccc;">---</span>
+            </div>
+            <div style="padding:4px;background:rgba(0,0,0,.45);border-left:1px solid #444;">
+              <span style="color:rgba(255,255,255,.28);font-size:7px;display:block;">GENERATION</span>
+              <span id="e-gen" style="color:#ccc;">---</span>
+            </div>
+            <div style="padding:4px;background:rgba(0,0,0,.45);border-left:1px solid #444;">
+              <span style="color:rgba(255,255,255,.28);font-size:7px;display:block;">MSL RANGE</span>
+              <span id="e-rng" style="color:#ccc;">---</span>
+            </div>
+            <div style="padding:4px;background:rgba(0,0,0,.45);border-left:1px solid #444;">
+              <span style="color:rgba(255,255,255,.28);font-size:7px;display:block;">MSL SPEED</span>
+              <span id="e-mspd" style="color:#ccc;">---</span>
+            </div>
+            <div style="padding:4px;background:rgba(0,0,0,.45);border-left:1px solid #444;">
+              <span style="color:rgba(255,255,255,.28);font-size:7px;display:block;">AA CAPABLE</span>
+              <span id="e-aa" style="color:#ccc;">---</span>
+            </div>
+          </div>
+          <div id="gen-adv-box" style="margin-top:4px;padding:4px 7px;border:1px solid rgba(0,212,176,.18);font-size:8px;display:flex;justify-content:space-between;align-items:center;">
+            <span style="color:rgba(0,212,176,.4);letter-spacing:.08em;">GENERATION ADVANTAGE</span>
+            <span id="gen-adv-val" style="font-weight:700;color:#00d4b0;">--</span>
+          </div>
+        </div>
+      </div>
+    </div>
+
+    <div class="panel" style="display:flex;flex-direction:column;align-items:center;">
+      <div class="bc bc-tl"></div><div class="bc bc-tr"></div>
+      <div class="bc bc-bl"></div><div class="bc bc-br"></div>
+
+      <div class="panel-label" style="width:100%;text-align:center;margin-bottom:10px;">THREAT KINEMATICS</div>
+
+      <svg width="138" height="138" viewBox="0 0 132 132" style="flex-shrink:0;">
+        <circle cx="66" cy="66" r="58" fill="none" stroke="rgba(0,212,176,.05)" stroke-width="1" stroke-dasharray="2,4"/>
+        <circle cx="66" cy="66" r="52" fill="none" stroke="rgba(0,212,176,.1)" stroke-width="4"/>
+        <circle id="threat-arc" cx="66" cy="66" r="52" fill="none" stroke="#00d4b0" stroke-width="5"
+          stroke-dasharray="0 326" transform="rotate(-90 66 66)" style="transition:stroke-dasharray .5s ease-out,stroke .3s;"/>
+        <circle cx="66" cy="66" r="44" fill="none" stroke="rgba(0,212,176,.09)" stroke-width="9"/>
+        <circle cx="66" cy="66" r="35" fill="rgba(4,4,9,.92)" stroke="rgba(0,212,176,.14)" stroke-width=".8"/>
+        <line x1="66" y1="5" x2="66" y2="14" stroke="rgba(0,212,176,.4)" stroke-width="1"/>
+        <line x1="127" y1="66" x2="118" y2="66" stroke="rgba(0,212,176,.4)" stroke-width="1"/>
+        <line x1="66" y1="127" x2="66" y2="118" stroke="rgba(0,212,176,.4)" stroke-width="1"/>
+        <line x1="5" y1="66" x2="14" y2="66" stroke="rgba(0,212,176,.4)" stroke-width="1"/>
+        <text id="threat-val" x="66" y="59" text-anchor="middle" fill="#00d4b0"
+          font-family="'IBM Plex Mono',monospace" font-size="22" font-weight="700">0%</text>
+        <text id="threat-label" x="66" y="71" text-anchor="middle" fill="rgba(0,212,176,.5)"
+          font-family="'IBM Plex Mono',monospace" font-size="8" letter-spacing="2">SAFE</text>
+        <text x="66" y="81" text-anchor="middle" fill="rgba(0,212,176,.3)"
+          font-family="'IBM Plex Mono',monospace" font-size="5.5" letter-spacing="0">HIT PROBABILITY</text>
+      </svg>
+
+      <div style="width:100%;margin-top:10px;display:flex;flex-direction:column;gap:0;">
+        <div class="data-row">
+          <span class="data-key">EVASION ETA</span>
+          <span id="eta-val" class="data-val">N/A</span>
+        </div>
+        <div class="data-row">
+          <span class="data-key">CLOSURE RATE</span>
+          <span id="close-val" class="data-val">0 M/S</span>
+        </div>
+        <div class="data-row">
+          <span class="data-key">DIST TO TGT</span>
+          <span id="dist-val" class="data-val">---</span>
+        </div>
+        <div class="data-row">
+          <span class="data-key">ASPECT ANGLE</span>
+          <span id="aspect-val" class="data-val">---</span>
+        </div>
+        <div class="data-row" style="border-bottom:1px solid rgba(0,212,176,.08);padding-bottom:4px;margin-bottom:8px;">
+          <span class="data-key">MSL PHASE</span>
+          <span id="phase-val" class="data-val">---</span>
+        </div>
+      </div>
+
+      <div id="cm-toggle" class="cm-btn" onclick="toggleCM()">
+        CM: <span id="cm-status">NOT DEPLOYED</span>
+      </div>
+    </div>
+  </div>
+
+  <div style="display:flex;gap:10px;margin-bottom:8px;align-items:stretch;min-height:42px;">
+    <div id="tac-banner" style="flex:1;border:1px solid rgba(0,212,176,.2);background:rgba(0,212,176,.05);padding:10px 14px;display:flex;align-items:center;justify-content:space-between;transition:all .3s;">
+      <span style="font-size:9px;color:rgba(0,212,176,.45);letter-spacing:.15em;white-space:nowrap;flex-shrink:0;">SYS ACTION</span>
+      <span id="tac-text" style="font-size:12px;font-weight:700;letter-spacing:.15em;color:#00d4b0;text-align:center;flex:1;">CONTINUE SCANNING SECTOR</span>
+      <span id="conf-text" style="font-size:9px;color:rgba(0,212,176,.45);white-space:nowrap;flex-shrink:0;">SYS NORMAL</span>
+    </div>
+  </div>
+
+  <div style="display:flex;justify-content:space-between;font-size:8px;color:rgba(0,212,176,.22);letter-spacing:.05em;">
+    <span>CLASSIFIER: EFFICIENTNETV2-L · TOP-1 78.08% · TOP-5 92.02%</span>
+    <span>ETA: XGBOOST · R²=0.9939 · MAE=0.46s</span>
+    <span>HIT CLASSIFIER: XGBOOST · F1=0.9968</span>
+    <span>SYSTEM: ATAS PIPELINE ONLINE</span>
+  </div>
+
+</div>
+</div>
+
+<script>
+// ──────────────────────────────────────────────
+// CLOCK (Indian Standard Time)
+// ──────────────────────────────────────────────
+setInterval(() => {
+  const now = new Date();
+  const options = { timeZone: 'Asia/Kolkata', hour12: false, hour: '2-digit', minute: '2-digit', second: '2-digit' };
+  document.getElementById('sys-time').innerText = now.toLocaleTimeString('en-IN', options) + ' IST';
+}, 1000);
+
+// ──────────────────────────────────────────────
+// ENEMY DATABASE (For Left Column UI population)
+// ──────────────────────────────────────────────
+const enemyDB = {
+  "A10":         {aircraft_max_speed:222,  enemy_generation:4.0, maneuverability:1},
+  "A400M":       {aircraft_max_speed:255,  enemy_generation:4.0, maneuverability:0},
+  "AG600":       {aircraft_max_speed:155,  enemy_generation:4.0, maneuverability:0},
+  "AH64":        {aircraft_max_speed:101,  enemy_generation:4.0, maneuverability:1},
+  "B21":         {aircraft_max_speed:277,  enemy_generation:5.0, maneuverability:0},
+  "F16":         {aircraft_max_speed:590,  enemy_generation:4.0, maneuverability:2},
+  "F22":         {aircraft_max_speed:669,  enemy_generation:5.0, maneuverability:2},
+  "Il76":        {aircraft_max_speed:250,  enemy_generation:3.5, maneuverability:0},
+  "J20":         {aircraft_max_speed:590,  enemy_generation:5.0, maneuverability:2},
+  "MQ9":         {aircraft_max_speed:134,  enemy_generation:4.0, maneuverability:1},
+  "Rafale":      {aircraft_max_speed:531,  enemy_generation:4.5, maneuverability:2},
+  "WZ9":         {aircraft_max_speed:85,   enemy_generation:4.0, maneuverability:1},
+  "Su57":        {aircraft_max_speed:590,  enemy_generation:5.0, maneuverability:2}
+};
+
+const fullList = [
+  'A10','A400M','AG600','AH64','AKINCI','AV8B','An124','An22','An225','An72',
+  'B1','B2','B21','B52','Be200','C1','C130','C17','C2','C390','C5','CH47','CH53',
+  'CL415','E2','E7','EF2000','EMB314','F117','F14','F15','F16','F18','F2','F22',
+  'F35','F4','FCK1','H6','Il76','J10','J20','J35','J36','J50','JAS39','JF17',
+  'JH7','KAAN','KC135','KF21','KIZILELMA','KJ600','Ka27','Ka52','MQ20','MQ25',
+  'MQ28','MQ9','Mi24','Mi26','Mi28','Mi8','Mig29','Mig31','Mirage2000','NH90',
+  'P3','RQ4','Rafale','SR71','Su24','Su25','Su34','Su47','Su57','T50','TB001',
+  'TB2','Tejas','Tornado','Tu160','Tu22M','Tu95','U2','UH60','US2','V22','V280',
+  'Vulcan','WZ10','WZ7','WZ9','X29','X32','XB70','XQ58','Y20','YF23','Z10','Z19','Z21'
+];
+fullList.forEach(ac => {
+  if (!enemyDB[ac]) enemyDB[ac] = {
+    aircraft_max_speed:400, enemy_generation:4.0, maneuverability:1
+  };
+});
+
+let currentOwn = {};
+const ownSelect = document.getElementById('own-platform-select');
+fullList.forEach(ac => {
+  const opt = document.createElement('option');
+  opt.value = ac;
+  opt.innerText = ac + ' (GEN ' + enemyDB[ac].enemy_generation.toFixed(1) + ')';
+  ownSelect.appendChild(opt);
+});
+ownSelect.value = 'Tejas';
+
+// ──────────────────────────────────────────────
+// STATE & TOGGLES
+// ──────────────────────────────────────────────
+let cmDeployed = false;
+let isUI_Locked = false;
+let uploadedFile = null;
+
+function toggleCM() {
+  if (isUI_Locked) return;
+  cmDeployed = !cmDeployed;
+  const btn = document.getElementById('cm-toggle');
+  document.getElementById('cm-status').innerText = cmDeployed ? 'DEPLOYED' : 'NOT DEPLOYED';
+  cmDeployed ? btn.classList.add('deployed') : btn.classList.remove('deployed');
+}
+
+function updateOwnPlatform() {
+  if (isUI_Locked) return;
+  const key  = document.getElementById('own-platform-select').value;
+  const data = enemyDB[key];
+  const cat  = getCategory(key);
+  const machMax = parseFloat(Math.max(0.8, (data.aircraft_max_speed / 340)).toFixed(2));
+  const altMax  = cat === 'fighter' ? 60000 : (cat === 'helo' ? 20000 : 42000);
+
+  currentOwn = {
+    max_spd: machMax, max_alt: altMax,
+    gen: data.enemy_generation, man: data.maneuverability,
+    name_man: data.maneuverability >= 2 ? 'HIGH (' + data.maneuverability + ')'
+            : (data.maneuverability === 1 ? 'MED (1)' : 'LOW (0)')
+  };
+
+  document.getElementById('max-spd-label').innerText = currentOwn.max_spd;
+  document.getElementById('max-alt-label').innerText = (currentOwn.max_alt / 1000) + 'k';
+  document.getElementById('own-gen-label').innerText = currentOwn.gen.toFixed(1);
+  document.getElementById('own-man-label').innerText = currentOwn.name_man;
+  document.getElementById('slide-spd').max = currentOwn.max_spd;
+  document.getElementById('slide-alt').max = currentOwn.max_alt;
+  
+  if (parseFloat(document.getElementById('slide-spd').value) > currentOwn.max_spd)
+    document.getElementById('slide-spd').value = currentOwn.max_spd;
+  if (parseFloat(document.getElementById('slide-alt').value) > currentOwn.max_alt)
+    document.getElementById('slide-alt').value = currentOwn.max_alt;
+    
+  updateOwnSliders();
+}
+
+function updateOwnSliders() {
+  if (isUI_Locked) return;
+  document.getElementById('val-spd').innerText = parseFloat(document.getElementById('slide-spd').value).toFixed(2);
+  document.getElementById('val-alt').innerText = parseInt(document.getElementById('slide-alt').value).toLocaleString();
+}
+
+function setLockState(locked) {
+  isUI_Locked = locked;
+  document.getElementById('slide-spd').disabled = locked;
+  document.getElementById('slide-alt').disabled = locked;
+  document.getElementById('own-platform-select').disabled = locked;
+  
+  const opacityVal = locked ? '0.3' : '1';
+  const pointerVal = locked ? 'none' : 'auto';
+  
+  document.getElementById('start-btn').style.opacity = opacityVal;
+  document.getElementById('start-btn').style.pointerEvents = pointerVal;
+  
+  document.getElementById('cm-toggle').style.opacity = opacityVal;
+  document.getElementById('cm-toggle').style.pointerEvents = pointerVal;
+  
+  document.getElementById('sensor-feed').style.pointerEvents = pointerVal;
+}
+
+// ────────��─────────────────────────────────────
+// AIRCRAFT BLUEPRINTS (SVG) & CATEGORIES
+// ──────────────────────────────────────────────
+const premiumSVGs = {
+  none: `
+    <circle cx="60" cy="60" r="50" fill="none" stroke="rgba(0,212,176,0.08)" stroke-width="1"/>
+    <circle cx="60" cy="60" r="25" fill="none" stroke="rgba(0,212,176,0.05)" stroke-width="1" stroke-dasharray="3 5"/>
+    <line x1="60" y1="10" x2="60" y2="110" stroke="rgba(0,212,176,0.12)" stroke-width="1"/>
+    <line x1="10" y1="60" x2="110" y2="60" stroke="rgba(0,212,176,0.12)" stroke-width="1"/>
+    <circle cx="60" cy="60" r="3" fill="#00d4b0" opacity="0.5"/>`,
+
+  fighter: `
+    <g transform="translate(0,5)">
+      <polygon points="60,30 20,70 20,85 45,80 60,85" fill="rgba(232,68,68,0.1)" stroke="currentColor" stroke-width="1.2"/>
+      <polygon points="60,30 100,70 100,85 75,80 60,85" fill="rgba(232,68,68,0.1)" stroke="currentColor" stroke-width="1.2"/>
+      <path d="M60,10 L65,30 L70,70 L68,100 L52,100 L50,70 L55,30 Z" fill="rgba(232,68,68,0.15)" stroke="currentColor" stroke-width="1.5"/>
+      <path d="M58,25 L62,25 L63,40 L57,40 Z" fill="none" stroke="currentColor" stroke-width="0.8" opacity="0.8"/>
+      <line x1="60" y1="40" x2="60" y2="100" stroke="currentColor" stroke-width="1" opacity="0.3"/>
+      <rect x="53" y="95" width="4" height="10" fill="rgba(0,0,0,0.8)" stroke="currentColor" stroke-width="1"/>
+      <rect x="63" y="95" width="4" height="10" fill="rgba(0,0,0,0.8)" stroke="currentColor" stroke-width="1"/>
+      <polygon points="52,80 40,105 48,105 54,90" fill="rgba(232,68,68,0.2)" stroke="currentColor" stroke-width="1"/>
+      <polygon points="68,80 80,105 72,105 66,90" fill="rgba(232,68,68,0.2)" stroke="currentColor" stroke-width="1"/>
+    </g>`,
+
+  bomber: `
+    <g transform="translate(0,15)">
+      <path d="M60,10 L110,60 L90,75 L75,65 L60,75 L45,65 L30,75 L10,60 Z"
+        fill="rgba(232,68,68,0.15)" stroke="currentColor" stroke-width="1.5"/>
+      <path d="M55,25 L65,25 L62,30 L58,30 Z" fill="none" stroke="currentColor" stroke-width="0.8" opacity="0.8"/>
+      <line x1="50" y1="60" x2="50" y2="70" stroke="currentColor" stroke-width="1" opacity="0.5"/>
+      <line x1="70" y1="60" x2="70" y2="70" stroke="currentColor" stroke-width="1" opacity="0.5"/>
+      <path d="M45,65 Q60,55 75,65" fill="none" stroke="currentColor" stroke-width="1" stroke-dasharray="2,2" opacity="0.6"/>
+    </g>`,
+
+  helo: `
+    <g transform="translate(0,5)">
+      <g class="rotor-spin">
+        <circle cx="60" cy="45" r="45" fill="rgba(240,160,48,0.04)" stroke="currentColor" stroke-width="0.5" stroke-dasharray="2 6"/>
+        <line x1="60" y1="2" x2="60" y2="88" stroke="currentColor" stroke-width="1.5" opacity="0.7"/>
+        <line x1="17" y1="45" x2="103" y2="45" stroke="currentColor" stroke-width="1.5" opacity="0.7"/>
+        <circle cx="60" cy="45" r="3" fill="currentColor" opacity="0.9"/>
+      </g>
+      <path d="M60,15 C63,15 65,20 65,35 L62,65 L58,65 L55,35 C55,20 57,15 60,15 Z"
+        fill="rgba(240,160,48,0.2)" stroke="currentColor" stroke-width="1.5"/>
+      <rect x="42" y="42" width="13" height="4" fill="rgba(240,160,48,0.1)" stroke="currentColor" stroke-width="1"/>
+      <rect x="65" y="42" width="13" height="4" fill="rgba(240,160,48,0.1)" stroke="currentColor" stroke-width="1"/>
+      <polygon points="58,65 62,65 61,95 59,95" fill="rgba(240,160,48,0.15)" stroke="currentColor" stroke-width="1.2"/>
+      <polygon points="50,88 70,88 70,92 50,92" fill="rgba(240,160,48,0.1)" stroke="currentColor" stroke-width="1"/>
+      <g style="transform-origin:60px 98px;" class="rotor-spin">
+        <line x1="52" y1="98" x2="68" y2="98" stroke="currentColor" stroke-width="1"/>
+        <line x1="60" y1="90" x2="60" y2="106" stroke="currentColor" stroke-width="1"/>
+      </g>
+    </g>`,
+
+  cargo: `
+    <g transform="translate(0,5)">
+      <defs>
+        <filter id="eGlow" x="-50%" y="-50%" width="200%" height="200%">
+          <feGaussianBlur stdDeviation="1.5" result="b"/>
+          <feComposite in="SourceGraphic" in2="b" operator="over"/>
+        </filter>
+      </defs>
+      <path d="M54,40 L10,65 L8,75 L25,72 L54,55 Z" fill="rgba(0,212,176,0.1)" stroke="currentColor" stroke-width="1.2"/>
+      <path d="M66,40 L110,65 L112,75 L95,72 L66,55 Z" fill="rgba(0,212,176,0.1)" stroke="currentColor" stroke-width="1.2"/>
+      <rect x="28" y="52" width="5" height="12" rx="2" fill="rgba(0,0,0,.8)" stroke="currentColor" stroke-width="1"/>
+      <rect x="40" y="47" width="5" height="14" rx="2" fill="rgba(0,0,0,.8)" stroke="currentColor" stroke-width="1"/>
+      <rect x="87" y="52" width="5" height="12" rx="2" fill="rgba(0,0,0,.8)" stroke="currentColor" stroke-width="1"/>
+      <rect x="75" y="47" width="5" height="14" rx="2" fill="rgba(0,0,0,.8)" stroke="currentColor" stroke-width="1"/>
+      <path d="M29,64 L32,64 L30.5,72 Z" fill="#ffaa00" opacity="0.8"/>
+      <path d="M41,61 L44,61 L42.5,70 Z" fill="#ffaa00" opacity="0.8"/>
+      <path d="M88,64 L91,64 L89.5,72 Z" fill="#ffaa00" opacity="0.8"/>
+      <path d="M76,61 L79,61 L77.5,70 Z" fill="#ffaa00" opacity="0.8"/>
+      <path d="M60,10 C68,10 70,18 70,45 L66,95 L54,95 L50,45 C50,18 52,10 60,10 Z"
+        fill="rgba(0,212,176,0.15)" stroke="currentColor" stroke-width="1.5"/>
+      <polygon points="60,100 85,108 85,112 60,105 35,112 35,108"
+        fill="rgba(0,212,176,0.2)" stroke="currentColor" stroke-width="1.5"/>
+    </g>`
+};
+
+function getCategory(name) {
+  const bombers = ['B1','B2','B21','B52','H6','Tu160','Tu22M','Tu95','Vulcan','XB70'];
+  const cargo   = ['A400M','An124','An22','An225','An72','C1','C130','C17','C2','C390',
+                   'C5','Il76','KC135','Y20','P3','E2','E7','KJ600','AG600','Be200','CL415','US2'];
+  const helo    = ['AH64','CH47','CH53','Ka27','Ka52','Mi24','Mi26','Mi28','Mi8',
+                   'NH90','UH60','WZ10','WZ9','Z10','Z19','Z21','V22','V280'];
+  if (bombers.includes(name)) return 'bomber';
+  if (cargo.includes(name))   return 'cargo';
+  if (helo.includes(name))    return 'helo';
+  return 'fighter';
+}
+
+function aspectDescriptor(angle) {
+  if (angle < 30)  return '(HEAD-ON)';
+  if (angle > 150) return '(TAIL)';
+  if (angle > 75 && angle < 105) return '(BEAM)';
+  return '(OBLIQUE)';
+}
+
+// ──────────────────────────────────────────────
+// BACKEND API INTEGRATION (The Real Brain)
+// ──────────────────────────────────────────────
+function handleImageUpload(event) {
+  const file = event.target.files[0];
+  if (!file) return;
+  uploadedFile = file;
+
+  const img = document.getElementById('sensor-img');
+  img.src = URL.createObjectURL(file);
+  img.style.display = 'block';
+  document.getElementById('sensor-text').style.display = 'none';
+  document.getElementById('sensor-status').innerText  = 'TARGET ACQUIRED';
+  document.getElementById('sensor-status').style.color = '#00d4b0';
+}
+
+async function startSimulation() {
+  if (isUI_Locked) return;
+  if (!uploadedFile) {
+    alert("Please upload a target photo to initiate pipeline.");
+    return;
+  }
+
+  // 1. Enter Loading State & Lock UI
+  setLockState(true);
+  document.getElementById('sensor-status').innerText  = 'UPLINK ACTIVE...';
+  document.getElementById('sensor-status').style.color = '#f0a030';
+  document.getElementById('scan-line').style.display   = 'block';
+  document.getElementById('clear-btn').style.display = 'none';
+  
+  // 2. Gather UI values and generate plausible geometry for missing fields
+  const ownSpeedMS = parseFloat(document.getElementById('slide-spd').value) * 340;
+  const ownAltM    = parseInt(document.getElementById('slide-alt').value) * 0.3048;
+  const enemyAltM  = Math.floor(Math.random() * 9000 + 1000); 
+  const launchDist = Math.floor(Math.random() * 25000 + 10000);
+  const remainDist = Math.floor(Math.random() * launchDist);
+  const azimuth    = Math.floor(Math.random() * 180);
+  const elevation  = Math.floor(Math.random() * 20);
+  const friendly   = document.getElementById('own-platform-select').value || 'Tejas';
+
+  // 3. Construct Payload
+  const formData = new FormData();
+  formData.append('image', uploadedFile);
+  formData.append('your_speed', ownSpeedMS);
+  formData.append('your_altitude', ownAltM);
+  formData.append('enemy_altitude', enemyAltM);
+  formData.append('countermeasure_deployed', cmDeployed ? 1 : 0);
+  formData.append('launch_distance', launchDist);
+  formData.append('remaining_distance', remainDist);
+  formData.append('azimuth', azimuth);
+  formData.append('elevation', elevation);
+  formData.append('friendly_aircraft', friendly);
+
+  try {
+    // 4. Fire to FastAPI Backend
+    const response = await fetch('http://localhost:8000/analyze', {
+        method: 'POST',
+        body: formData
+    });
+
+    if (!response.ok) throw new Error("API Connection Failed");
+    const data = await response.json();
+
+    // 5. Success - Proceed to DOM update
+    document.getElementById('scan-line').style.display  = 'none';
+    document.getElementById('sensor-status').innerText  = 'LOCKED';
+    document.getElementById('sensor-status').style.color = '#e84444';
+    
+    updateDOM(data, remainDist, azimuth, ownSpeedMS, enemyAltM);
+
+  } catch (error) {
+    console.error(error);
+    document.getElementById('scan-line').style.display  = 'none';
+    document.getElementById('sensor-status').innerText  = 'UPLINK FAILED';
+    document.getElementById('sensor-status').style.color = '#e84444';
+  } finally {
+    setLockState(false);
+  }
+}
+
+// ──────────────────────────────────────────────
+// HUD UPDATER (Maps Real JSON to UI)
+// ──────────────────────────────────────────────
+function updateDOM(data, distMeters, azimuth, ownSpeedMS, enemyAltM) {
+  console.log(data);
+  const name = data.aircraft_name;
+  const cat = getCategory(name);
+  const isAA = data.no_aa_capability === 0;
+  const threatProb = isAA ? Math.floor(data.hit_probability * 100) : 0;
+
+  // Determine colour based on threat
+  let col = '#00d4b0'; // Safe
+  if (isAA) {
+    if (threatProb > 60 || data.eta_seconds < 10) col = '#e84444'; // Crit
+    else if (threatProb > 30) col = '#f0a030'; // Warn
+  }
+
+  // ── Header ──
+  const hdr = document.getElementById('header-alert');
+  hdr.innerText = 'TARGET LOCKED: ' + name.toUpperCase();
+  hdr.style.color = '#e84444';
+  hdr.classList.add('hb');
+
+  // ── SVG Canvas ──
+  const svgNode = document.getElementById('target-svg');
+  svgNode.style.color = col;
+  svgNode.innerHTML = premiumSVGs[cat];
+
+  document.getElementById('target-info').innerHTML = `
+    <div style="font-size:15px;font-weight:700;letter-spacing:.1em;color:${col};margin-top:5px;">${name.toUpperCase()}</div>
+    <div style="font-size:8px;color:rgba(255,255,255,.35);letter-spacing:.1em;margin-top:2px;">CLASS: ${cat.toUpperCase()}</div>
+  `;
+
+  // ── True Metadata ──
+  document.getElementById('true-specs-box').style.display = 'block';
+  document.getElementById('e-man').innerText  = data.maneuverability >= 2 ? "HIGH (2)" : (data.maneuverability === 1 ? "MED (1)" : "LOW (0)"); 
+  document.getElementById('e-alt').textContent = Math.round(enemyAltM / 0.3048).toLocaleString() + ' FT';
+  document.getElementById('e-gen').innerText  = 'GEN ' + data.enemy_generation.toFixed(1);
+  document.getElementById('e-rng').innerText  = data.missile_range  > 0 ? (data.missile_range  / 1000).toFixed(0) + ' KM' : 'NONE';
+  document.getElementById('e-mspd').innerText = data.missile_speed  > 0 ? data.missile_speed  + ' M/S' : 'N/A';
+  document.getElementById('e-aa').innerText   = isAA ? 'YES' : 'NO';
+
+  // ── Generation Advantage ──
+  if (isAA) {
+    const genDiff = currentOwn.gen - data.enemy_generation;
+    const advEl   = document.getElementById('gen-adv-val');
+    const advBox  = document.getElementById('gen-adv-box');
+
+    if (genDiff > 0) {
+      advEl.innerText = '+' + genDiff.toFixed(1) + ' FRIENDLY ADV';
+      advEl.style.color = '#2dd4a0';
+      advBox.style.borderColor = 'rgba(45,212,160,.28)';
+    } else if (genDiff < 0) {
+      advEl.innerText = genDiff.toFixed(1) + ' ENEMY ADV';
+      advEl.style.color = '#e84444';
+      advBox.style.borderColor = 'rgba(232,68,68,.28)';
+    } else {
+      advEl.innerText = 'PARITY';
+      advEl.style.color = '#f0a030';
+      advBox.style.borderColor = 'rgba(240,160,48,.28)';
+    }
+  } else {
+    document.getElementById('gen-adv-val').textContent = 'N/A';
+    document.getElementById('gen-adv-box').style.borderColor = 'rgba(0,212,176,.18)';
+  }
+
+  // ── Threat Gauge ──
+  const dash = (threatProb / 100) * 326;
+  document.getElementById('threat-arc').style.strokeDasharray = `${dash} 326`;
+  document.getElementById('threat-arc').style.stroke = col;
+  document.getElementById('threat-val').textContent = `${threatProb}%`;
+  document.getElementById('threat-val').setAttribute('fill', col);
+  
+  let labelText = threatProb > 60 ? 'CRITICAL' : (threatProb > 30 ? 'ELEVATED' : 'LOW');
+  document.getElementById('threat-label').textContent = labelText;
+  document.getElementById('threat-label').setAttribute('fill', col);
+
+  // ── Kinematics ──
+  document.getElementById('eta-val').innerText   = isAA ? data.eta_seconds.toFixed(1) + 's' : 'N/A';
+  document.getElementById('eta-val').style.color = col;
+  
+  const approxClose = data.missile_speed > 0 ? Math.floor(data.missile_speed + ownSpeedMS) : 0;
+  document.getElementById('close-val').innerText = isAA ? approxClose.toLocaleString() + ' M/S' : '0 M/S';
+  
+  document.getElementById('dist-val').innerText  = (distMeters / 1000).toFixed(1) + ' KM';
+  document.getElementById('aspect-val').innerText = azimuth + '\u00b0 ' + aspectDescriptor(azimuth);
+  document.getElementById('phase-val').innerText  = isAA ? 'TRACKING' : 'N/A';
+
+  // ── Banner ──
+  const banner = document.getElementById('tac-banner');
+  banner.style.borderColor = col;
+  banner.style.background  = col === '#00d4b0' ? 'rgba(0,212,176,.05)'
+                           : col === '#f0a030' ? 'rgba(240,160,48,.05)'
+                           : 'rgba(232,68,68,.05)';
+  document.getElementById('tac-text').innerText   = data.recommendation;
+  document.getElementById('tac-text').style.color = col;
+  
+  document.getElementById('conf-text').innerText  = threatProb > 70 ? 'HIGH CONF'
+                                                  : threatProb > 35 ? 'MED CONF'
+                                                  : 'SYS NORMAL';
+
+  // ── Critical Pulse ──
+  const hud = document.getElementById('main-hud');
+  threatProb > 60 ? hud.classList.add('threat-critical') : hud.classList.remove('threat-critical');
+
+  document.getElementById('clear-btn').style.display = 'block';
+}
+
+// ──────────────────────────────────────────────
+// RESET HUD TO NEUTRAL
+// ──────────────────────────────────────────────
+function clearTarget() {
+  uploadedFile = null;
+  // SVG
+  const svgNode = document.getElementById('target-svg');
+  svgNode.style.color = '#00d4b0';
+  svgNode.innerHTML = premiumSVGs.none;
+  document.getElementById('target-info').innerHTML =
+    `<div style="font-size:13px;font-weight:700;color:#00d4b0;margin-top:12px;letter-spacing:.1em;">SCANNING AIRSPACE...</div>`;
+  document.getElementById('true-specs-box').style.display = 'none';
+
+  // Threat gauge
+  document.getElementById('threat-arc').style.strokeDasharray = '0 326';
+  document.getElementById('threat-arc').style.stroke = '#00d4b0';
+  document.getElementById('threat-val').textContent = '0%';
+  document.getElementById('threat-val').setAttribute('fill', '#00d4b0');
+  document.getElementById('threat-label').textContent = 'SAFE';
+  document.getElementById('threat-label').setAttribute('fill', 'rgba(0,212,176,.5)');
+
+  // Kinematics
+  document.getElementById('eta-val').innerText   = 'N/A';
+  document.getElementById('eta-val').style.color = '#00d4b0';
+  document.getElementById('close-val').innerText = '0 M/S';
+  document.getElementById('dist-val').innerText  = '---';
+  document.getElementById('aspect-val').innerText = '---';
+  document.getElementById('phase-val').innerText  = '---';
+
+  // Banner
+  const banner = document.getElementById('tac-banner');
+  banner.style.borderColor = 'rgba(0,212,176,.2)';
+  banner.style.background  = 'rgba(0,212,176,.05)';
+  document.getElementById('tac-text').innerText   = 'CONTINUE SCANNING SECTOR';
+  document.getElementById('tac-text').style.color = '#00d4b0';
+  document.getElementById('conf-text').innerText  = 'SYS NORMAL';
+
+  // Header & Feed Reset
+  const hdr = document.getElementById('header-alert');
+  hdr.innerText    = 'SCANNING AIRSPACE';
+  hdr.style.color  = 'rgba(0,212,176,.3)';
+  hdr.classList.remove('hb');
+  
+  document.getElementById('main-hud').classList.remove('threat-critical');
+  document.getElementById('clear-btn').style.display   = 'none';
+  document.getElementById('sensor-status').innerText   = 'STANDBY';
+  document.getElementById('sensor-status').style.color = 'rgba(0,212,176,.3)';
+  document.getElementById('sensor-img').style.display  = 'none';
+  document.getElementById('sensor-text').style.display = 'block';
+  document.getElementById('file-upload').value = '';
+}
+
+// ── INIT ──
+updateOwnPlatform();
+</script>
+</body>
+</html>

release_models/aircraft_classifier/atas_final_fine_tuned_aircraft_classifier_model.keras

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f35c73c33c0318b07ec261f6d1cf37e5ce15543a32867a398841793c9f1fca2d
+size 941789413

release_models/eta/atas_final_eta_regressor_model.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:599498f9d68a49295abe2d221e8e5aaf20ba071b2c524b640e176b52bc5a19d7
+size 42999244

release_models/hit/atas_final_hit_classifier_model.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d4a4900f2e5e1a4027d6652ff605e363f5c79a97abb640aba09708d6792a78c1
+size 425666

requirements.txt

@@ -0,0 +1,13 @@
+fastapi==0.136.3
+uvicorn==0.49.0
+python-multipart==0.0.32
+pydantic==2.13.4
+tensorflow==2.16.1
+tf-keras==2.16.0
+tensorflow-hub==0.16.1
+numpy==1.26.4
+pandas
+pillow
+scikit-learn
+xgboost
+joblib

src/analysis_for_physics_generator.ipynb

@@ -0,0 +1,2770 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "id": "f8aa68cc",
+   "metadata": {},
+   "source": [
+    "# **Physics Generator - Design Philosophy**\n",
+    "\n",
+    "The physics generator creates synthetic missile engagement scenarios for ATAS.\n",
+    "\n",
+    "Real engagement data is difficult and unrealistic to obtain at scale, so synthetic data is used to generate controlled training scenarios with known ground-truth labels.\n",
+    "\n",
+    "The generator uses metadata-derived aircraft capability ranges from `aircraft_metadata.csv` instead of fully hardcoded values.\n"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "57887b61",
+   "metadata": {},
+   "source": [
+    "### **aircraft_metadata.csv**\n",
+    "\n",
+    "This CSV is the threat database for ATAS.\n",
+    "\n",
+    "The classifier only predicts:\n",
+    "```python\n",
+    "\"F22\"\n",
+    "````\n",
+    "\n",
+    "The metadata adds tactical information:\n",
+    "\n",
+    "* missile speed\n",
+    "* missile range\n",
+    "* aircraft generation\n",
+    "* maneuverability\n",
+    "* combat capability\n",
+    "\n",
+    "This allows the system to simulate engagement scenarios.\n",
+    "\n",
+    "---\n",
+    "\n",
+    "### Columns\n",
+    "\n",
+    "#### `aircraft`\n",
+    "\n",
+    "Aircraft class name.\n",
+    "\n",
+    "Used to match classifier output with metadata.\n",
+    "\n",
+    "---\n",
+    "\n",
+    "#### `missile_speed`\n",
+    "\n",
+    "Approx missile speed (m/s).\n",
+    "\n",
+    "Used for:\n",
+    "\n",
+    "* closure rate\n",
+    "* evasion time\n",
+    "* threat level\n",
+    "\n",
+    "Higher speed = less reaction time.\n",
+    "\n",
+    "---\n",
+    "\n",
+    "#### `missile_range`\n",
+    "\n",
+    "Approx max missile range (m).\n",
+    "\n",
+    "Used for:\n",
+    "\n",
+    "* launch distance generation\n",
+    "* engagement realism\n",
+    "\n",
+    "Higher range = longer reach.\n",
+    "\n",
+    "---\n",
+    "\n",
+    "#### `enemy_generation`\n",
+    "\n",
+    "Aircraft technology level.\n",
+    "\n",
+    "Values:\n",
+    "\n",
+    "* 3.5\n",
+    "* 4\n",
+    "* 4.5\n",
+    "* 5\n",
+    "\n",
+    "Used for:\n",
+    "\n",
+    "* threat weighting\n",
+    "* hit probability modifiers\n",
+    "\n",
+    "Higher generation = more dangerous.\n",
+    "\n",
+    "---\n",
+    "\n",
+    "#### `maneuverability`\n",
+    "\n",
+    "Aircraft agility.\n",
+    "\n",
+    "Values:\n",
+    "\n",
+    "* 0 = low\n",
+    "* 1 = medium\n",
+    "* 2 = high\n",
+    "\n",
+    "Used for:\n",
+    "\n",
+    "* evasion logic\n",
+    "* survival probability\n",
+    "\n",
+    "Higher maneuverability = harder to hit.\n",
+    "\n",
+    "---\n",
+    "\n",
+    "#### `no_aa_capability`\n",
+    "\n",
+    "Whether aircraft lacks air-to-air combat capability.\n",
+    "\n",
+    "Values:\n",
+    "\n",
+    "* 0 = combat capable\n",
+    "* 1 = not combat capable\n",
+    "\n",
+    "Used to:\n",
+    "\n",
+    "* lower threat score\n",
+    "* skip missile logic for support aircraft\n",
+    "\n",
+    "---\n",
+    "\n",
+    "## Why This Exists\n",
+    "\n",
+    "The metadata converts:\n",
+    "\n",
+    "```python\n",
+    "\"What aircraft is this?\"\n",
+    "```\n",
+    "\n",
+    "into:\n",
+    "\n",
+    "```python\n",
+    "\"How dangerous is this aircraft?\"\n",
+    "```"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "id": "ce0f85d7",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Importing the libraries\n",
+    "\n",
+    "import pandas as pd\n",
+    "import numpy as np"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 3,
+   "id": "6831bd71",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/html": [
+       "<div>\n",
+       "<style scoped>\n",
+       "    .dataframe tbody tr th:only-of-type {\n",
+       "        vertical-align: middle;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe tbody tr th {\n",
+       "        vertical-align: top;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe thead th {\n",
+       "        text-align: right;\n",
+       "    }\n",
+       "</style>\n",
+       "<table border=\"1\" class=\"dataframe\">\n",
+       "  <thead>\n",
+       "    <tr style=\"text-align: right;\">\n",
+       "      <th></th>\n",
+       "      <th>aircraft</th>\n",
+       "      <th>aircraft_max_speed</th>\n",
+       "      <th>missile_speed</th>\n",
+       "      <th>missile_range</th>\n",
+       "      <th>enemy_generation</th>\n",
+       "      <th>maneuverability</th>\n",
+       "      <th>no_aa_capability</th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <th>0</th>\n",
+       "      <td>A10</td>\n",
+       "      <td>222</td>\n",
+       "      <td>857</td>\n",
+       "      <td>35000</td>\n",
+       "      <td>4.0</td>\n",
+       "      <td>1</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>1</th>\n",
+       "      <td>A400M</td>\n",
+       "      <td>255</td>\n",
+       "      <td>-1</td>\n",
+       "      <td>-1</td>\n",
+       "      <td>4.0</td>\n",
+       "      <td>0</td>\n",
+       "      <td>1</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>2</th>\n",
+       "      <td>AG600</td>\n",
+       "      <td>155</td>\n",
+       "      <td>-1</td>\n",
+       "      <td>-1</td>\n",
+       "      <td>4.0</td>\n",
+       "      <td>0</td>\n",
+       "      <td>1</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>3</th>\n",
+       "      <td>AH64</td>\n",
+       "      <td>101</td>\n",
+       "      <td>750</td>\n",
+       "      <td>8000</td>\n",
+       "      <td>4.0</td>\n",
+       "      <td>1</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>4</th>\n",
+       "      <td>AKINCI</td>\n",
+       "      <td>100</td>\n",
+       "      <td>1372</td>\n",
+       "      <td>65000</td>\n",
+       "      <td>4.0</td>\n",
+       "      <td>1</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>...</th>\n",
+       "      <td>...</td>\n",
+       "      <td>...</td>\n",
+       "      <td>...</td>\n",
+       "      <td>...</td>\n",
+       "      <td>...</td>\n",
+       "      <td>...</td>\n",
+       "      <td>...</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>97</th>\n",
+       "      <td>Y20</td>\n",
+       "      <td>255</td>\n",
+       "      <td>-1</td>\n",
+       "      <td>-1</td>\n",
+       "      <td>4.0</td>\n",
+       "      <td>0</td>\n",
+       "      <td>1</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>98</th>\n",
+       "      <td>YF23</td>\n",
+       "      <td>648</td>\n",
+       "      <td>1372</td>\n",
+       "      <td>160000</td>\n",
+       "      <td>5.0</td>\n",
+       "      <td>2</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>99</th>\n",
+       "      <td>Z10</td>\n",
+       "      <td>83</td>\n",
+       "      <td>686</td>\n",
+       "      <td>8000</td>\n",
+       "      <td>4.0</td>\n",
+       "      <td>1</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>100</th>\n",
+       "      <td>Z19</td>\n",
+       "      <td>78</td>\n",
+       "      <td>686</td>\n",
+       "      <td>8000</td>\n",
+       "      <td>4.0</td>\n",
+       "      <td>1</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>101</th>\n",
+       "      <td>Z21</td>\n",
+       "      <td>83</td>\n",
+       "      <td>686</td>\n",
+       "      <td>8000</td>\n",
+       "      <td>4.0</td>\n",
+       "      <td>1</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n",
+       "<p>102 rows × 7 columns</p>\n",
+       "</div>"
+      ],
+      "text/plain": [
+       "    aircraft  aircraft_max_speed  missile_speed  missile_range  \\\n",
+       "0        A10                 222            857          35000   \n",
+       "1      A400M                 255             -1             -1   \n",
+       "2      AG600                 155             -1             -1   \n",
+       "3       AH64                 101            750           8000   \n",
+       "4     AKINCI                 100           1372          65000   \n",
+       "..       ...                 ...            ...            ...   \n",
+       "97       Y20                 255             -1             -1   \n",
+       "98      YF23                 648           1372         160000   \n",
+       "99       Z10                  83            686           8000   \n",
+       "100      Z19                  78            686           8000   \n",
+       "101      Z21                  83            686           8000   \n",
+       "\n",
+       "     enemy_generation  maneuverability  no_aa_capability  \n",
+       "0                 4.0                1                 0  \n",
+       "1                 4.0                0                 1  \n",
+       "2                 4.0                0                 1  \n",
+       "3                 4.0                1                 0  \n",
+       "4                 4.0                1                 0  \n",
+       "..                ...              ...               ...  \n",
+       "97                4.0                0                 1  \n",
+       "98                5.0                2                 0  \n",
+       "99                4.0                1                 0  \n",
+       "100               4.0                1                 0  \n",
+       "101               4.0                1                 0  \n",
+       "\n",
+       "[102 rows x 7 columns]"
+      ]
+     },
+     "execution_count": 3,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "# Importing the CSV to work with\n",
+    "\n",
+    "df = pd.read_csv(\"../data/aircraft_metadata.csv\")\n",
+    "df"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 4,
+   "id": "3ac9444c",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "<class 'pandas.DataFrame'>\n",
+      "RangeIndex: 102 entries, 0 to 101\n",
+      "Data columns (total 7 columns):\n",
+      " #   Column              Non-Null Count  Dtype  \n",
+      "---  ------              --------------  -----  \n",
+      " 0   aircraft            102 non-null    str    \n",
+      " 1   aircraft_max_speed  102 non-null    int64  \n",
+      " 2   missile_speed       102 non-null    int64  \n",
+      " 3   missile_range       102 non-null    int64  \n",
+      " 4   enemy_generation    102 non-null    float64\n",
+      " 5   maneuverability     102 non-null    int64  \n",
+      " 6   no_aa_capability    102 non-null    int64  \n",
+      "dtypes: float64(1), int64(5), str(1)\n",
+      "memory usage: 5.7 KB\n"
+     ]
+    }
+   ],
+   "source": [
+    "df.info()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 5,
+   "id": "c29a7656",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/html": [
+       "<div>\n",
+       "<style scoped>\n",
+       "    .dataframe tbody tr th:only-of-type {\n",
+       "        vertical-align: middle;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe tbody tr th {\n",
+       "        vertical-align: top;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe thead th {\n",
+       "        text-align: right;\n",
+       "    }\n",
+       "</style>\n",
+       "<table border=\"1\" class=\"dataframe\">\n",
+       "  <thead>\n",
+       "    <tr style=\"text-align: right;\">\n",
+       "      <th></th>\n",
+       "      <th>aircraft</th>\n",
+       "      <th>aircraft_max_speed</th>\n",
+       "      <th>missile_speed</th>\n",
+       "      <th>missile_range</th>\n",
+       "      <th>enemy_generation</th>\n",
+       "      <th>maneuverability</th>\n",
+       "      <th>no_aa_capability</th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <th>count</th>\n",
+       "      <td>102</td>\n",
+       "      <td>102.000000</td>\n",
+       "      <td>102.000000</td>\n",
+       "      <td>102.000000</td>\n",
+       "      <td>102.000000</td>\n",
+       "      <td>102.000000</td>\n",
+       "      <td>102.000000</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>unique</th>\n",
+       "      <td>102</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>top</th>\n",
+       "      <td>A10</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>freq</th>\n",
+       "      <td>1</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>mean</th>\n",
+       "      <td>NaN</td>\n",
+       "      <td>345.764706</td>\n",
+       "      <td>636.460784</td>\n",
+       "      <td>64724.990196</td>\n",
+       "      <td>4.117647</td>\n",
+       "      <td>0.852941</td>\n",
+       "      <td>0.500000</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>std</th>\n",
+       "      <td>NaN</td>\n",
+       "      <td>226.619242</td>\n",
+       "      <td>692.537831</td>\n",
+       "      <td>102267.602554</td>\n",
+       "      <td>0.478118</td>\n",
+       "      <td>0.825306</td>\n",
+       "      <td>0.502469</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>min</th>\n",
+       "      <td>NaN</td>\n",
+       "      <td>61.000000</td>\n",
+       "      <td>-1.000000</td>\n",
+       "      <td>-1.000000</td>\n",
+       "      <td>3.500000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>25%</th>\n",
+       "      <td>NaN</td>\n",
+       "      <td>164.500000</td>\n",
+       "      <td>-1.000000</td>\n",
+       "      <td>-1.000000</td>\n",
+       "      <td>4.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>50%</th>\n",
+       "      <td>NaN</td>\n",
+       "      <td>268.000000</td>\n",
+       "      <td>342.500000</td>\n",
+       "      <td>3999.500000</td>\n",
+       "      <td>4.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>0.500000</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>75%</th>\n",
+       "      <td>NaN</td>\n",
+       "      <td>544.000000</td>\n",
+       "      <td>1372.000000</td>\n",
+       "      <td>108750.000000</td>\n",
+       "      <td>4.500000</td>\n",
+       "      <td>2.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>max</th>\n",
+       "      <td>NaN</td>\n",
+       "      <td>983.000000</td>\n",
+       "      <td>2058.000000</td>\n",
+       "      <td>400000.000000</td>\n",
+       "      <td>5.000000</td>\n",
+       "      <td>2.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n",
+       "</div>"
+      ],
+      "text/plain": [
+       "       aircraft  aircraft_max_speed  missile_speed  missile_range  \\\n",
+       "count       102          102.000000     102.000000     102.000000   \n",
+       "unique      102                 NaN            NaN            NaN   \n",
+       "top         A10                 NaN            NaN            NaN   \n",
+       "freq          1                 NaN            NaN            NaN   \n",
+       "mean        NaN          345.764706     636.460784   64724.990196   \n",
+       "std         NaN          226.619242     692.537831  102267.602554   \n",
+       "min         NaN           61.000000      -1.000000      -1.000000   \n",
+       "25%         NaN          164.500000      -1.000000      -1.000000   \n",
+       "50%         NaN          268.000000     342.500000    3999.500000   \n",
+       "75%         NaN          544.000000    1372.000000  108750.000000   \n",
+       "max         NaN          983.000000    2058.000000  400000.000000   \n",
+       "\n",
+       "        enemy_generation  maneuverability  no_aa_capability  \n",
+       "count         102.000000       102.000000        102.000000  \n",
+       "unique               NaN              NaN               NaN  \n",
+       "top                  NaN              NaN               NaN  \n",
+       "freq                 NaN              NaN               NaN  \n",
+       "mean            4.117647         0.852941          0.500000  \n",
+       "std             0.478118         0.825306          0.502469  \n",
+       "min             3.500000         0.000000          0.000000  \n",
+       "25%             4.000000         0.000000          0.000000  \n",
+       "50%             4.000000         1.000000          0.500000  \n",
+       "75%             4.500000         2.000000          1.000000  \n",
+       "max             5.000000         2.000000          1.000000  "
+      ]
+     },
+     "execution_count": 5,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "df.describe(include=\"all\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 6,
+   "id": "ce49133d",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "[4.  3.5 4.5 5. ]\n",
+      "[1 0 2]\n",
+      "[0 1]\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(df[\"enemy_generation\"].unique())\n",
+    "print(df[\"maneuverability\"].unique())\n",
+    "print(df[\"no_aa_capability\"].unique())"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 7,
+   "id": "0297a837",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/html": [
+       "<div>\n",
+       "<style scoped>\n",
+       "    .dataframe tbody tr th:only-of-type {\n",
+       "        vertical-align: middle;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe tbody tr th {\n",
+       "        vertical-align: top;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe thead th {\n",
+       "        text-align: right;\n",
+       "    }\n",
+       "</style>\n",
+       "<table border=\"1\" class=\"dataframe\">\n",
+       "  <thead>\n",
+       "    <tr style=\"text-align: right;\">\n",
+       "      <th></th>\n",
+       "      <th>aircraft</th>\n",
+       "      <th>aircraft_max_speed</th>\n",
+       "      <th>missile_speed</th>\n",
+       "      <th>missile_range</th>\n",
+       "      <th>enemy_generation</th>\n",
+       "      <th>maneuverability</th>\n",
+       "      <th>no_aa_capability</th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <th>0</th>\n",
+       "      <td>A10</td>\n",
+       "      <td>222</td>\n",
+       "      <td>857</td>\n",
+       "      <td>35000</td>\n",
+       "      <td>4.0</td>\n",
+       "      <td>1</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>1</th>\n",
+       "      <td>AH64</td>\n",
+       "      <td>101</td>\n",
+       "      <td>750</td>\n",
+       "      <td>8000</td>\n",
+       "      <td>4.0</td>\n",
+       "      <td>1</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>2</th>\n",
+       "      <td>AKINCI</td>\n",
+       "      <td>100</td>\n",
+       "      <td>1372</td>\n",
+       "      <td>65000</td>\n",
+       "      <td>4.0</td>\n",
+       "      <td>1</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>3</th>\n",
+       "      <td>AV8B</td>\n",
+       "      <td>300</td>\n",
+       "      <td>1372</td>\n",
+       "      <td>160000</td>\n",
+       "      <td>4.0</td>\n",
+       "      <td>1</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>4</th>\n",
+       "      <td>EF2000</td>\n",
+       "      <td>590</td>\n",
+       "      <td>1372</td>\n",
+       "      <td>200000</td>\n",
+       "      <td>4.5</td>\n",
+       "      <td>2</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n",
+       "</div>"
+      ],
+      "text/plain": [
+       "  aircraft  aircraft_max_speed  missile_speed  missile_range  \\\n",
+       "0      A10                 222            857          35000   \n",
+       "1     AH64                 101            750           8000   \n",
+       "2   AKINCI                 100           1372          65000   \n",
+       "3     AV8B                 300           1372         160000   \n",
+       "4   EF2000                 590           1372         200000   \n",
+       "\n",
+       "   enemy_generation  maneuverability  no_aa_capability  \n",
+       "0               4.0                1                 0  \n",
+       "1               4.0                1                 0  \n",
+       "2               4.0                1                 0  \n",
+       "3               4.0                1                 0  \n",
+       "4               4.5                2                 0  "
+      ]
+     },
+     "execution_count": 7,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "# Getting data of aircarfts that has air-to-air combact ability\n",
+    "combat_df = df[df[\"no_aa_capability\"]==0].reset_index(drop=True)\n",
+    "combat_df.head()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 8,
+   "id": "c578c9c1",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "<class 'pandas.DataFrame'>\n",
+      "RangeIndex: 51 entries, 0 to 50\n",
+      "Data columns (total 7 columns):\n",
+      " #   Column              Non-Null Count  Dtype  \n",
+      "---  ------              --------------  -----  \n",
+      " 0   aircraft            51 non-null     str    \n",
+      " 1   aircraft_max_speed  51 non-null     int64  \n",
+      " 2   missile_speed       51 non-null     int64  \n",
+      " 3   missile_range       51 non-null     int64  \n",
+      " 4   enemy_generation    51 non-null     float64\n",
+      " 5   maneuverability     51 non-null     int64  \n",
+      " 6   no_aa_capability    51 non-null     int64  \n",
+      "dtypes: float64(1), int64(5), str(1)\n",
+      "memory usage: 2.9 KB\n"
+     ]
+    }
+   ],
+   "source": [
+    "combat_df.info()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 9,
+   "id": "2dc32a13",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/html": [
+       "<div>\n",
+       "<style scoped>\n",
+       "    .dataframe tbody tr th:only-of-type {\n",
+       "        vertical-align: middle;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe tbody tr th {\n",
+       "        vertical-align: top;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe thead th {\n",
+       "        text-align: right;\n",
+       "    }\n",
+       "</style>\n",
+       "<table border=\"1\" class=\"dataframe\">\n",
+       "  <thead>\n",
+       "    <tr style=\"text-align: right;\">\n",
+       "      <th></th>\n",
+       "      <th>aircraft</th>\n",
+       "      <th>aircraft_max_speed</th>\n",
+       "      <th>missile_speed</th>\n",
+       "      <th>missile_range</th>\n",
+       "      <th>enemy_generation</th>\n",
+       "      <th>maneuverability</th>\n",
+       "      <th>no_aa_capability</th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <th>count</th>\n",
+       "      <td>51</td>\n",
+       "      <td>51.000000</td>\n",
+       "      <td>51.000000</td>\n",
+       "      <td>51.000000</td>\n",
+       "      <td>51.000000</td>\n",
+       "      <td>51.000000</td>\n",
+       "      <td>51.0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>unique</th>\n",
+       "      <td>51</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>top</th>\n",
+       "      <td>A10</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>freq</th>\n",
+       "      <td>1</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "      <td>NaN</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>mean</th>\n",
+       "      <td>NaN</td>\n",
+       "      <td>440.235294</td>\n",
+       "      <td>1273.921569</td>\n",
+       "      <td>129450.980392</td>\n",
+       "      <td>4.284314</td>\n",
+       "      <td>1.529412</td>\n",
+       "      <td>0.0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>std</th>\n",
+       "      <td>NaN</td>\n",
+       "      <td>225.509609</td>\n",
+       "      <td>373.955497</td>\n",
+       "      <td>112160.655085</td>\n",
+       "      <td>0.461030</td>\n",
+       "      <td>0.504101</td>\n",
+       "      <td>0.0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>min</th>\n",
+       "      <td>NaN</td>\n",
+       "      <td>78.000000</td>\n",
+       "      <td>686.000000</td>\n",
+       "      <td>8000.000000</td>\n",
+       "      <td>3.500000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>0.0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>25%</th>\n",
+       "      <td>NaN</td>\n",
+       "      <td>246.500000</td>\n",
+       "      <td>857.000000</td>\n",
+       "      <td>37500.000000</td>\n",
+       "      <td>4.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>0.0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>50%</th>\n",
+       "      <td>NaN</td>\n",
+       "      <td>532.000000</td>\n",
+       "      <td>1372.000000</td>\n",
+       "      <td>110000.000000</td>\n",
+       "      <td>4.000000</td>\n",
+       "      <td>2.000000</td>\n",
+       "      <td>0.0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>75%</th>\n",
+       "      <td>NaN</td>\n",
+       "      <td>590.000000</td>\n",
+       "      <td>1372.000000</td>\n",
+       "      <td>160000.000000</td>\n",
+       "      <td>4.500000</td>\n",
+       "      <td>2.000000</td>\n",
+       "      <td>0.0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>max</th>\n",
+       "      <td>NaN</td>\n",
+       "      <td>833.000000</td>\n",
+       "      <td>2058.000000</td>\n",
+       "      <td>400000.000000</td>\n",
+       "      <td>5.000000</td>\n",
+       "      <td>2.000000</td>\n",
+       "      <td>0.0</td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n",
+       "</div>"
+      ],
+      "text/plain": [
+       "       aircraft  aircraft_max_speed  missile_speed  missile_range  \\\n",
+       "count        51           51.000000      51.000000      51.000000   \n",
+       "unique       51                 NaN            NaN            NaN   \n",
+       "top         A10                 NaN            NaN            NaN   \n",
+       "freq          1                 NaN            NaN            NaN   \n",
+       "mean        NaN          440.235294    1273.921569  129450.980392   \n",
+       "std         NaN          225.509609     373.955497  112160.655085   \n",
+       "min         NaN           78.000000     686.000000    8000.000000   \n",
+       "25%         NaN          246.500000     857.000000   37500.000000   \n",
+       "50%         NaN          532.000000    1372.000000  110000.000000   \n",
+       "75%         NaN          590.000000    1372.000000  160000.000000   \n",
+       "max         NaN          833.000000    2058.000000  400000.000000   \n",
+       "\n",
+       "        enemy_generation  maneuverability  no_aa_capability  \n",
+       "count          51.000000        51.000000              51.0  \n",
+       "unique               NaN              NaN               NaN  \n",
+       "top                  NaN              NaN               NaN  \n",
+       "freq                 NaN              NaN               NaN  \n",
+       "mean            4.284314         1.529412               0.0  \n",
+       "std             0.461030         0.504101               0.0  \n",
+       "min             3.500000         1.000000               0.0  \n",
+       "25%             4.000000         1.000000               0.0  \n",
+       "50%             4.000000         2.000000               0.0  \n",
+       "75%             4.500000         2.000000               0.0  \n",
+       "max             5.000000         2.000000               0.0  "
+      ]
+     },
+     "execution_count": 9,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "combat_df.describe(include=\"all\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 10,
+   "id": "b51a060f",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Missile Speed ranges: \n",
+      "\n",
+      "Max → 2058 \n",
+      "Min → 686\n"
+     ]
+    }
+   ],
+   "source": [
+    "# Getting missile speed ranges\n",
+    "print(\"Missile Speed ranges: \\n\")\n",
+    "print(f\"Max → {combat_df['missile_speed'].max()} \\nMin → {combat_df['missile_speed'].min()}\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 11,
+   "id": "d76aabf2",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Missile range: \n",
+      "\n",
+      "Max → 400000 \n",
+      "Min → 8000\n"
+     ]
+    }
+   ],
+   "source": [
+    "# Getting missile speed ranges\n",
+    "print(\"Missile range: \\n\")\n",
+    "print(f\"Max → {combat_df['missile_range'].max()} \\nMin → {combat_df['missile_range'].min()}\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 12,
+   "id": "1e7207e0",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Aircraft speed range: \n",
+      "\n",
+      "Max → 983 \n",
+      "Min → 61\n"
+     ]
+    }
+   ],
+   "source": [
+    "# Getting aircraft speed ranges\n",
+    "print(\"Aircraft speed range: \\n\")\n",
+    "print(f\"Max → {df['aircraft_max_speed'].max()} \\nMin → {df['aircraft_max_speed'].min()}\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 13,
+   "id": "21a1d9d3",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "enemy_generation\n",
+       "4.0    25\n",
+       "4.5    11\n",
+       "5.0    11\n",
+       "3.5     4\n",
+       "Name: count, dtype: int64"
+      ]
+     },
+     "execution_count": 13,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "# Getting the number of unique generation\n",
+    "combat_df[\"enemy_generation\"].value_counts()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 14,
+   "id": "d47323a4",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "maneuverability\n",
+       "2    27\n",
+       "1    24\n",
+       "Name: count, dtype: int64"
+      ]
+     },
+     "execution_count": 14,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "# Getting the maneverability level\n",
+    "combat_df[\"maneuverability\"].value_counts()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 15,
+   "id": "7471829e",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/html": [
+       "<div>\n",
+       "<style scoped>\n",
+       "    .dataframe tbody tr th:only-of-type {\n",
+       "        vertical-align: middle;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe tbody tr th {\n",
+       "        vertical-align: top;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe thead th {\n",
+       "        text-align: right;\n",
+       "    }\n",
+       "</style>\n",
+       "<table border=\"1\" class=\"dataframe\">\n",
+       "  <thead>\n",
+       "    <tr style=\"text-align: right;\">\n",
+       "      <th></th>\n",
+       "      <th>aircraft</th>\n",
+       "      <th>aircraft_max_speed</th>\n",
+       "      <th>missile_speed</th>\n",
+       "      <th>missile_range</th>\n",
+       "      <th>enemy_generation</th>\n",
+       "      <th>maneuverability</th>\n",
+       "      <th>no_aa_capability</th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <th>14</th>\n",
+       "      <td>FCK1</td>\n",
+       "      <td>532</td>\n",
+       "      <td>2058</td>\n",
+       "      <td>100000</td>\n",
+       "      <td>4.0</td>\n",
+       "      <td>2</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>39</th>\n",
+       "      <td>Su57</td>\n",
+       "      <td>590</td>\n",
+       "      <td>2058</td>\n",
+       "      <td>400000</td>\n",
+       "      <td>5.0</td>\n",
+       "      <td>2</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>32</th>\n",
+       "      <td>Mig31</td>\n",
+       "      <td>833</td>\n",
+       "      <td>2058</td>\n",
+       "      <td>400000</td>\n",
+       "      <td>4.0</td>\n",
+       "      <td>1</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>18</th>\n",
+       "      <td>J36</td>\n",
+       "      <td>590</td>\n",
+       "      <td>1715</td>\n",
+       "      <td>400000</td>\n",
+       "      <td>5.0</td>\n",
+       "      <td>2</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>19</th>\n",
+       "      <td>J50</td>\n",
+       "      <td>590</td>\n",
+       "      <td>1715</td>\n",
+       "      <td>400000</td>\n",
+       "      <td>5.0</td>\n",
+       "      <td>2</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>17</th>\n",
+       "      <td>J35</td>\n",
+       "      <td>590</td>\n",
+       "      <td>1715</td>\n",
+       "      <td>300000</td>\n",
+       "      <td>5.0</td>\n",
+       "      <td>2</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>16</th>\n",
+       "      <td>J20</td>\n",
+       "      <td>590</td>\n",
+       "      <td>1715</td>\n",
+       "      <td>300000</td>\n",
+       "      <td>5.0</td>\n",
+       "      <td>2</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>15</th>\n",
+       "      <td>J10</td>\n",
+       "      <td>650</td>\n",
+       "      <td>1715</td>\n",
+       "      <td>300000</td>\n",
+       "      <td>4.5</td>\n",
+       "      <td>2</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>10</th>\n",
+       "      <td>F2</td>\n",
+       "      <td>590</td>\n",
+       "      <td>1544</td>\n",
+       "      <td>105000</td>\n",
+       "      <td>4.5</td>\n",
+       "      <td>2</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>41</th>\n",
+       "      <td>Tejas</td>\n",
+       "      <td>550</td>\n",
+       "      <td>1544</td>\n",
+       "      <td>110000</td>\n",
+       "      <td>4.5</td>\n",
+       "      <td>2</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n",
+       "</div>"
+      ],
+      "text/plain": [
+       "   aircraft  aircraft_max_speed  missile_speed  missile_range  \\\n",
+       "14     FCK1                 532           2058         100000   \n",
+       "39     Su57                 590           2058         400000   \n",
+       "32    Mig31                 833           2058         400000   \n",
+       "18      J36                 590           1715         400000   \n",
+       "19      J50                 590           1715         400000   \n",
+       "17      J35                 590           1715         300000   \n",
+       "16      J20                 590           1715         300000   \n",
+       "15      J10                 650           1715         300000   \n",
+       "10       F2                 590           1544         105000   \n",
+       "41    Tejas                 550           1544         110000   \n",
+       "\n",
+       "    enemy_generation  maneuverability  no_aa_capability  \n",
+       "14               4.0                2                 0  \n",
+       "39               5.0                2                 0  \n",
+       "32               4.0                1                 0  \n",
+       "18               5.0                2                 0  \n",
+       "19               5.0                2                 0  \n",
+       "17               5.0                2                 0  \n",
+       "16               5.0                2                 0  \n",
+       "15               4.5                2                 0  \n",
+       "10               4.5                2                 0  \n",
+       "41               4.5                2                 0  "
+      ]
+     },
+     "execution_count": 15,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "# get values of ranges of missile speed\n",
+    "combat_df.sort_values(\"missile_speed\", ascending=False).head(10)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 16,
+   "id": "f14cb2cf",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/html": [
+       "<div>\n",
+       "<style scoped>\n",
+       "    .dataframe tbody tr th:only-of-type {\n",
+       "        vertical-align: middle;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe tbody tr th {\n",
+       "        vertical-align: top;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe thead th {\n",
+       "        text-align: right;\n",
+       "    }\n",
+       "</style>\n",
+       "<table border=\"1\" class=\"dataframe\">\n",
+       "  <thead>\n",
+       "    <tr style=\"text-align: right;\">\n",
+       "      <th></th>\n",
+       "      <th>aircraft</th>\n",
+       "      <th>aircraft_max_speed</th>\n",
+       "      <th>missile_speed</th>\n",
+       "      <th>missile_range</th>\n",
+       "      <th>enemy_generation</th>\n",
+       "      <th>maneuverability</th>\n",
+       "      <th>no_aa_capability</th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <th>19</th>\n",
+       "      <td>J50</td>\n",
+       "      <td>590</td>\n",
+       "      <td>1715</td>\n",
+       "      <td>400000</td>\n",
+       "      <td>5.0</td>\n",
+       "      <td>2</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>39</th>\n",
+       "      <td>Su57</td>\n",
+       "      <td>590</td>\n",
+       "      <td>2058</td>\n",
+       "      <td>400000</td>\n",
+       "      <td>5.0</td>\n",
+       "      <td>2</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>18</th>\n",
+       "      <td>J36</td>\n",
+       "      <td>590</td>\n",
+       "      <td>1715</td>\n",
+       "      <td>400000</td>\n",
+       "      <td>5.0</td>\n",
+       "      <td>2</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>32</th>\n",
+       "      <td>Mig31</td>\n",
+       "      <td>833</td>\n",
+       "      <td>2058</td>\n",
+       "      <td>400000</td>\n",
+       "      <td>4.0</td>\n",
+       "      <td>1</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>17</th>\n",
+       "      <td>J35</td>\n",
+       "      <td>590</td>\n",
+       "      <td>1715</td>\n",
+       "      <td>300000</td>\n",
+       "      <td>5.0</td>\n",
+       "      <td>2</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>16</th>\n",
+       "      <td>J20</td>\n",
+       "      <td>590</td>\n",
+       "      <td>1715</td>\n",
+       "      <td>300000</td>\n",
+       "      <td>5.0</td>\n",
+       "      <td>2</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>15</th>\n",
+       "      <td>J10</td>\n",
+       "      <td>650</td>\n",
+       "      <td>1715</td>\n",
+       "      <td>300000</td>\n",
+       "      <td>4.5</td>\n",
+       "      <td>2</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>34</th>\n",
+       "      <td>Rafale</td>\n",
+       "      <td>531</td>\n",
+       "      <td>1372</td>\n",
+       "      <td>200000</td>\n",
+       "      <td>4.5</td>\n",
+       "      <td>2</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>20</th>\n",
+       "      <td>JAS39</td>\n",
+       "      <td>590</td>\n",
+       "      <td>1372</td>\n",
+       "      <td>200000</td>\n",
+       "      <td>4.5</td>\n",
+       "      <td>2</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>24</th>\n",
+       "      <td>KF21</td>\n",
+       "      <td>532</td>\n",
+       "      <td>1372</td>\n",
+       "      <td>200000</td>\n",
+       "      <td>4.5</td>\n",
+       "      <td>2</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n",
+       "</div>"
+      ],
+      "text/plain": [
+       "   aircraft  aircraft_max_speed  missile_speed  missile_range  \\\n",
+       "19      J50                 590           1715         400000   \n",
+       "39     Su57                 590           2058         400000   \n",
+       "18      J36                 590           1715         400000   \n",
+       "32    Mig31                 833           2058         400000   \n",
+       "17      J35                 590           1715         300000   \n",
+       "16      J20                 590           1715         300000   \n",
+       "15      J10                 650           1715         300000   \n",
+       "34   Rafale                 531           1372         200000   \n",
+       "20    JAS39                 590           1372         200000   \n",
+       "24     KF21                 532           1372         200000   \n",
+       "\n",
+       "    enemy_generation  maneuverability  no_aa_capability  \n",
+       "19               5.0                2                 0  \n",
+       "39               5.0                2                 0  \n",
+       "18               5.0                2                 0  \n",
+       "32               4.0                1                 0  \n",
+       "17               5.0                2                 0  \n",
+       "16               5.0                2                 0  \n",
+       "15               4.5                2                 0  \n",
+       "34               4.5                2                 0  \n",
+       "20               4.5                2                 0  \n",
+       "24               4.5                2                 0  "
+      ]
+     },
+     "execution_count": 16,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "# Get sorted decending missile ranges\n",
+    "combat_df.sort_values(\"missile_range\", ascending=False)[:10]"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 17,
+   "id": "5e723e87",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/html": [
+       "<div>\n",
+       "<style scoped>\n",
+       "    .dataframe tbody tr th:only-of-type {\n",
+       "        vertical-align: middle;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe tbody tr th {\n",
+       "        vertical-align: top;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe thead th {\n",
+       "        text-align: right;\n",
+       "    }\n",
+       "</style>\n",
+       "<table border=\"1\" class=\"dataframe\">\n",
+       "  <thead>\n",
+       "    <tr style=\"text-align: right;\">\n",
+       "      <th></th>\n",
+       "      <th>aircraft</th>\n",
+       "      <th>aircraft_max_speed</th>\n",
+       "      <th>missile_speed</th>\n",
+       "      <th>missile_range</th>\n",
+       "      <th>enemy_generation</th>\n",
+       "      <th>maneuverability</th>\n",
+       "      <th>no_aa_capability</th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <th>11</th>\n",
+       "      <td>F22</td>\n",
+       "      <td>669</td>\n",
+       "      <td>1372</td>\n",
+       "      <td>160000</td>\n",
+       "      <td>5.0</td>\n",
+       "      <td>2</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n",
+       "</div>"
+      ],
+      "text/plain": [
+       "   aircraft  aircraft_max_speed  missile_speed  missile_range  \\\n",
+       "11      F22                 669           1372         160000   \n",
+       "\n",
+       "    enemy_generation  maneuverability  no_aa_capability  \n",
+       "11               5.0                2                 0  "
+      ]
+     },
+     "execution_count": 17,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "combat_df[combat_df[\"aircraft\"]==\"F22\"]"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 18,
+   "id": "41f0712c",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/html": [
+       "<div>\n",
+       "<style scoped>\n",
+       "    .dataframe tbody tr th:only-of-type {\n",
+       "        vertical-align: middle;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe tbody tr th {\n",
+       "        vertical-align: top;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe thead th {\n",
+       "        text-align: right;\n",
+       "    }\n",
+       "</style>\n",
+       "<table border=\"1\" class=\"dataframe\">\n",
+       "  <thead>\n",
+       "    <tr style=\"text-align: right;\">\n",
+       "      <th></th>\n",
+       "      <th>aircraft_max_speed</th>\n",
+       "      <th>missile_speed</th>\n",
+       "      <th>missile_range</th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <th>aircraft_max_speed</th>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>0.753865</td>\n",
+       "      <td>0.633663</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>missile_speed</th>\n",
+       "      <td>0.753865</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>0.820434</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>missile_range</th>\n",
+       "      <td>0.633663</td>\n",
+       "      <td>0.820434</td>\n",
+       "      <td>1.000000</td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n",
+       "</div>"
+      ],
+      "text/plain": [
+       "                    aircraft_max_speed  missile_speed  missile_range\n",
+       "aircraft_max_speed            1.000000       0.753865       0.633663\n",
+       "missile_speed                 0.753865       1.000000       0.820434\n",
+       "missile_range                 0.633663       0.820434       1.000000"
+      ]
+     },
+     "execution_count": 18,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "# correlation between aircraft speed and missile \n",
+    "combat_df[[\"aircraft_max_speed\", \"missile_speed\", \"missile_range\"]].corr()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 19,
+   "id": "468e508e",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "enemy_generation\n",
+       "3.5     26500.000000\n",
+       "4.0     83240.000000\n",
+       "4.5    159090.909091\n",
+       "5.0    242272.727273\n",
+       "Name: missile_range, dtype: float64"
+      ]
+     },
+     "execution_count": 19,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "combat_df.groupby(\"enemy_generation\")[\"missile_range\"].mean()"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "f7850284",
+   "metadata": {},
+   "source": [
+    "## **Extracted Metadata Insights**\n",
+    "\n",
+    "### **Combat-Capable Aircraft**\n",
+    "- Total aircraft: 102\n",
+    "- Combat-capable aircraft: 56\n",
+    "- Non combat-capable aircraft: 46\n",
+    "\n",
+    "---\n",
+    "\n",
+    "### Aircraft Speed Range\n",
+    "\n",
+    "```python\n",
+    "61 m/s → 983 m/s\n",
+    "```\n",
+    "\n",
+    "---\n",
+    "\n",
+    "### Missile Speed Range\n",
+    "\n",
+    "```python \n",
+    "686 m/s → 2058 m/s\n",
+    "````\n",
+    "\n",
+    "---\n",
+    "\n",
+    "### Missile Range\n",
+    "\n",
+    "```python id=\"e2h15d\"\n",
+    "8,000 m → 400,000 m\n",
+    "```\n",
+    "\n",
+    "---\n",
+    "\n",
+    "### Enemy Generation Distribution\n",
+    "\n",
+    "```python id=\"txut2q\"\n",
+    "4.0  → 23 aircraft\n",
+    "5.0  → 13 aircraft\n",
+    "4.5  → 10 aircraft\n",
+    "3.5  → 5 aircraft\n",
+    "```\n",
+    "\n",
+    "---\n",
+    "\n",
+    "### Maneuverability Distribution\n",
+    "\n",
+    "```python id=\"nt11ao\"\n",
+    "2 (high)   → 27 aircraft\n",
+    "1 (medium) → 24 aircraft\n",
+    "```\n",
+    "\n",
+    "---\n",
+    "\n",
+    "### Key Observations\n",
+    "\n",
+    "* Most combat aircraft belong to Gen 4 and Gen 5.\n",
+    "* Most combat aircraft have medium or high maneuverability.\n",
+    "* Advanced aircraft generally have longer missile ranges.\n",
+    "* High-end aircraft like Su57, Mig31, J35, J36, and J50 dominate the upper missile range limits.\n",
+    "\n",
+    "The goal of the generator is not perfect aerospace simulation, but generation of believable and internally consistent combat scenarios that allow ML models to learn meaningful relationships.\n",
+    "\n",
+    "---"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "8832d356",
+   "metadata": {},
+   "source": [
+    "## **Plan to build physics generator**\n",
+    "\n",
+    "### **How it should be called**\n",
+    "\n",
+    "```python\n",
+    "from src.physics_generator import generate_dataset\n",
+    "generate_dataset()\n",
+    "```\n",
+    "\n",
+    "### **How it will be build**\n",
+    "\n",
+    "```markdown\n",
+    "generate_dataset()          ← the one function the notebook calls\n",
+    "    └── load_metadata()     ← reads aircraft_metadata.csv\n",
+    "    └── generate_row()      ← builds one scenario\n",
+    "            └── derive_missile_phase()\n",
+    "            └── derive_closure_rate()\n",
+    "            └── derive_evasion_time()\n",
+    "            └── derive_hit_label()\n",
+    "    └── save_dataset()      ← saves to CSV\n",
+    "```\n",
+    "---\n",
+    "\n",
+    "### **The sequence of `physics_generator.py` script**\n",
+    "\n",
+    "```python\n",
+    "_generate_row()\n",
+    "│\n",
+    "├── 1. Pick a random enemy aircraft from metadata\n",
+    "│      (combat-capable only — no_aa_capability = 0)\n",
+    "│\n",
+    "├── 2. Sample raw feature values\n",
+    "│      your_speed, your_altitude, azimuth, elevation,\n",
+    "│      maneuverability, countermeasure_deployed\n",
+    "│      launch_distance (within that aircraft's missile_range)\n",
+    "│      remaining_distance (between 0 and launch_distance)\n",
+    "│\n",
+    "├── 3. _derive_missile_phase()\n",
+    "│      \"How far has the missile already traveled?\"\n",
+    "│      → returns 0, 1, or 2\n",
+    "│\n",
+    "├── 4. _derive_closure_rate()\n",
+    "│      \"How fast is the gap closing in 3D space?\"\n",
+    "│      → returns a speed in m/s\n",
+    "│\n",
+    "├── 5. _derive_evasion_time()\n",
+    "│      \"How many seconds before it reaches you?\"\n",
+    "│      → returns a float in seconds\n",
+    "│\n",
+    "├── 6. _derive_hit_label()\n",
+    "│      \"After your evasion attempt, does it hit?\"\n",
+    "│      → returns 0 or 1\n",
+    "│\n",
+    "└── 7. Package everything into a dict → one row\n",
+    "```\n",
+    "\n",
+    "---"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "c3d32245",
+   "metadata": {},
+   "source": [
+    "## **Sanity Check - Physics Generator**\n",
+    "\n",
+    "Quick validation that `generate_dataset()` produces clean, physically sensible output.\n",
+    "Run with `N_ROWS = 1_000` before the full 1M generation."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 26,
+   "id": "37b78e79",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import sys\n",
+    "from pathlib import Path\n",
+    "\n",
+    "# Add src/ to path so we can import physics_generator\n",
+    "sys.path.append(str(Path.cwd().resolve().parent.parent / \"src\"))\n",
+    "\n",
+    "from physics_generator import generate_dataset"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "4181ca14",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Done. 1,000 rows saved to /mnt/d/Eakem/Learning ML and DS/Project_ATAS/ATAS_Project/data/synthetic_engagements.csv\n"
+     ]
+    }
+   ],
+   "source": [
+    "# # Running the generator for N_ROWS = 1_000\n",
+    "# df = generate_dataset()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "54d3fc2c",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Rows: 1000\n",
+      "Columns: 16\n",
+      "Shape: (1000, 16)\n"
+     ]
+    }
+   ],
+   "source": [
+    "# import pandas as pd\n",
+    "# df = pd.read_csv(\"../data/synthetic_engagements_1000.csv\")\n",
+    "# print(f\"Rows: {len(df)}\")\n",
+    "# print(f\"Columns: {len(df.columns)}\")\n",
+    "# print(f\"Shape: {df.shape}\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "14d1f643",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/html": [
+       "<div>\n",
+       "<style scoped>\n",
+       "    .dataframe tbody tr th:only-of-type {\n",
+       "        vertical-align: middle;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe tbody tr th {\n",
+       "        vertical-align: top;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe thead th {\n",
+       "        text-align: right;\n",
+       "    }\n",
+       "</style>\n",
+       "<table border=\"1\" class=\"dataframe\">\n",
+       "  <thead>\n",
+       "    <tr style=\"text-align: right;\">\n",
+       "      <th></th>\n",
+       "      <th>launch_distance</th>\n",
+       "      <th>remaining_distance</th>\n",
+       "      <th>closure_rate</th>\n",
+       "      <th>azimuth</th>\n",
+       "      <th>elevation</th>\n",
+       "      <th>missile_phase</th>\n",
+       "      <th>your_speed</th>\n",
+       "      <th>your_altitude</th>\n",
+       "      <th>your_maneuverability</th>\n",
+       "      <th>enemy_altitude</th>\n",
+       "      <th>missile_speed</th>\n",
+       "      <th>missile_range</th>\n",
+       "      <th>enemy_generation</th>\n",
+       "      <th>countermeasure_deployed</th>\n",
+       "      <th>evasion_time</th>\n",
+       "      <th>hit</th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <th>0</th>\n",
+       "      <td>22570.893367</td>\n",
+       "      <td>2961.980067</td>\n",
+       "      <td>1185.763992</td>\n",
+       "      <td>61.005549</td>\n",
+       "      <td>-34.707563</td>\n",
+       "      <td>2</td>\n",
+       "      <td>825.050818</td>\n",
+       "      <td>6512.661972</td>\n",
+       "      <td>1</td>\n",
+       "      <td>3655.798699</td>\n",
+       "      <td>857</td>\n",
+       "      <td>35000</td>\n",
+       "      <td>4.0</td>\n",
+       "      <td>1</td>\n",
+       "      <td>2.229421</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>1</th>\n",
+       "      <td>2367.007196</td>\n",
+       "      <td>1652.662358</td>\n",
+       "      <td>625.878279</td>\n",
+       "      <td>212.466382</td>\n",
+       "      <td>68.630275</td>\n",
+       "      <td>0</td>\n",
+       "      <td>751.777144</td>\n",
+       "      <td>18782.156933</td>\n",
+       "      <td>1</td>\n",
+       "      <td>19508.648175</td>\n",
+       "      <td>857</td>\n",
+       "      <td>35000</td>\n",
+       "      <td>4.0</td>\n",
+       "      <td>0</td>\n",
+       "      <td>2.772577</td>\n",
+       "      <td>1</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>2</th>\n",
+       "      <td>6602.346475</td>\n",
+       "      <td>2558.752556</td>\n",
+       "      <td>871.179381</td>\n",
+       "      <td>281.112640</td>\n",
+       "      <td>45.266055</td>\n",
+       "      <td>1</td>\n",
+       "      <td>104.527357</td>\n",
+       "      <td>3625.183143</td>\n",
+       "      <td>2</td>\n",
+       "      <td>29894.396018</td>\n",
+       "      <td>857</td>\n",
+       "      <td>35000</td>\n",
+       "      <td>4.0</td>\n",
+       "      <td>0</td>\n",
+       "      <td>3.230825</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>3</th>\n",
+       "      <td>18281.792405</td>\n",
+       "      <td>18049.709650</td>\n",
+       "      <td>648.636953</td>\n",
+       "      <td>115.135435</td>\n",
+       "      <td>-37.372887</td>\n",
+       "      <td>0</td>\n",
+       "      <td>617.267514</td>\n",
+       "      <td>5759.552561</td>\n",
+       "      <td>1</td>\n",
+       "      <td>15002.849014</td>\n",
+       "      <td>857</td>\n",
+       "      <td>35000</td>\n",
+       "      <td>4.0</td>\n",
+       "      <td>1</td>\n",
+       "      <td>32.140344</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>4</th>\n",
+       "      <td>9834.379665</td>\n",
+       "      <td>1522.983460</td>\n",
+       "      <td>1088.038782</td>\n",
+       "      <td>316.104680</td>\n",
+       "      <td>-45.805628</td>\n",
+       "      <td>2</td>\n",
+       "      <td>459.932597</td>\n",
+       "      <td>28821.640152</td>\n",
+       "      <td>2</td>\n",
+       "      <td>4643.721244</td>\n",
+       "      <td>857</td>\n",
+       "      <td>35000</td>\n",
+       "      <td>4.0</td>\n",
+       "      <td>1</td>\n",
+       "      <td>1.308767</td>\n",
+       "      <td>0</td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n",
+       "</div>"
+      ],
+      "text/plain": [
+       "   launch_distance  remaining_distance  closure_rate     azimuth  elevation  \\\n",
+       "0     22570.893367         2961.980067   1185.763992   61.005549 -34.707563   \n",
+       "1      2367.007196         1652.662358    625.878279  212.466382  68.630275   \n",
+       "2      6602.346475         2558.752556    871.179381  281.112640  45.266055   \n",
+       "3     18281.792405        18049.709650    648.636953  115.135435 -37.372887   \n",
+       "4      9834.379665         1522.983460   1088.038782  316.104680 -45.805628   \n",
+       "\n",
+       "   missile_phase  your_speed  your_altitude  your_maneuverability  \\\n",
+       "0              2  825.050818    6512.661972                     1   \n",
+       "1              0  751.777144   18782.156933                     1   \n",
+       "2              1  104.527357    3625.183143                     2   \n",
+       "3              0  617.267514    5759.552561                     1   \n",
+       "4              2  459.932597   28821.640152                     2   \n",
+       "\n",
+       "   enemy_altitude  missile_speed  missile_range  enemy_generation  \\\n",
+       "0     3655.798699            857          35000               4.0   \n",
+       "1    19508.648175            857          35000               4.0   \n",
+       "2    29894.396018            857          35000               4.0   \n",
+       "3    15002.849014            857          35000               4.0   \n",
+       "4     4643.721244            857          35000               4.0   \n",
+       "\n",
+       "   countermeasure_deployed  evasion_time  hit  \n",
+       "0                        1      2.229421    0  \n",
+       "1                        0      2.772577    1  \n",
+       "2                        0      3.230825    0  \n",
+       "3                        1     32.140344    0  \n",
+       "4                        1      1.308767    0  "
+      ]
+     },
+     "execution_count": 29,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "# df.head()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "19f98851",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Negative closure_rate: 0\n",
+      "Negative evasion_rate: 0\n"
+     ]
+    }
+   ],
+   "source": [
+    "# # CHecking for negatives values\n",
+    "# print(f\"Negative closure_rate: {(df['closure_rate'] < 0).sum()}\")\n",
+    "# print(f\"Negative evasion_rate: {(df['evasion_time'] < 0).sum()}\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "3b3d531f",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "missile_speed\n",
+       "1372    473\n",
+       "857     154\n",
+       "686     100\n",
+       "1715     97\n",
+       "2058     58\n",
+       "847      40\n",
+       "1544     39\n",
+       "750      20\n",
+       "1475     19\n",
+       "Name: count, dtype: int64"
+      ]
+     },
+     "execution_count": 32,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "# df.missile_speed.value_counts()  # From metadata csv"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "e7bd832a",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/html": [
+       "<div>\n",
+       "<style scoped>\n",
+       "    .dataframe tbody tr th:only-of-type {\n",
+       "        vertical-align: middle;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe tbody tr th {\n",
+       "        vertical-align: top;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe thead th {\n",
+       "        text-align: right;\n",
+       "    }\n",
+       "</style>\n",
+       "<table border=\"1\" class=\"dataframe\">\n",
+       "  <thead>\n",
+       "    <tr style=\"text-align: right;\">\n",
+       "      <th></th>\n",
+       "      <th>0</th>\n",
+       "      <th>1</th>\n",
+       "      <th>2</th>\n",
+       "      <th>3</th>\n",
+       "      <th>4</th>\n",
+       "      <th>5</th>\n",
+       "      <th>6</th>\n",
+       "      <th>7</th>\n",
+       "      <th>8</th>\n",
+       "      <th>9</th>\n",
+       "      <th>...</th>\n",
+       "      <th>990</th>\n",
+       "      <th>991</th>\n",
+       "      <th>992</th>\n",
+       "      <th>993</th>\n",
+       "      <th>994</th>\n",
+       "      <th>995</th>\n",
+       "      <th>996</th>\n",
+       "      <th>997</th>\n",
+       "      <th>998</th>\n",
+       "      <th>999</th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <th>launch_distance</th>\n",
+       "      <td>22570.893367</td>\n",
+       "      <td>2367.007196</td>\n",
+       "      <td>6602.346475</td>\n",
+       "      <td>18281.792405</td>\n",
+       "      <td>9834.379665</td>\n",
+       "      <td>12661.488033</td>\n",
+       "      <td>25901.724724</td>\n",
+       "      <td>8890.584630</td>\n",
+       "      <td>983.734396</td>\n",
+       "      <td>13859.562006</td>\n",
+       "      <td>...</td>\n",
+       "      <td>4981.478801</td>\n",
+       "      <td>32056.695360</td>\n",
+       "      <td>53419.689040</td>\n",
+       "      <td>19021.028127</td>\n",
+       "      <td>128594.277243</td>\n",
+       "      <td>64845.033514</td>\n",
+       "      <td>83532.658041</td>\n",
+       "      <td>146555.411805</td>\n",
+       "      <td>77971.657118</td>\n",
+       "      <td>2253.861997</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>remaining_distance</th>\n",
+       "      <td>2961.980067</td>\n",
+       "      <td>1652.662358</td>\n",
+       "      <td>2558.752556</td>\n",
+       "      <td>18049.709650</td>\n",
+       "      <td>1522.983460</td>\n",
+       "      <td>3400.938832</td>\n",
+       "      <td>17955.494364</td>\n",
+       "      <td>7886.781225</td>\n",
+       "      <td>483.793860</td>\n",
+       "      <td>13391.098584</td>\n",
+       "      <td>...</td>\n",
+       "      <td>1605.088087</td>\n",
+       "      <td>4117.674334</td>\n",
+       "      <td>26058.452754</td>\n",
+       "      <td>3767.541060</td>\n",
+       "      <td>40310.301403</td>\n",
+       "      <td>60646.537540</td>\n",
+       "      <td>13843.586940</td>\n",
+       "      <td>94701.450863</td>\n",
+       "      <td>17925.445277</td>\n",
+       "      <td>1103.757518</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>closure_rate</th>\n",
+       "      <td>1185.763992</td>\n",
+       "      <td>625.878279</td>\n",
+       "      <td>871.179381</td>\n",
+       "      <td>648.636953</td>\n",
+       "      <td>1088.038782</td>\n",
+       "      <td>725.990269</td>\n",
+       "      <td>844.960185</td>\n",
+       "      <td>1020.875119</td>\n",
+       "      <td>1066.347972</td>\n",
+       "      <td>386.272373</td>\n",
+       "      <td>...</td>\n",
+       "      <td>590.826598</td>\n",
+       "      <td>1312.773778</td>\n",
+       "      <td>1663.712597</td>\n",
+       "      <td>438.758129</td>\n",
+       "      <td>1394.591033</td>\n",
+       "      <td>1186.846963</td>\n",
+       "      <td>1211.445463</td>\n",
+       "      <td>1803.707956</td>\n",
+       "      <td>1636.533985</td>\n",
+       "      <td>405.627102</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>azimuth</th>\n",
+       "      <td>61.005549</td>\n",
+       "      <td>212.466382</td>\n",
+       "      <td>281.112640</td>\n",
+       "      <td>115.135435</td>\n",
+       "      <td>316.104680</td>\n",
+       "      <td>259.107852</td>\n",
+       "      <td>211.731931</td>\n",
+       "      <td>283.740414</td>\n",
+       "      <td>349.132944</td>\n",
+       "      <td>176.945221</td>\n",
+       "      <td>...</td>\n",
+       "      <td>240.738983</td>\n",
+       "      <td>191.345225</td>\n",
+       "      <td>72.020487</td>\n",
+       "      <td>206.765055</td>\n",
+       "      <td>69.911001</td>\n",
+       "      <td>251.992973</td>\n",
+       "      <td>243.510094</td>\n",
+       "      <td>41.013277</td>\n",
+       "      <td>305.568549</td>\n",
+       "      <td>117.061738</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>elevation</th>\n",
+       "      <td>-34.707563</td>\n",
+       "      <td>68.630275</td>\n",
+       "      <td>45.266055</td>\n",
+       "      <td>-37.372887</td>\n",
+       "      <td>-45.805628</td>\n",
+       "      <td>-2.007360</td>\n",
+       "      <td>-83.509892</td>\n",
+       "      <td>-13.091010</td>\n",
+       "      <td>-43.024865</td>\n",
+       "      <td>-55.574971</td>\n",
+       "      <td>...</td>\n",
+       "      <td>73.492841</td>\n",
+       "      <td>-85.265763</td>\n",
+       "      <td>-30.731328</td>\n",
+       "      <td>27.430184</td>\n",
+       "      <td>-76.994103</td>\n",
+       "      <td>-14.818047</td>\n",
+       "      <td>-43.173902</td>\n",
+       "      <td>-54.366817</td>\n",
+       "      <td>-32.225887</td>\n",
+       "      <td>-20.136454</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>missile_phase</th>\n",
+       "      <td>2.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>2.000000</td>\n",
+       "      <td>2.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>...</td>\n",
+       "      <td>2.000000</td>\n",
+       "      <td>2.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>2.000000</td>\n",
+       "      <td>2.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>2.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>2.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>your_speed</th>\n",
+       "      <td>825.050818</td>\n",
+       "      <td>751.777144</td>\n",
+       "      <td>104.527357</td>\n",
+       "      <td>617.267514</td>\n",
+       "      <td>459.932597</td>\n",
+       "      <td>693.742453</td>\n",
+       "      <td>125.237980</td>\n",
+       "      <td>708.341294</td>\n",
+       "      <td>291.592189</td>\n",
+       "      <td>833.848608</td>\n",
+       "      <td>...</td>\n",
+       "      <td>685.281835</td>\n",
+       "      <td>731.899168</td>\n",
+       "      <td>451.183218</td>\n",
+       "      <td>527.764085</td>\n",
+       "      <td>292.249519</td>\n",
+       "      <td>619.546047</td>\n",
+       "      <td>493.576101</td>\n",
+       "      <td>982.046031</td>\n",
+       "      <td>537.593520</td>\n",
+       "      <td>656.393231</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>your_altitude</th>\n",
+       "      <td>6512.661972</td>\n",
+       "      <td>18782.156933</td>\n",
+       "      <td>3625.183143</td>\n",
+       "      <td>5759.552561</td>\n",
+       "      <td>28821.640152</td>\n",
+       "      <td>9790.692112</td>\n",
+       "      <td>1510.779084</td>\n",
+       "      <td>14553.522284</td>\n",
+       "      <td>4020.288191</td>\n",
+       "      <td>15714.406817</td>\n",
+       "      <td>...</td>\n",
+       "      <td>4796.945631</td>\n",
+       "      <td>4000.690960</td>\n",
+       "      <td>7211.059032</td>\n",
+       "      <td>2657.273049</td>\n",
+       "      <td>16145.505669</td>\n",
+       "      <td>23094.841893</td>\n",
+       "      <td>24159.528537</td>\n",
+       "      <td>26605.676886</td>\n",
+       "      <td>22146.398479</td>\n",
+       "      <td>1624.780195</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>your_maneuverability</th>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>2.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>2.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>...</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>2.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>2.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>enemy_altitude</th>\n",
+       "      <td>3655.798699</td>\n",
+       "      <td>19508.648175</td>\n",
+       "      <td>29894.396018</td>\n",
+       "      <td>15002.849014</td>\n",
+       "      <td>4643.721244</td>\n",
+       "      <td>26557.889641</td>\n",
+       "      <td>11515.140368</td>\n",
+       "      <td>4119.563278</td>\n",
+       "      <td>2974.936095</td>\n",
+       "      <td>28617.057195</td>\n",
+       "      <td>...</td>\n",
+       "      <td>13657.745440</td>\n",
+       "      <td>24585.421109</td>\n",
+       "      <td>28243.550517</td>\n",
+       "      <td>16566.700223</td>\n",
+       "      <td>15476.262795</td>\n",
+       "      <td>26693.600073</td>\n",
+       "      <td>17534.961728</td>\n",
+       "      <td>7711.296412</td>\n",
+       "      <td>4263.977523</td>\n",
+       "      <td>17091.003655</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>missile_speed</th>\n",
+       "      <td>857.000000</td>\n",
+       "      <td>857.000000</td>\n",
+       "      <td>857.000000</td>\n",
+       "      <td>857.000000</td>\n",
+       "      <td>857.000000</td>\n",
+       "      <td>857.000000</td>\n",
+       "      <td>857.000000</td>\n",
+       "      <td>857.000000</td>\n",
+       "      <td>857.000000</td>\n",
+       "      <td>857.000000</td>\n",
+       "      <td>...</td>\n",
+       "      <td>686.000000</td>\n",
+       "      <td>1372.000000</td>\n",
+       "      <td>1544.000000</td>\n",
+       "      <td>857.000000</td>\n",
+       "      <td>1372.000000</td>\n",
+       "      <td>1372.000000</td>\n",
+       "      <td>1372.000000</td>\n",
+       "      <td>1372.000000</td>\n",
+       "      <td>1372.000000</td>\n",
+       "      <td>686.000000</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>missile_range</th>\n",
+       "      <td>35000.000000</td>\n",
+       "      <td>35000.000000</td>\n",
+       "      <td>35000.000000</td>\n",
+       "      <td>35000.000000</td>\n",
+       "      <td>35000.000000</td>\n",
+       "      <td>35000.000000</td>\n",
+       "      <td>35000.000000</td>\n",
+       "      <td>35000.000000</td>\n",
+       "      <td>35000.000000</td>\n",
+       "      <td>35000.000000</td>\n",
+       "      <td>...</td>\n",
+       "      <td>8000.000000</td>\n",
+       "      <td>160000.000000</td>\n",
+       "      <td>105000.000000</td>\n",
+       "      <td>40000.000000</td>\n",
+       "      <td>160000.000000</td>\n",
+       "      <td>65000.000000</td>\n",
+       "      <td>160000.000000</td>\n",
+       "      <td>200000.000000</td>\n",
+       "      <td>110000.000000</td>\n",
+       "      <td>8000.000000</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>enemy_generation</th>\n",
+       "      <td>4.000000</td>\n",
+       "      <td>4.000000</td>\n",
+       "      <td>4.000000</td>\n",
+       "      <td>4.000000</td>\n",
+       "      <td>4.000000</td>\n",
+       "      <td>4.000000</td>\n",
+       "      <td>4.000000</td>\n",
+       "      <td>4.000000</td>\n",
+       "      <td>4.000000</td>\n",
+       "      <td>4.000000</td>\n",
+       "      <td>...</td>\n",
+       "      <td>4.000000</td>\n",
+       "      <td>5.000000</td>\n",
+       "      <td>4.500000</td>\n",
+       "      <td>4.000000</td>\n",
+       "      <td>4.000000</td>\n",
+       "      <td>4.500000</td>\n",
+       "      <td>4.500000</td>\n",
+       "      <td>4.500000</td>\n",
+       "      <td>4.000000</td>\n",
+       "      <td>4.000000</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>countermeasure_deployed</th>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>...</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>evasion_time</th>\n",
+       "      <td>2.229421</td>\n",
+       "      <td>2.772577</td>\n",
+       "      <td>3.230825</td>\n",
+       "      <td>32.140344</td>\n",
+       "      <td>1.308767</td>\n",
+       "      <td>4.599058</td>\n",
+       "      <td>23.375118</td>\n",
+       "      <td>8.922964</td>\n",
+       "      <td>0.453692</td>\n",
+       "      <td>40.040966</td>\n",
+       "      <td>...</td>\n",
+       "      <td>2.667103</td>\n",
+       "      <td>2.771440</td>\n",
+       "      <td>17.229116</td>\n",
+       "      <td>8.430119</td>\n",
+       "      <td>24.569035</td>\n",
+       "      <td>53.653813</td>\n",
+       "      <td>10.684553</td>\n",
+       "      <td>60.641844</td>\n",
+       "      <td>10.753401</td>\n",
+       "      <td>3.142886</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>hit</th>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>...</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>1.000000</td>\n",
+       "      <td>0.000000</td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n",
+       "<p>16 rows × 1000 columns</p>\n",
+       "</div>"
+      ],
+      "text/plain": [
+       "                                  0             1             2    \\\n",
+       "launch_distance          22570.893367   2367.007196   6602.346475   \n",
+       "remaining_distance        2961.980067   1652.662358   2558.752556   \n",
+       "closure_rate              1185.763992    625.878279    871.179381   \n",
+       "azimuth                     61.005549    212.466382    281.112640   \n",
+       "elevation                  -34.707563     68.630275     45.266055   \n",
+       "missile_phase                2.000000      0.000000      1.000000   \n",
+       "your_speed                 825.050818    751.777144    104.527357   \n",
+       "your_altitude             6512.661972  18782.156933   3625.183143   \n",
+       "your_maneuverability         1.000000      1.000000      2.000000   \n",
+       "enemy_altitude            3655.798699  19508.648175  29894.396018   \n",
+       "missile_speed              857.000000    857.000000    857.000000   \n",
+       "missile_range            35000.000000  35000.000000  35000.000000   \n",
+       "enemy_generation             4.000000      4.000000      4.000000   \n",
+       "countermeasure_deployed      1.000000      0.000000      0.000000   \n",
+       "evasion_time                 2.229421      2.772577      3.230825   \n",
+       "hit                          0.000000      1.000000      0.000000   \n",
+       "\n",
+       "                                  3             4             5    \\\n",
+       "launch_distance          18281.792405   9834.379665  12661.488033   \n",
+       "remaining_distance       18049.709650   1522.983460   3400.938832   \n",
+       "closure_rate               648.636953   1088.038782    725.990269   \n",
+       "azimuth                    115.135435    316.104680    259.107852   \n",
+       "elevation                  -37.372887    -45.805628     -2.007360   \n",
+       "missile_phase                0.000000      2.000000      2.000000   \n",
+       "your_speed                 617.267514    459.932597    693.742453   \n",
+       "your_altitude             5759.552561  28821.640152   9790.692112   \n",
+       "your_maneuverability         1.000000      2.000000      1.000000   \n",
+       "enemy_altitude           15002.849014   4643.721244  26557.889641   \n",
+       "missile_speed              857.000000    857.000000    857.000000   \n",
+       "missile_range            35000.000000  35000.000000  35000.000000   \n",
+       "enemy_generation             4.000000      4.000000      4.000000   \n",
+       "countermeasure_deployed      1.000000      1.000000      0.000000   \n",
+       "evasion_time                32.140344      1.308767      4.599058   \n",
+       "hit                          0.000000      0.000000      1.000000   \n",
+       "\n",
+       "                                  6             7             8    \\\n",
+       "launch_distance          25901.724724   8890.584630    983.734396   \n",
+       "remaining_distance       17955.494364   7886.781225    483.793860   \n",
+       "closure_rate               844.960185   1020.875119   1066.347972   \n",
+       "azimuth                    211.731931    283.740414    349.132944   \n",
+       "elevation                  -83.509892    -13.091010    -43.024865   \n",
+       "missile_phase                0.000000      0.000000      1.000000   \n",
+       "your_speed                 125.237980    708.341294    291.592189   \n",
+       "your_altitude             1510.779084  14553.522284   4020.288191   \n",
+       "your_maneuverability         1.000000      1.000000      1.000000   \n",
+       "enemy_altitude           11515.140368   4119.563278   2974.936095   \n",
+       "missile_speed              857.000000    857.000000    857.000000   \n",
+       "missile_range            35000.000000  35000.000000  35000.000000   \n",
+       "enemy_generation             4.000000      4.000000      4.000000   \n",
+       "countermeasure_deployed      0.000000      1.000000      0.000000   \n",
+       "evasion_time                23.375118      8.922964      0.453692   \n",
+       "hit                          1.000000      0.000000      1.000000   \n",
+       "\n",
+       "                                  9    ...           990            991  \\\n",
+       "launch_distance          13859.562006  ...   4981.478801   32056.695360   \n",
+       "remaining_distance       13391.098584  ...   1605.088087    4117.674334   \n",
+       "closure_rate               386.272373  ...    590.826598    1312.773778   \n",
+       "azimuth                    176.945221  ...    240.738983     191.345225   \n",
+       "elevation                  -55.574971  ...     73.492841     -85.265763   \n",
+       "missile_phase                0.000000  ...      2.000000       2.000000   \n",
+       "your_speed                 833.848608  ...    685.281835     731.899168   \n",
+       "your_altitude            15714.406817  ...   4796.945631    4000.690960   \n",
+       "your_maneuverability         0.000000  ...      0.000000       2.000000   \n",
+       "enemy_altitude           28617.057195  ...  13657.745440   24585.421109   \n",
+       "missile_speed              857.000000  ...    686.000000    1372.000000   \n",
+       "missile_range            35000.000000  ...   8000.000000  160000.000000   \n",
+       "enemy_generation             4.000000  ...      4.000000       5.000000   \n",
+       "countermeasure_deployed      1.000000  ...      1.000000       0.000000   \n",
+       "evasion_time                40.040966  ...      2.667103       2.771440   \n",
+       "hit                          0.000000  ...      0.000000       1.000000   \n",
+       "\n",
+       "                                   992           993            994  \\\n",
+       "launch_distance           53419.689040  19021.028127  128594.277243   \n",
+       "remaining_distance        26058.452754   3767.541060   40310.301403   \n",
+       "closure_rate               1663.712597    438.758129    1394.591033   \n",
+       "azimuth                      72.020487    206.765055      69.911001   \n",
+       "elevation                   -30.731328     27.430184     -76.994103   \n",
+       "missile_phase                 1.000000      2.000000       2.000000   \n",
+       "your_speed                  451.183218    527.764085     292.249519   \n",
+       "your_altitude              7211.059032   2657.273049   16145.505669   \n",
+       "your_maneuverability          1.000000      0.000000       0.000000   \n",
+       "enemy_altitude            28243.550517  16566.700223   15476.262795   \n",
+       "missile_speed              1544.000000    857.000000    1372.000000   \n",
+       "missile_range            105000.000000  40000.000000  160000.000000   \n",
+       "enemy_generation              4.500000      4.000000       4.000000   \n",
+       "countermeasure_deployed       0.000000      1.000000       0.000000   \n",
+       "evasion_time                 17.229116      8.430119      24.569035   \n",
+       "hit                           0.000000      0.000000       1.000000   \n",
+       "\n",
+       "                                  995            996            997  \\\n",
+       "launch_distance          64845.033514   83532.658041  146555.411805   \n",
+       "remaining_distance       60646.537540   13843.586940   94701.450863   \n",
+       "closure_rate              1186.846963    1211.445463    1803.707956   \n",
+       "azimuth                    251.992973     243.510094      41.013277   \n",
+       "elevation                  -14.818047     -43.173902     -54.366817   \n",
+       "missile_phase                0.000000       2.000000       1.000000   \n",
+       "your_speed                 619.546047     493.576101     982.046031   \n",
+       "your_altitude            23094.841893   24159.528537   26605.676886   \n",
+       "your_maneuverability         1.000000       2.000000       1.000000   \n",
+       "enemy_altitude           26693.600073   17534.961728    7711.296412   \n",
+       "missile_speed             1372.000000    1372.000000    1372.000000   \n",
+       "missile_range            65000.000000  160000.000000  200000.000000   \n",
+       "enemy_generation             4.500000       4.500000       4.500000   \n",
+       "countermeasure_deployed      1.000000       0.000000       0.000000   \n",
+       "evasion_time                53.653813      10.684553      60.641844   \n",
+       "hit                          0.000000       0.000000       1.000000   \n",
+       "\n",
+       "                                   998           999  \n",
+       "launch_distance           77971.657118   2253.861997  \n",
+       "remaining_distance        17925.445277   1103.757518  \n",
+       "closure_rate               1636.533985    405.627102  \n",
+       "azimuth                     305.568549    117.061738  \n",
+       "elevation                   -32.225887    -20.136454  \n",
+       "missile_phase                 2.000000      1.000000  \n",
+       "your_speed                  537.593520    656.393231  \n",
+       "your_altitude             22146.398479   1624.780195  \n",
+       "your_maneuverability          1.000000      0.000000  \n",
+       "enemy_altitude             4263.977523  17091.003655  \n",
+       "missile_speed              1372.000000    686.000000  \n",
+       "missile_range            110000.000000   8000.000000  \n",
+       "enemy_generation              4.000000      4.000000  \n",
+       "countermeasure_deployed       0.000000      1.000000  \n",
+       "evasion_time                 10.753401      3.142886  \n",
+       "hit                           1.000000      0.000000  \n",
+       "\n",
+       "[16 rows x 1000 columns]"
+      ]
+     },
+     "execution_count": 33,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "# df.T"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "afbdd5d1",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "       launch_distance  remaining_distance  closure_rate  evasion_time  \\\n",
+      "count      1000.000000         1000.000000   1000.000000   1000.000000   \n",
+      "mean      64915.002129        31821.659117   1266.905104     24.195931   \n",
+      "std       74767.756596        45271.208405    482.164719     31.584890   \n",
+      "min         669.396283            2.403273      4.027710      0.006151   \n",
+      "25%       10515.861139         3803.500738    872.679155      4.102251   \n",
+      "50%       39195.914463        14983.450620   1327.613182     11.872938   \n",
+      "75%       93328.236924        40924.330595   1573.072050     31.334417   \n",
+      "max      395478.660029       389130.451998   2858.542394    284.080232   \n",
+      "\n",
+      "        your_speed  your_altitude  \n",
+      "count  1000.000000    1000.000000  \n",
+      "mean    521.111816   14957.021066  \n",
+      "std     263.082784    8549.393126  \n",
+      "min      62.358418      72.422424  \n",
+      "25%     297.653412    7338.047586  \n",
+      "50%     518.151709   15324.311391  \n",
+      "75%     751.001456   22210.948597  \n",
+      "max     982.046031   29963.768605  \n"
+     ]
+    }
+   ],
+   "source": [
+    "# # Check feature ranges are within expected bounds\n",
+    "# print(df[['launch_distance', 'remaining_distance', 'closure_rate', \n",
+    "#           'evasion_time', 'your_speed', 'your_altitude']].describe())"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "ea4cf1bf",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "remaining > launch: 0\n"
+     ]
+    }
+   ],
+   "source": [
+    "# # Check remaining_distance never exceeds launch_distance\n",
+    "# print(f\"remaining > launch: {(df['remaining_distance'] > df['launch_distance']).sum()}\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "d16d1952",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "hit\n",
+      "0    602\n",
+      "1    398\n",
+      "Name: count, dtype: int64\n",
+      "hit\n",
+      "0    0.602\n",
+      "1    0.398\n",
+      "Name: proportion, dtype: float64\n"
+     ]
+    }
+   ],
+   "source": [
+    "# # Check hit label distribution\n",
+    "# print(df['hit'].value_counts())\n",
+    "# print(df['hit'].value_counts(normalize=True))  # It shows the distribution as percentages instead of counts."
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "6955deb6",
+   "metadata": {},
+   "source": [
+    "- `remaining > launch: 0` - physical constraint holds ✅\n",
+    "- All ranges look sensible - speeds, altitudes, distances all within expected bounds ✅\n",
+    "- evasion_time min is 0.006s (very close missile, terminal phase) - realistic ✅\n",
+    "- **Hit distribution:** 60% miss, 40% hit - reasonable balance, not skewed ✅\n",
+    "\n",
+    "---"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "a56f6f02",
+   "metadata": {},
+   "source": [
+    "## **Full Dataset Generation**\n",
+    "\n",
+    "Sanity check passed on **1,000** rows, no negatives, no constraint violations.\n",
+    "\n",
+    "Changing `N_ROWS` to `1_000_000` and running full generation."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 40,
+   "id": "c8722b7c",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import importlib\n",
+    "import physics_generator\n",
+    "\n",
+    "# Reload the module to pick up any changes made to the script\n",
+    "importlib.reload(physics_generator)\n",
+    "from physics_generator import generate_dataset"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 41,
+   "id": "0ef8538e",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Done. 1,000,000 rows saved to /mnt/d/Eakem/Learning ML and DS/Project_ATAS/ATAS_Project/data/synthetic_engagements.csv\n"
+     ]
+    }
+   ],
+   "source": [
+    "# Generate the data with ~1M rows\n",
+    "df_full = generate_dataset()"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "881b8593",
+   "metadata": {},
+   "source": [
+    "**Now from here moving to Hit and ETA classifier notebooks**\n",
+    "\n",
+    "---"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "env",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.11.15"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

src/classifier.py

@@ -0,0 +1,190 @@
+"""
+classifier.py
+-------------
+Loads the trained EfficientNetV2-L aircraft classifier and runs inference
+with Test Time Augmentation (TTA).
+
+Responsibilities:
+    - Set TF_USE_LEGACY_KERAS before any TensorFlow import
+    - Load the saved .keras model from disk once at module import time
+    - Accept an image file path and return a predicted aircraft class name
+    - Apply TTA (N=15) using horizontal flip, vertical flip,
+      brightness and contrast jitter
+
+Used by: main.py (FastAPI endpoint)
+"""
+
+# Must be set before any TensorFlow or TF Hub import.
+# The model was trained with tf_keras (Keras 2) and saved in that format.
+# Without this flag, TF Hub's KerasLayer will fail to deserialize correctly
+# on TF 2.13+ which defaults to Keras 3.
+import os
+os.environ["TF_USE_LEGACY_KERAS"] = "1"
+
+
+import tensorflow as tf
+import tensorflow_hub as hub
+from src.schemas import MODEL_PATHS
+import numpy as np
+
+# Prevent TF from allocating all VRAM at startup — allocate as needed instead.
+# Skipped automatically on CPU-only environments (e.g. Hugging Face Spaces).
+gpus = tf.config.list_physical_devices('GPU')
+if gpus:
+    tf.config.experimental.set_memory_growth(gpus[0], True)
+
+# Load the Keras model
+
+try:
+    classifier_model = tf.keras.models.load_model(
+        MODEL_PATHS['classifier'],
+        custom_objects={"KerasLayer": hub.KerasLayer}
+    )
+    print("Classifier model loaded successfully.")
+except FileNotFoundError:
+    print(f"Error: model file not found at {MODEL_PATHS['classifier']}")
+except Exception as e:
+    print(f"Error loading classifier model: {e}")
+
+
+CLASS_NAMES = [
+    'A10', 'A400M', 'AG600', 'AH64', 'AKINCI', 'AV8B', 'An124', 'An22',
+    'An225', 'An72', 'B1', 'B2', 'B21', 'B52', 'Be200', 'C1', 'C130',
+    'C17', 'C2', 'C390', 'C5', 'CH47', 'CH53', 'CL415', 'E2', 'E7',
+    'EF2000', 'EMB314', 'F117', 'F14', 'F15', 'F16', 'F18', 'F2',
+    'F22', 'F35', 'F4', 'FCK1', 'H6', 'Il76', 'J10', 'J20', 'J35',
+    'J36', 'J50', 'JAS39', 'JF17', 'JH7', 'KAAN', 'KC135', 'KF21',
+    'KIZILELMA', 'KJ600', 'Ka27', 'Ka52', 'MQ20', 'MQ25', 'MQ28',
+    'MQ9', 'Mi24', 'Mi26', 'Mi28', 'Mi8', 'Mig29', 'Mig31',
+    'Mirage2000', 'NH90', 'P3', 'RQ4', 'Rafale', 'SR71', 'Su24',
+    'Su25', 'Su34', 'Su47', 'Su57', 'T50', 'TB001', 'TB2', 'Tejas',
+    'Tornado', 'Tu160', 'Tu22M', 'Tu95', 'U2', 'UH60', 'US2', 'V22',
+    'V280', 'Vulcan', 'WZ10', 'WZ7', 'WZ9', 'X29', 'X32', 'XB70',
+    'XQ58', 'Y20', 'YF23', 'Z10', 'Z19'
+]
+
+
+# Create a function to preprocess the image
+def _process_image(image_path:str, image_size=(480, 480)):
+    """
+    Load and preprocess a single image from disk.
+    - Reads raw bytes from the filepath
+    - Decodes into an RGB tensor
+    - Resizes to the target image size
+    - Normalizes pixel values to [0, 1]
+
+    Args:
+        image_path: Path to the image file (string)
+        image_size: Target image size
+
+    Returns:
+        Preprocessed image tensor
+    """
+
+    # Read the image as raw bytes from the filepath
+    image = tf.io.read_file(image_path)
+
+    # Decode into an RGB tensor
+    image = tf.image.decode_jpeg(image, channels=3)
+
+    # Convert pixel values from 0-255 to 0-1
+    image = tf.image.convert_image_dtype(image, tf.float32)
+
+    # Resize the image to the desired shape
+    image = tf.image.resize(image, image_size)
+
+    return image
+
+
+
+# Function to apply augmentation to Validation or Test images
+def _tta_augment(image):
+    """
+    Applies random augmentations to a single image at inference time.
+    Matches training augmentation exactly: flips, brightness, contrast.
+    Used during TTA to generate N augmented versions of the same image.
+
+    Args:
+        image: preprocessed image tensor, shape (Hight, Width, 3 channels)
+
+    Returns:
+        Augmented image tensor, same shape as input.
+    """
+    
+    image = tf.image.random_flip_left_right(image)
+    image = tf.image.random_flip_up_down(image)
+    image = tf.image.random_brightness(image, max_delta=0.1)
+    image = tf.image.random_contrast(image, lower=0.9, upper=1.1)
+    
+    return image   
+
+
+# Function to do prediction and apply augmentaion during predictions
+
+def _tta_predict(model, image_path, n_augments=15):
+    """
+    Performs Test-Time Augmentation (TTA) prediction on a single image.
+
+    This function loads an image, generates multiple augmented versions of it 
+    (creating an on-the-fly batch), and passes the entire batch through the 
+    model in a single forward pass. The final output is the average of the 
+    softmax probabilities across all versions, which improves prediction robustness.
+
+    Args:
+        model (tf.keras.Model): The loaded aircraft classifier model.
+        image_path (str or Path): The file path to the input image.
+        n_augments (int, optional): The total number of image variations to 
+                                    evaluate (1 original + N-1 augmentations). Defaults to 7.
+
+    Returns:
+        numpy.ndarray: A 1D array containing the averaged softmax probability 
+        vector for the target image.
+    """
+    # 1. Load and preprocess the image
+    image = _process_image(image_path=image_path)
+    
+    # 2. Build batch of N versions
+    versions = [image]
+    for _ in range(n_augments - 1):
+        versions.append(_tta_augment(image))
+        
+    # 3. Stack into one batch (N, H, W, 3) - default batch size 7
+    batch = tf.stack(versions, axis=0)
+    
+    # 4. One predict call
+    # FAST INFERENCE: Call the model directly instead of .predict()
+    # training=False ensures layers like Dropout and BatchNorm behave correctly for inference (Using a trained model to make predictions on new data)
+    predictions = model(batch, training=False).numpy()
+    
+    # predictions = model.predict(batch, verbose=2)
+    
+    # 5. Average the N Softmax vectors
+    avg_pred = np.mean(predictions, axis=0)
+    
+    return avg_pred 
+
+
+
+
+# Function to predict the class name of the aircraft
+def predict_aircraft(image_path:str):
+    """
+    Run TTA inference on a single image and return the predicted aircraft class name.
+
+    Args:
+        image_path (str): Path to the input image file.
+
+    Returns:
+        str: Predicted aircraft class name (e.g. 'F22', 'Rafale').
+    """
+    
+    # Make the predictions
+    average_tta_predictions = _tta_predict(model=classifier_model,
+                                           image_path=image_path,
+                                           n_augments=15)
+    
+    # Get the maximum probaility class index
+    class_label_idx = np.argmax(average_tta_predictions)
+    class_label = CLASS_NAMES[class_label_idx]
+    
+    return class_label

src/decision.py

@@ -0,0 +1,118 @@
+"""
+decision.py — Tactical Decision Layer
+---------------------------------------
+Receives ETA and hit probability from models.py and converts them into
+a human-readable threat level and tactical recommendation.
+
+This module knows nothing about models, physics, or metadata.
+Its only job is: predictions in, decision out.
+"""
+
+from src.schemas import DECISION_THRESHOLDS
+
+def get_recommendation(eta_seconds, hit_probability, no_aa_capability, enemy_generation, friendly_generation):
+    """
+    Converts model predictions and platform data into a tactical recommendation.
+
+    Args:
+        eta_seconds (float): Predicted evasion time in seconds, clipped at 0.
+        hit_probability (float): Probability of hit after evasion maneuvers (0.0 - 1.0).
+        no_aa_capability (int): 1 if enemy has no air-to-air capability, 0 otherwise.
+        enemy_generation (float): Enemy aircraft generation (e.g. 4.0, 4.5, 5.0).
+        friendly_generation (float): Friendly aircraft generation (e.g. 4.0, 4.5, 5.0).
+
+    Returns:
+        str: Tactical recommendation string for the pilot.
+    """
+    
+    # Check first air-to-air capability
+    if no_aa_capability == 1:
+        return "NON-THREAT PLATFORM - MONITOR"
+    
+    # Calculate the generational gap
+    gen_gap = friendly_generation - enemy_generation
+    
+    # Define combat posture
+    if gen_gap >= DECISION_THRESHOLDS['gen_advantage_threshold']:
+        combat_posture = "Friendly advantage"
+    elif gen_gap <= -DECISION_THRESHOLDS['gen_disadvantage_threshold']:
+        combat_posture = "Enemy advantage"
+    else:
+        combat_posture = "Parity"
+        
+    # Get the ETA status
+    if eta_seconds <= DECISION_THRESHOLDS['eta_critical']:
+        eta_status = "Low ETA"
+    else:
+        eta_status = "High ETA"
+        
+        
+    # Get the Hit probability status
+    if hit_probability >= DECISION_THRESHOLDS['hit_prob_high']:
+        hit_status = "High hit probability"
+    elif DECISION_THRESHOLDS['hit_prob_medium'] <= hit_probability <= DECISION_THRESHOLDS['hit_prob_high']:
+        hit_status = "Medium hit probability"
+    else:
+        hit_status = "Low hit probability"
+        
+        
+    # Get the recommendation based on decisions
+    if eta_status == "Low ETA":
+        if hit_status == "High hit probability":
+            # Critical threat - missile close, high chance of hit
+            if combat_posture == "Friendly advantage":
+                return "BREAK HARD + DEPLOY CM + COUNTER-ENGAGE"
+            elif combat_posture == "Parity":
+                return "BREAK HARD + DEPLOY CM"
+            else:
+                # Outmatched - survive first, disengage
+                return "BREAK HARD + DEPLOY CM + DISENGAGE IMMEDIATELY"
+
+        elif hit_status == "Medium hit probability":
+            # Elevated threat - missile close, uncertain outcome
+            if combat_posture == "Friendly advantage":
+                return "DEPLOY CM + DEFENSIVE MANEUVERS + ENGAGE"
+            elif combat_posture == "Parity":
+                return "DEPLOY CM + DEFENSIVE MANEUVERS"
+            else:
+                # Disadvantaged - don't risk engagement
+                return "DEPLOY CM + DEFENSIVE MANEUVERS + DISENGAGE"
+
+        else:
+            # Missile close but likely evadable - redirect and reassess
+            if combat_posture == "Friendly advantage":
+                return "DEPLOY CM + CHANGE FLIGHT PATH + ENGAGE"
+            elif combat_posture == "Parity":
+                return "DEPLOY CM + CHANGE FLIGHT PATH"
+            else:
+                return "DEPLOY CM + CHANGE FLIGHT PATH + DISENGAGE"
+
+    else:
+        if hit_status == "High hit probability":
+            # Time available but threat is serious - prepare now
+            if combat_posture == "Friendly advantage":
+                return "DEFENSIVE MANEUVERS + PREPARE CM + COUNTER-ENGAGE"
+            elif combat_posture == "Parity":
+                return "DEFENSIVE MANEUVERS + PREPARE CM"
+            else:
+                # Outmatched - use the time to create distance
+                return "DEFENSIVE MANEUVERS + PREPARE CM + MAINTAIN DISTANCE"
+
+        elif hit_status == "Medium hit probability":
+            # Uncertain threat with time to monitor - stay ready
+            if combat_posture == "Friendly advantage":
+                return "PREPARE CM + MONITOR + ENGAGE WHEN READY"
+            elif combat_posture == "Parity":
+                return "PREPARE CM + MONITOR TRAJECTORY"
+            else:
+                return "PREPARE CM + MAINTAIN DISTANCE"
+
+        else:
+            # Low threat - missile unlikely to hit, time available
+            if combat_posture == "Friendly advantage":
+                return "STAY SHARP + CLOSE AND ENGAGE"
+            elif combat_posture == "Parity":
+                return "STAY SHARP + MONITOR TRAJECTORY"
+            else:
+                # Disadvantaged even in low threat - keep distance
+                return "STAY SHARP + MAINTAIN DISTANCE"

src/metadata.py

@@ -0,0 +1,40 @@
+"""
+What is metadata.py
+
+When the classifier identifies an aircraft say "F22", the pipeline needs its specs: missile speed, missile range, enemy generation, maneuverability. Those live in your aircraft_metadata.csv.
+metadata.py has one job: take an aircraft name, look it up in the CSV, return its specs as a dictionary.
+"""
+
+import pandas as pd
+from src.schemas import BASE_DIR
+
+CSV_PATH = BASE_DIR / "data" / "aircraft_metadata.csv"
+
+# Load the aircraft metadata
+aircraft_df = pd.read_csv(CSV_PATH)
+
+# Function to get the Aircraft specs
+def get_aircraft_metadata(aircraft_name:str) -> dict:
+    """
+    Look up aircraft specifications from the metadata CSV by aircraft name.
+
+    Args:
+        aircraft_name (str): Aircraft name (case-insensitive). e.g. "F22", "rafale"
+
+    Returns:
+        dict: All CSV columns for that aircraft. Empty dict if not found.
+    """
+    
+    # Upper case everything to search
+    aircraft_name = aircraft_name.upper()
+    
+    # Find the row
+    aircraft_row = aircraft_df[aircraft_df['aircraft'].str.upper() == aircraft_name]
+    
+    # Handling egde cases
+    if not aircraft_row.empty:
+        return aircraft_row.iloc[0].to_dict()
+    else:
+        return {}
+
+    

src/models.py

@@ -0,0 +1,71 @@
+"""
+models.py — Inference Layer
+----------------------------
+Loads the trained ETA regressor and hit classifier from disk once at module
+import time. Exposes a single function that accepts a feature dict, converts
+it to an ordered numpy array using FEATURE_COLUMNS, runs both models, and
+returns ETA in seconds and hit probability as a float.
+
+This module knows nothing about physics, metadata, or decisions.
+Its only job is: feature array in, predictions out.
+"""
+
+
+from src.schemas import FEATURE_COLUMNS, MODEL_PATHS
+from joblib import load
+import numpy as np
+
+# Empty dictionary to hold the active loaded model
+loaded_model = {}
+
+# Load the model once in the memory
+for model_name, path in MODEL_PATHS.items():
+    if model_name in ('eta', 'hit'):
+        try:
+            loaded_model[model_name] = load(path)
+            print(f"Successfully loaded: {model_name} model")
+        except FileNotFoundError:
+            print(f"Error: The file at {path} could not be found.")
+        except Exception as e:
+            print(f"An error occurred while loading model {model_name}: {e}")
+            
+            
+            
+# Function to make predictions using loaded models
+def make_predictions(feature_dict):
+    """
+    Converts a feature dict to a numpy array and runs both models.
+
+    Args:
+        feature_dict (dict): 14-feature dict in FEATURE_COLUMNS order,
+                             produced by build_feature_array().
+
+    Returns:
+        dict: {
+            "eta_seconds": float,       # predicted evasion time, clipped at 0
+            "hit_probability": float    # probability of hit after evasion (0.0 - 1.0)
+        }
+    """
+    
+    # Convert the dictionary into ndarray using the order of FEATURE_COLUMNS
+    feature_values = np.array([feature_dict[col] for col in FEATURE_COLUMNS])
+    
+    # Reshape the the feature_values into (1, 14) as a line of table
+    feature_values = feature_values.reshape(1, -1)
+    
+    # Performing predictions
+    for model_name, model in loaded_model.items():
+        if model_name == 'eta':
+            eta_prediction = model.predict(feature_values)
+            # Clip the negative time value to 0
+            eta_prediction = np.maximum(eta_prediction, 0)[0]
+            
+        else:
+            hit_prediction = model.predict_proba(feature_values)[0][1]   # [probability of miss, probability of hit]
+            
+            
+    return {
+        "eta_seconds": float(eta_prediction),
+        "hit_probability": float(hit_prediction)
+    }
+    

src/physics_generator.py

@@ -0,0 +1,461 @@
+# Importing the required libraries
+import numpy as np
+import pandas as pd
+import math
+from pathlib import Path
+
+# ── Paths ─────────────────────────────────────────────────────────────────────
+PROJECT_ROOT  = Path(__file__).resolve().parent.parent
+DATA_ROOT     = PROJECT_ROOT / "data"
+OUTPUT_PATH   = DATA_ROOT / "synthetic_engagements.csv"
+METADATA_PATH = DATA_ROOT / "aircraft_metadata.csv"
+
+# ── Generation config ─────────────────────────────────────────────────────────
+N_ROWS = 1_000_000           # Python ignores underscores in numbers, just readability
+
+# ── Feature ranges ────────────────────────────────────────────────────────────
+# Launch distance: how far the missile was fired from (metres)
+# Floor is 500m — below this, guns are more effective than missiles and
+# the missile doesn't have enough distance to arm and track properly
+# Upper bound is capped to each aircraft's missile_range at row generation
+LAUNCH_DISTANCE_MIN = 500
+
+# Your aircraft speed range - 61 m/s (TB2 drone) → 983 m/s (SR-71), full metadata range
+YOUR_SPEED_RANGE       = (61, 983)
+
+# Your aircraft altitude (metres)
+# 0 = ground level, 30,000m = upper combat/operational ceiling
+# SR-71 operates at ~24,000m, most fighters top out around 20,000m
+# 30,000m gives headroom for all 102 aircraft in metadata
+YOUR_ALTITUDE_RANGE    = (0, 30_000)
+
+# Enemy aircraft altitude (metres) — same envelope as your own aircraft
+# Determines vertical geometry of the engagement alongside elevation angle
+ENEMY_ALTITUDE_RANGE   = (0, 30_000)
+
+# Azimuth: horizontal angle of incoming threat, clockwise from North
+# 0° = head-on, 90° = right side, 180° = tail-chase, 270° = left side
+AZIMUTH_RANGE          = (0, 360)
+
+# Elevation: vertical angle of incoming threat
+# 0° = same altitude, +90° = directly above, -90° = directly below
+ELEVATION_RANGE        = (-90, 90)
+
+# Maneuverability: 0 = low (bomber/transport), 1 = medium (older jets),
+#                  2 = high (modern fighters)
+MANEUVERABILITY_VALUES = [0, 1, 2]
+
+# Countermeasure: 0 = not deployed, 1 = deployed (flares/chaff)
+COUNTERMEASURE_VALUES  = [0, 1]
+
+
+# ── Runs once — loads combat-capable aircraft from metadata CSV ───────────────
+def _load_metadata():
+    """
+    Loads combat-capable aircraft from the metadata CSV (no_aa_capability == 0).
+
+    Returns:
+        pd.DataFrame: Filtered metadata containing only aircraft with air-to-air missile capability.
+    """
+    
+    if not METADATA_PATH.exists():
+        raise FileNotFoundError(f"Metadata CSV not found: {METADATA_PATH}")
+    metadata_df = pd.read_csv(METADATA_PATH)
+    metadata_df = metadata_df[metadata_df["no_aa_capability"]==0]
+    return metadata_df
+
+
+# ── Derives missile phase from how far it has already traveled ────────────────
+# phase 0 = boost (just launched), 1 = mid-course, 2 = terminal (final approach)
+def _derive_missile_phase(remaining_distance, launch_distance):
+    """
+    Derives missile flight phase from how much of its journey it has completed.
+
+    Args:
+        remaining_distance (float): Distance remaining between missile and target (metres).
+        launch_distance (float): Total distance at the moment of launch (metres).
+
+    Returns:
+        int: 0 = boost, 1 = mid-course, 2 = terminal.
+        
+        Phase 0 — engine burning, accelerating, just launched
+        Phase 1 — flying toward your predicted position, guided but not actively tracking you yet
+        Phase 2 — active seeker on, tracking you specifically, hardest to fool
+    """
+    
+    traveled_distance = launch_distance - remaining_distance
+    ratio = traveled_distance / launch_distance
+    
+    if ratio < 0.33:
+        return 0
+    elif 0.33 <= ratio < 0.66:
+        return 1
+    else:
+        return 2
+
+
+# ── Derives closure rate using 3D geometry (azimuth + elevation + speeds) ─────
+# How fast the gap between you and the missile is closing, in m/s
+# Enemy altitude adds the vertical dimension to the engagement geometry
+def _derive_closure_rate(missile_speed, your_speed,
+                         azimuth, elevation):
+    """
+    Derives the closure rate - how fast the gap between the missile and the target is closing (m/s).
+
+    Uses 3D geometry: azimuth accounts for horizontal approach angle,
+    elevation accounts for vertical approach angle.
+
+    Args:
+        missile_speed (float): Speed of the incoming missile (m/s).
+        your_speed (float): Speed of the friendly aircraft (m/s).
+        azimuth (float): Horizontal angle of incoming threat in degrees (0° = head-on).
+        elevation (float): Vertical angle of incoming threat in degrees (0° = same altitude).
+
+    Returns:
+        float: Closure rate in m/s.
+    """
+    
+    # Convert the angles into radians and prepare for cosine
+    azimuth = math.radians(azimuth)
+    elevation = math.radians(elevation)
+    
+    # Extract the closure rate
+    closure_rate = missile_speed + (your_speed * (math.cos(azimuth) * math.cos(elevation)))
+    
+    return closure_rate
+
+
+"""
+Note: enemy_speed is not included as a feature. This is a known simplification, in real engagements,
+enemy aircraft velocity at launch contributes to effective missile speed. This can be added in a future iteration 
+of the physics generator.
+"""
+
+
+def _derive_evasion_time(remaining_distance, closure_rate,
+                         missile_phase, enemy_generation,
+                         your_speed, your_altitude, enemy_altitude):
+    """
+    Derives the minimum evasion time - seconds before the missile reaches you.
+
+    Base calculation is pure kinematics: remaining_distance / closure_rate.
+    Four modifiers are applied to account for factors the base formula cannot capture.
+
+    Modifiers:
+        - missile_phase == 2 (terminal): seeker has locked on, countermeasures
+          need 2-3s overhead to be effective. Shrinks window by 15%. (x 0.85)
+        - enemy_generation == 5: HOBS (High Off-Boresight)
+          seeker + ECCM make the missile harder to
+          defeat, compressing effective reaction time by ~10%. (x 0.90)
+        - your_speed > 522 m/s (above median): high energy state gives more
+          lateral geometry per second during evasion. Slight expansion. (x 1.05)
+        - abs(your_altitude - enemy_altitude) > 5000m: large altitude gap pushes
+          engagement toward edge of missile performance envelope, degrading
+          terminal accuracy. Slight expansion. (x 1.10)
+
+    Args:
+        remaining_distance (float): Distance between missile and you right now (metres).
+        closure_rate (float): Combined closing speed from _derive_closure_rate() (m/s).
+        missile_phase (int): 0 = boost, 1 = mid-course, 2 = terminal.
+        enemy_generation (float): Enemy aircraft generation (3.5, 4, 4.5, or 5).
+        your_speed (float): Your current airspeed (m/s).
+        your_altitude (float): Your current altitude (metres).
+        enemy_altitude (float): Enemy aircraft altitude (metres).
+
+    Returns:
+        float: Evasion time in seconds.
+    """
+    
+    # Calculate the evasion time
+    evasion_time = remaining_distance / closure_rate
+    
+    # Terminal phase - seeker locked on, countermeasures need 2–3s overhead
+    if missile_phase == 2:
+        evasion_time *= 0.85
+        
+    # Gen 5 aircraft missile (HOBS + ECCM) compresses effective reaction time
+    if enemy_generation == 5:
+        evasion_time *= 0.90
+        
+    # High speed — more room to maneuver
+    if your_speed > 522:
+        evasion_time *= 1.05
+        
+    # Large altitude gap - missile at edge of performance envelope
+    if abs(your_altitude - enemy_altitude) > 5000:
+        evasion_time *= 1.10
+        
+    return evasion_time
+    
+    
+
+def _derive_hit_label(countermeasure_deployed, your_maneuverability,
+                      azimuth, elevation, missile_phase, enemy_generation,
+                      your_altitude, enemy_altitude):
+    """
+    Derives whether the missile hits after an evasion attempt.
+
+    Starts from a neutral survival score of 0.5. Modifiers push it up (more
+    likely to hit) or down (more likely to miss). Final label is 1 if the
+    missile hits, 0 if it misses.
+
+    This is where azimuth and elevation belong — they affect survival after
+    evasion, not when the missile arrives (that is handled by closure_rate
+    in _derive_evasion_time).
+
+    Modifiers:
+        - countermeasure_deployed == 1: active countermeasures significantly
+          reduce hit probability. Score drops.
+        - your_maneuverability == 2: high maneuverability makes evasion more
+          effective. Score drops.
+        - azimuth near 0° (head-on): least time and geometry to evade.
+          Score rises.
+        - missile_phase == 2 (terminal): seeker locked on, hardest to defeat.
+          Score rises.
+        - enemy_generation == 5: HOBS + ECCM make the missile more lethal.
+          Score rises.
+        - elevation far from 0°: steep approach angle reduces evasion options.
+          Score rises.
+        - large altitude differential > 5000m: pushes missile toward edge of
+          performance envelope. Score drops.
+
+    Args:
+        countermeasure_deployed (int): 0 = not deployed, 1 = deployed.
+        your_maneuverability (int): 0 = low, 1 = medium, 2 = high.
+        azimuth (float): Horizontal angle of incoming threat in degrees (0° = head-on).
+        elevation (float): Vertical angle of incoming threat in degrees (0° = level).
+        missile_phase (int): 0 = boost, 1 = mid-course, 2 = terminal.
+        enemy_generation (float): Enemy aircraft generation (3.5, 4, 4.5, or 5).
+        your_altitude (float): Your current altitude (metres).
+        enemy_altitude (float): Enemy aircraft altitude (metres).
+
+    Returns:
+        int: 1 if missile hits, 0 if missile misses.
+    """
+    
+    # Neutral starting score - modifiers push it toward hit or miss
+    score = 0.5
+    
+    # Countermeasures active - significantly reduces hit chance
+    if countermeasure_deployed == 1:
+        score -= 0.20
+        
+    # High maneuverability - evasion more effective
+    if your_maneuverability == 2:
+        score -= 0.10
+        
+    # Head-on approach - least time and geometry to evade
+    if azimuth < 30:
+        score += 0.15
+        
+    # Terminal phase - seeker locked on, hardest to defeat
+    if missile_phase == 2:
+        score += 0.15
+        
+    # Gen 5 missile - HOBS + ECCM make it more lethal
+    if enemy_generation == 5:
+        score += 0.10
+        
+    # Steep approach angle - reduces evasion options
+    if abs(elevation) > 45:
+        score += 0.10
+        
+    # Large altitude gap - missile at edge of performance envelope
+    if abs(your_altitude - enemy_altitude) > 5000:
+        score -= 0.10
+
+    # Hit if score crosses 0.5
+    return 1 if score >= 0.5 else 0
+    
+
+# ── Generates one complete engagement scenario as a dict ─────────────────────
+def _generate_row(aircraft_row):
+    """
+    Generates one synthetic engagement scenario for a given aircraft.
+
+    Samples random values for all situational features, derives computed
+    features using physics, and returns a complete row with labels.
+
+    Args:
+        aircraft_row (pd.Series): One row from the metadata CSV.
+
+    Returns:
+        dict: All 14 features plus evasion_time and hit labels.
+    """
+    
+    # Sample situational features
+    your_speed    = np.random.uniform(*YOUR_SPEED_RANGE)
+    your_altitude = np.random.uniform(*YOUR_ALTITUDE_RANGE)
+    enemy_altitude = np.random.uniform(*ENEMY_ALTITUDE_RANGE)
+    azimuth       = np.random.uniform(*AZIMUTH_RANGE)
+    elevation     = np.random.uniform(*ELEVATION_RANGE)
+    
+    # Sample categorical features
+    your_maneuverability    = np.random.choice(MANEUVERABILITY_VALUES)
+    countermeasure_deployed = np.random.choice(COUNTERMEASURE_VALUES)
+    
+    # Pull threat specs from metadata
+    missile_speed    = aircraft_row["missile_speed"]
+    missile_range    = aircraft_row["missile_range"]
+    enemy_generation = aircraft_row["enemy_generation"]
+    
+    # Sample engagement distances
+    launch_distance    = np.random.uniform(LAUNCH_DISTANCE_MIN, missile_range)
+    remaining_distance = np.random.uniform(0, launch_distance)    # remaining_distance <= launch_distance always
+    
+    # Derive computed features
+    missile_phase = _derive_missile_phase(remaining_distance, launch_distance)
+    closure_rate  = _derive_closure_rate(missile_speed, your_speed,
+                                         azimuth, elevation)
+
+    # Derive labels
+    evasion_time = _derive_evasion_time(remaining_distance, closure_rate,
+                                        missile_phase, enemy_generation,
+                                        your_speed, your_altitude, enemy_altitude)
+    hit          = _derive_hit_label(countermeasure_deployed, your_maneuverability,
+                                     azimuth, elevation, missile_phase, enemy_generation,
+                                     your_altitude, enemy_altitude)
+    
+    # Return complete row
+    return {
+        "launch_distance":        launch_distance,
+        "remaining_distance":     remaining_distance,
+        "closure_rate":           closure_rate,
+        "azimuth":                azimuth,
+        "elevation":              elevation,
+        "missile_phase":          missile_phase,
+        "your_speed":             your_speed,
+        "your_altitude":          your_altitude,
+        "your_maneuverability":   your_maneuverability,
+        "enemy_altitude":         enemy_altitude,
+        "missile_speed":          missile_speed,
+        "missile_range":          missile_range,
+        "enemy_generation":       enemy_generation,
+        "countermeasure_deployed": countermeasure_deployed,
+        "evasion_time":           evasion_time,
+        "hit":                    hit
+    }
+    
+
+
+# ── Saves completed DataFrame to CSV ─────────────────────────────────────────
+def _save_dataset(df):
+    """
+    Saves the generated dataset to a CSV file.
+
+    Args:
+        df (pd.DataFrame): The complete synthetic engagement dataset.
+    """
+    
+    # Create data directory if it doesn't exist
+    OUTPUT_PATH.parent.mkdir(parents=True, exist_ok=True)
+    
+    # Save to CSV
+    df.to_csv(OUTPUT_PATH, index=False)
+
+
+# ── Public entry point — the only function the notebook calls ─────────────────
+def generate_dataset():
+    """
+    Generates 1,000,000 synthetic engagement scenarios and saves to CSV.
+
+    Loads combat-capable aircraft from metadata, samples each aircraft
+    equally across all rows, derives all features and labels using physics,
+    and writes the final dataset to disk.
+
+    Only valid threat scenarios are included (closure_rate > 0).
+
+    Returns:
+        pd.DataFrame: The complete synthetic engagement dataset.
+    """
+    metadata = _load_metadata()
+    
+    rows_per_aircraft = N_ROWS // len(metadata)  # 1_000_000 // 56 → 17,857 rows per aircraft
+    rows = []
+    
+    for _, aircraft_row in metadata.iterrows():
+        count = 0
+        while count < rows_per_aircraft:
+            row = _generate_row(aircraft_row)
+            if row["closure_rate"] > 0:
+                rows.append(row)
+                count += 1
+    
+    # Fill remaining rows to hit exactly N_ROWS
+    remaining_rows = N_ROWS - len(rows)
+    if remaining_rows:
+        count = 0
+        while count < remaining_rows:
+            row = _generate_row(metadata.sample(1).iloc[0])
+            if row["closure_rate"] > 0:
+                rows.append(row)
+                count += 1
+    
+    df = pd.DataFrame(rows)
+    _save_dataset(df)
+    print(f"Done. {len(rows):,} rows saved to {OUTPUT_PATH}")
+    
+    
+    
+# Inference only - assembles 14-feature array from HUD inputs and metadata for model prediction    
+def build_feature_array(
+    # From HUD sliders
+    launch_distance, remaining_distance,
+    azimuth, elevation,
+    your_speed, your_altitude, enemy_altitude,
+    countermeasure_deployed,
+    # From metadata lookup (enemy aircraft)
+    missile_speed, missile_range, enemy_generation,
+    # From metadata lookup (friendly aircraft)
+    your_maneuverability
+):
+    """
+    Assembles the 14-feature array required by the ETA and hit models at inference time.
+
+    All inputs arrive pre-collected from the HUD sliders and metadata lookups.
+    Closure rate and missile phase are derived here from those inputs using the
+    same physics helpers used during training. The returned dict is in the exact
+    column order the models were trained on.
+
+    Args:
+        launch_distance (float): Distance at moment of missile launch (metres).
+        remaining_distance (float): Distance remaining between missile and target (metres).
+        azimuth (float): Horizontal angle of incoming threat in degrees (0 = head-on).
+        elevation (float): Vertical angle of incoming threat in degrees (0 = level).
+        your_speed (float): Friendly aircraft airspeed (m/s).
+        your_altitude (float): Friendly aircraft altitude (metres).
+        enemy_altitude (float): Enemy aircraft altitude (metres).
+        countermeasure_deployed (int): 0 = not deployed, 1 = deployed.
+        missile_speed (float): Incoming missile speed (m/s), from enemy metadata.
+        missile_range (float): Missile maximum effective range (metres), from enemy metadata.
+        enemy_generation (float): Enemy aircraft generation, from enemy metadata.
+        your_maneuverability (int): Friendly aircraft maneuverability, from friendly metadata.
+            0 = low, 1 = medium, 2 = high. Maneuverability is an aircraft property
+            and follows the same metadata lookup pattern for both friendly and enemy platforms.
+
+    Returns:
+        dict: 14 features in training column order, ready for model inference.
+    """
+
+    # Derive the two features that are computed from inputs rather than sourced directly
+    missile_phase = _derive_missile_phase(remaining_distance, launch_distance)
+    closure_rate  = _derive_closure_rate(missile_speed, your_speed,
+                                         azimuth, elevation)
+
+    # Assemble and return all 14 features in exact training column order
+    return {
+        "launch_distance":         launch_distance,
+        "remaining_distance":      remaining_distance,
+        "closure_rate":            closure_rate,
+        "azimuth":                 azimuth,
+        "elevation":               elevation,
+        "missile_phase":           missile_phase,
+        "your_speed":              your_speed,
+        "your_altitude":           your_altitude,
+        "your_maneuverability":    your_maneuverability,
+        "enemy_altitude":          enemy_altitude,
+        "missile_speed":           missile_speed,
+        "missile_range":           missile_range,
+        "enemy_generation":        enemy_generation,
+        "countermeasure_deployed": countermeasure_deployed
+    }

src/schemas.py

@@ -0,0 +1,49 @@
+"""
+What is schemas.py
+
+A single Python file that holds shared constants — column names, their order, and any fixed values the pipeline depends on.
+
+Why it exists
+
+Three files will each build or consume the 14-feature array: physics_generator.py, models.py, and main.py. If each one hardcodes its own column list, one typo in one file breaks the whole pipeline silently.
+schemas.py means they all import from the same list. One source of truth.
+
+What goes inside it
+
+Three things:
+FEATURE_COLUMNS - the 14 feature names in exact training order
+MODEL_PATHS - file paths to your two saved .joblib models
+DECISION_THRESHOLDS - the ETA and hit probability cutoffs your decision layer will use
+"""
+
+
+from pathlib import Path
+
+# Path pointing towards project ATAS main folder
+BASE_DIR = Path(__file__).parent.parent
+# Path pointing towards saved models
+MODEL_DIR = BASE_DIR / "release_models"
+
+# Feature columns to be feed to models
+FEATURE_COLUMNS = [
+    'launch_distance', 'remaining_distance', 'closure_rate', 'azimuth', 'elevation',
+    'missile_phase', 'your_speed', 'your_altitude', 'your_maneuverability',
+    'enemy_altitude', 'missile_speed', 'missile_range', 'enemy_generation',
+    'countermeasure_deployed'
+]
+
+# Model paths as dict to use them to make predictions
+MODEL_PATHS = {
+    'classifier' : MODEL_DIR / "aircraft_classifier" / "atas_final_fine_tuned_aircraft_classifier_model.keras",
+    'eta'        : MODEL_DIR / "eta" / "atas_final_eta_regressor_model.joblib",
+    'hit'        : MODEL_DIR / "hit" / "atas_final_hit_classifier_model.joblib"
+}
+
+# Cutoffs used by decision.py to convert model outputs into pilot recommendations
+DECISION_THRESHOLDS = {
+    'hit_prob_high': 0.75,
+    'hit_prob_medium': 0.60,
+    'eta_critical': 10,
+    'gen_advantage_threshold': 0.5,    # friendly gen must be this much higher to be considered advantaged
+    'gen_disadvantage_threshold': 0.5  # enemy gen must be this much higher to be considered disadvantaged
+}