Add flight amenities: legroom, WiFi, power/USB, video per segment

Generate amenities deterministically based on aircraft type, cabin class,
and flight distance. Shown in Google Flights-style right-side panel in
the expanded flight card view, with icons and strikethrough for unavailable.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

backend/config.py

@@ -118,6 +118,51 @@ BEST_WEIGHT_PRICE = 0.45
 BEST_WEIGHT_DURATION = 0.35
 BEST_WEIGHT_STOPS = 0.20
 
+# Amenity generation rules
+# Legroom ranges (inches) by cabin class
+LEGROOM_RANGES = {
+    "economy": (29, 32),
+    "premium_economy": (34, 38),
+    "business": (38, 78),
+    "first": (78, 86),
+}
+
+# WiFi probability by aircraft family (widebody long-haul more likely)
+WIFI_AIRCRAFT = {
+    "787-8": 0.95, "787-9": 0.95, "787-10": 0.95,
+    "A350-900": 0.92, "A350-1000": 0.92,
+    "A330-300": 0.70, "777-300ER": 0.85,
+    "A380": 0.80,
+    "A321neo LR": 0.80, "757-200": 0.50, "767-300ER": 0.60,
+    "A320": 0.55, "A321": 0.60, "737-800": 0.50, "737 MAX 8": 0.75,
+    "E190": 0.25, "E175": 0.20, "CRJ-900": 0.15,
+}
+
+# Power/USB probability by aircraft family
+POWER_AIRCRAFT = {
+    "787-8": 0.95, "787-9": 0.95, "787-10": 0.95,
+    "A350-900": 0.95, "A350-1000": 0.95,
+    "A330-300": 0.75, "777-300ER": 0.85,
+    "A380": 0.90,
+    "A321neo LR": 0.80, "757-200": 0.40, "767-300ER": 0.55,
+    "A320": 0.45, "A321": 0.50, "737-800": 0.40, "737 MAX 8": 0.70,
+    "E190": 0.15, "E175": 0.10, "CRJ-900": 0.05,
+}
+
+# Video probability (seatback IFE) — mostly widebody long-haul
+VIDEO_AIRCRAFT = {
+    "787-8": 0.90, "787-9": 0.92, "787-10": 0.92,
+    "A350-900": 0.93, "A350-1000": 0.93,
+    "A330-300": 0.80, "777-300ER": 0.90,
+    "A380": 0.95,
+    "A321neo LR": 0.40, "757-200": 0.25, "767-300ER": 0.60,
+    "A320": 0.10, "A321": 0.12, "737-800": 0.08, "737 MAX 8": 0.15,
+    "E190": 0.0, "E175": 0.0, "CRJ-900": 0.0,
+}
+
+# Business/first class always get power and higher WiFi/video odds
+PREMIUM_CLASS_AMENITY_BOOST = 0.30  # +30% probability for business/first
+
 # Search limits
 MAX_RESULTS = 200
 MAX_AUTOCOMPLETE_RESULTS = 10

backend/flight_generator.py

@@ -13,12 +13,17 @@ from .config import (
     DEPARTURE_HOUR_MAX,
     DEPARTURE_HOUR_MIN,
     EMISSIONS_KG_PER_KM,
+    LEGROOM_RANGES,
     MAX_FLIGHTS_MULTI_CARRIER,
     MAX_FLIGHTS_SINGLE_CARRIER,
     MAX_LAYOVER_MINUTES,
     MIN_FLIGHTS_PER_DAY,
     MIN_LAYOVER_MINUTES,
+    POWER_AIRCRAFT,
+    PREMIUM_CLASS_AMENITY_BOOST,
     SAME_AIRLINE_CONNECTION_DISCOUNT,
+    VIDEO_AIRCRAFT,
+    WIFI_AIRCRAFT,
 )
 from .data_loader import Route, RouteGraph
 from .models import CabinClass, FlightOffer, FlightSegment
@@ -38,6 +43,58 @@ def _make_flight_number(carrier_iata: str, rng: random.Random) -> str:
     return f"{carrier_iata}{rng.randint(100, 9999)}"
 
 
+def _generate_amenities(
+    aircraft: str, cabin_class: CabinClass, distance_km: int, rng: random.Random,
+) -> dict:
+    """Generate seat amenities based on aircraft, cabin class, and distance."""
+    cls = cabin_class.value
+    lo, hi = LEGROOM_RANGES.get(cls, (29, 32))
+    legroom = rng.randint(lo, hi)
+
+    is_premium = cls in ("business", "first")
+    boost = PREMIUM_CLASS_AMENITY_BOOST if is_premium else 0.0
+
+    # WiFi
+    wifi_prob = min(1.0, WIFI_AIRCRAFT.get(aircraft, 0.3) + boost)
+    has_wifi = rng.random() < wifi_prob
+    wifi_type = None
+    if has_wifi:
+        # Free WiFi more common on premium classes and newer aircraft
+        free_prob = 0.7 if is_premium else (0.3 if aircraft.startswith(("787", "A35")) else 0.12)
+        wifi_type = "Free Wi-Fi" if rng.random() < free_prob else "Wi-Fi for a fee"
+
+    # Power & USB
+    power_prob = min(1.0, POWER_AIRCRAFT.get(aircraft, 0.2) + boost)
+    has_power = rng.random() < power_prob
+    # USB slightly more common than AC power
+    has_usb = has_power or rng.random() < min(1.0, power_prob + 0.15)
+
+    # Video/IFE
+    video_prob = min(1.0, VIDEO_AIRCRAFT.get(aircraft, 0.05) + boost)
+    # Long-haul (>4000km) gets a boost
+    if distance_km > 4000:
+        video_prob = min(1.0, video_prob + 0.15)
+    has_video = rng.random() < video_prob
+    video_type = None
+    if has_video:
+        if aircraft in ("A380", "777-300ER", "787-9", "787-10", "A350-900", "A350-1000"):
+            video_type = "On-demand video"
+        elif distance_km > 3000:
+            video_type = "On-demand video" if rng.random() < 0.7 else "Seatback screen"
+        else:
+            video_type = "Live TV" if rng.random() < 0.4 else "Seatback screen"
+
+    return {
+        "legroom_inches": legroom,
+        "has_wifi": has_wifi,
+        "wifi_type": wifi_type,
+        "has_power": has_power,
+        "has_usb": has_usb,
+        "has_video": has_video,
+        "video_type": video_type,
+    }
+
+
 def _get_timezone(graph: RouteGraph, iata: str) -> ZoneInfo:
     airport = graph.airports.get(iata)
     if airport and airport.timezone:
@@ -129,11 +186,14 @@ def _generate_direct_flights(
 
         flight_id = f"{origin}{destination}{departure_date.isoformat()}{dep_minutes}{carrier['iata']}"
 
+        aircraft = _pick_aircraft(leg.distance_km, rng)
+        amenities = _generate_amenities(aircraft, cabin_class, leg.distance_km, rng)
+
         segment = FlightSegment(
             airline_code=carrier["iata"],
             airline_name=carrier["name"],
             flight_number=_make_flight_number(carrier["iata"], rng),
-            aircraft=_pick_aircraft(leg.distance_km, rng),
+            aircraft=aircraft,
             origin=origin,
             origin_city=origin_airport.city_name,
             destination=destination,
@@ -141,6 +201,7 @@ def _generate_direct_flights(
             departure=departure_dt,
             arrival=arrival_dt,
             duration_minutes=leg.duration_min,
+            **amenities,
         )
 
         emissions = int(leg.distance_km * EMISSIONS_KG_PER_KM.get(cabin_class.value, 0.09))
@@ -221,11 +282,14 @@ def _generate_connecting_flights(
             leg_price *= CONNECTING_BASE_DISCOUNT
             total_price += leg_price
 
+            aircraft = _pick_aircraft(leg.distance_km, rng)
+            amenities = _generate_amenities(aircraft, cabin_class, leg.distance_km, rng)
+
             segments.append(FlightSegment(
                 airline_code=carrier["iata"],
                 airline_name=carrier["name"],
                 flight_number=_make_flight_number(carrier["iata"], rng),
-                aircraft=_pick_aircraft(leg.distance_km, rng),
+                aircraft=aircraft,
                 origin=leg_origin,
                 origin_city=origin_ap.city_name,
                 destination=leg_dest,
@@ -233,6 +297,7 @@ def _generate_connecting_flights(
                 departure=departure_dt,
                 arrival=arrival_dt,
                 duration_minutes=leg.duration_min,
+                **amenities,
             ))
 
             # Add layover time for next leg

backend/models.py

@@ -58,6 +58,14 @@ class FlightSegment(BaseModel):
     departure: datetime
     arrival: datetime
     duration_minutes: int
+    # Amenities
+    legroom_inches: int = 31
+    has_wifi: bool = False
+    wifi_type: Optional[str] = None       # "Free Wi-Fi", "Wi-Fi for a fee"
+    has_power: bool = False
+    has_usb: bool = False
+    has_video: bool = False
+    video_type: Optional[str] = None      # "On-demand video", "Live TV", "Seatback screen"
 
 
 # --- Flight offer (may have multiple segments) ---

frontend/src/api/types.ts

@@ -23,6 +23,14 @@ export interface FlightSegment {
   departure: string;
   arrival: string;
   duration_minutes: number;
+  // Amenities
+  legroom_inches: number;
+  has_wifi: boolean;
+  wifi_type: string | null;
+  has_power: boolean;
+  has_usb: boolean;
+  has_video: boolean;
+  video_type: string | null;
 }
 
 export interface FlightOffer {

frontend/src/components/results/FlightCard.tsx

@@ -95,35 +95,113 @@ export default function FlightCard({ flight }: Props) {
         <div className="border-t border-gray-100 px-4 pt-3 pb-4" data-testid="segments-detail">
           {flight.segments.map((seg, i) => (
             <div key={i}>
-              {/* Segment */}
-              <div className="flex gap-3">
-                {/* Timeline column */}
-                <div className="flex flex-col items-center w-5 flex-shrink-0">
-                  <div className="h-2.5 w-2.5 rounded-full border-2 border-[#1a73e8] bg-white" />
-                  <div className="flex-1 w-px bg-gray-300 my-0.5" />
-                  <div className="h-2.5 w-2.5 rounded-full border-2 border-[#1a73e8] bg-white" />
+              {/* Segment: timeline+info on left, amenities on right */}
+              <div className="flex gap-4">
+                {/* Left side: timeline + segment content */}
+                <div className="flex gap-3 flex-1 min-w-0">
+                  {/* Timeline column */}
+                  <div className="flex flex-col items-center w-5 flex-shrink-0">
+                    <div className="h-2.5 w-2.5 rounded-full border-2 border-[#1a73e8] bg-white" />
+                    <div className="flex-1 w-px bg-gray-300 my-0.5" />
+                    <div className="h-2.5 w-2.5 rounded-full border-2 border-[#1a73e8] bg-white" />
+                  </div>
+
+                  {/* Segment content */}
+                  <div className="flex-1 pb-1 min-w-0">
+                    {/* Departure */}
+                    <div className="flex items-baseline gap-2">
+                      <span className="text-sm font-medium w-16 flex-shrink-0">{formatTime(seg.departure)}</span>
+                      <span className="text-sm text-gray-500">·</span>
+                      <span className="text-sm text-gray-900">{seg.origin_city} ({seg.origin})</span>
+                    </div>
+
+                    {/* Flight info line */}
+                    <div className="ml-[72px] my-1.5 text-xs text-gray-400 leading-relaxed">
+                      <div>Travel time: {formatDuration(seg.duration_minutes)}</div>
+                      <div>{seg.airline_name} · {seg.flight_number} · {seg.aircraft}</div>
+                    </div>
+
+                    {/* Arrival */}
+                    <div className="flex items-baseline gap-2">
+                      <span className="text-sm font-medium w-16 flex-shrink-0">{formatTime(seg.arrival)}</span>
+                      <span className="text-sm text-gray-500">·</span>
+                      <span className="text-sm text-gray-900">{seg.destination_city} ({seg.destination})</span>
+                    </div>
+                  </div>
                 </div>
 
-                {/* Segment content */}
-                <div className="flex-1 pb-1">
-                  {/* Departure */}
-                  <div className="flex items-baseline gap-2">
-                    <span className="text-sm font-medium w-16">{formatTime(seg.departure)}</span>
-                    <span className="text-sm text-gray-500">·</span>
-                    <span className="text-sm text-gray-900">{seg.origin_city} ({seg.origin})</span>
+                {/* Right side: amenities */}
+                <div className="flex-shrink-0 w-44 border-l border-gray-100 pl-4 text-xs text-gray-500 space-y-1.5 pt-0.5" data-testid="segment-amenities">
+                  {/* Legroom */}
+                  <div className="flex items-center gap-2">
+                    <svg className="h-4 w-4 text-gray-400 flex-shrink-0" viewBox="0 0 24 24" fill="none" stroke="currentColor" strokeWidth="1.5">
+                      <path d="M6 19v-4a2 2 0 012-2h1V7a2 2 0 012-2h2a2 2 0 012 2v6h1a2 2 0 012 2v4" strokeLinecap="round" strokeLinejoin="round"/>
+                      <path d="M6 19h12" strokeLinecap="round"/>
+                    </svg>
+                    <span>{seg.legroom_inches} in legroom</span>
+                  </div>
+
+                  {/* WiFi */}
+                  <div className="flex items-center gap-2">
+                    {seg.has_wifi ? (
+                      <>
+                        <svg className="h-4 w-4 text-gray-400 flex-shrink-0" viewBox="0 0 24 24" fill="currentColor">
+                          <path d="M1 9l2 2c4.97-4.97 13.03-4.97 18 0l2-2C16.93 2.93 7.08 2.93 1 9zm8 8l3 3 3-3c-1.65-1.66-4.34-1.66-6 0zm-4-4l2 2c2.76-2.76 7.24-2.76 10 0l2-2C15.14 9.14 8.87 9.14 5 13z"/>
+                        </svg>
+                        <span>{seg.wifi_type || 'Wi-Fi'}</span>
+                      </>
+                    ) : (
+                      <>
+                        <svg className="h-4 w-4 text-gray-300 flex-shrink-0" viewBox="0 0 24 24" fill="currentColor">
+                          <path d="M1 9l2 2c4.97-4.97 13.03-4.97 18 0l2-2C16.93 2.93 7.08 2.93 1 9zm8 8l3 3 3-3c-1.65-1.66-4.34-1.66-6 0zm-4-4l2 2c2.76-2.76 7.24-2.76 10 0l2-2C15.14 9.14 8.87 9.14 5 13z"/>
+                        </svg>
+                        <span className="text-gray-300 line-through">Wi-Fi</span>
+                      </>
+                    )}
                   </div>
 
-                  {/* Flight info line */}
-                  <div className="ml-[72px] my-1.5 text-xs text-gray-400 leading-relaxed">
-                    <div>Travel time: {formatDuration(seg.duration_minutes)}</div>
-                    <div>{seg.airline_name} · {seg.flight_number} · {seg.aircraft}</div>
+                  {/* Power & USB */}
+                  <div className="flex items-center gap-2">
+                    {(seg.has_power || seg.has_usb) ? (
+                      <>
+                        <svg className="h-4 w-4 text-gray-400 flex-shrink-0" viewBox="0 0 24 24" fill="currentColor">
+                          <path d="M7 2v11h3v9l7-12h-4l4-8z"/>
+                        </svg>
+                        <span>
+                          {seg.has_power && seg.has_usb
+                            ? 'Power & USB'
+                            : seg.has_power
+                              ? 'In-seat power'
+                              : 'USB charging'}
+                        </span>
+                      </>
+                    ) : (
+                      <>
+                        <svg className="h-4 w-4 text-gray-300 flex-shrink-0" viewBox="0 0 24 24" fill="currentColor">
+                          <path d="M7 2v11h3v9l7-12h-4l4-8z"/>
+                        </svg>
+                        <span className="text-gray-300 line-through">Power</span>
+                      </>
+                    )}
                   </div>
 
-                  {/* Arrival */}
-                  <div className="flex items-baseline gap-2">
-                    <span className="text-sm font-medium w-16">{formatTime(seg.arrival)}</span>
-                    <span className="text-sm text-gray-500">·</span>
-                    <span className="text-sm text-gray-900">{seg.destination_city} ({seg.destination})</span>
+                  {/* Video/Entertainment */}
+                  <div className="flex items-center gap-2">
+                    {seg.has_video ? (
+                      <>
+                        <svg className="h-4 w-4 text-gray-400 flex-shrink-0" viewBox="0 0 24 24" fill="currentColor">
+                          <path d="M21 3H3c-1.1 0-2 .9-2 2v12c0 1.1.9 2 2 2h5v2h8v-2h5c1.1 0 2-.9 2-2V5c0-1.1-.9-2-2-2zm0 14H3V5h18v12z"/>
+                        </svg>
+                        <span>{seg.video_type || 'On-demand video'}</span>
+                      </>
+                    ) : (
+                      <>
+                        <svg className="h-4 w-4 text-gray-300 flex-shrink-0" viewBox="0 0 24 24" fill="currentColor">
+                          <path d="M21 3H3c-1.1 0-2 .9-2 2v12c0 1.1.9 2 2 2h5v2h8v-2h5c1.1 0 2-.9 2-2V5c0-1.1-.9-2-2-2zm0 14H3V5h18v12z"/>
+                        </svg>
+                        <span className="text-gray-300 line-through">Entertainment</span>
+                      </>
+                    )}
                   </div>
                 </div>
               </div>
@@ -143,13 +221,16 @@ export default function FlightCard({ flight }: Props) {
             </div>
           ))}
 
-          {/* Bottom info: emissions + total travel time */}
+          {/* Bottom info: emissions + cabin class + total travel time */}
           <div className="mt-3 pt-3 border-t border-gray-100 flex items-center gap-4 text-xs text-gray-400">
             <span>Total travel time: {formatDuration(flight.total_duration_minutes)}</span>
             {flight.emissions_kg > 0 && (
               <>
                 <span>·</span>
-                <span>{flight.emissions_kg} kg CO₂ per passenger</span>
+                <svg className="h-3.5 w-3.5 inline text-green-500" viewBox="0 0 24 24" fill="currentColor">
+                  <path d="M17 8C8 10 5.9 16.17 3.82 21.34l1.89.66.95-2.3c.48.17.98.3 1.34.3C19 20 22 3 22 3c-1 2-8 2.25-13 3.25S2 11.5 2 13.5s1.75 3.75 1.75 3.75C7 8 17 8 17 8z"/>
+                </svg>
+                <span className="text-green-600">{flight.emissions_kg} kg CO₂</span>
               </>
             )}
           </div>