СТО-ға арналған Document AI + Gemini 2.5 Flash: техпаспорт пен сынған бөлшектің суреті бойынша 3 секундта автобөлшек таңдау

import express, { Request, Response } from 'express';
import { DocumentProcessorServiceClient } from '@google-cloud/documentai';
import { GoogleGenAI, Type } from '@google/genai';
import { getFirestore, FieldValue } from 'firebase-admin/firestore';

const app = express();
app.use(express.json({ limit: '20mb' }));

// Инициализация клиентов Google Cloud
const docAiClient = new DocumentProcessorServiceClient();
const ai = new GoogleGenAI({ apiKey: process.env.GEMINI_API_KEY });
const db = getFirestore();

// Конфигурация процессора Document AI для распознавания техпаспорта (СРТС РК)
const DOCAI_PROCESSOR_NAME = process.env.DOCAI_PROCESSOR_NAME || 
  'projects/your-gcp-project/locations/eu/processors/kz-vehicle-id-processor';

interface AutoPartsMatchResult {
  vin: string;
  vehicle: {
    make: string;
    model: string;
    year: number;
    engineVolume: string;
  };
  detectedPart: {
    category: string;
    subsystem: 'ENGINE' | 'SUSPENSION' | 'BRAKES' | 'COOLING' | 'TRANSMISSION';
    visualDamage: string;
    estimatedOemNumber?: string;
  };
  recommendedMatches: Array<{
    oemCode: string;
    brand: string;
    titleRu: string;
    tier: 'ORIGINAL' | 'PREMIUM_AFTERMARKET' | 'BUDGET';
    priceKztAvg: number;
    inStockAlmaty: boolean;
  }>;
  confidenceScore: number;
}

/**
 * 1. Распознавание VIN и параметров авто из фото техпаспорта через Document AI
 */
async function parseVehicleRegistrationCertificate(imageBase64: string) {
  const request = {
    name: DOCAI_PROCESSOR_NAME,
    rawDocument: {
      content: imageBase64,
      mimeType: 'image/jpeg',
    },
  };

  const [result] = await docAiClient.processDocument(request);
  const { document } = result;

  // Извлечение сущностей (VIN, Год, Объем) из обработанного документа
  let vin = '';
  let makeModel = '';
  let year = new Date().getFullYear();

  if (document?.entities) {
    for (const entity of document.entities) {
      if (entity.type === 'VIN_NUMBER') vin = entity.mentionText?.toUpperCase().replace(/[^A-HJ-NPR-Z0-9]/g, '') || '';
      if (entity.type === 'VEHICLE_MODEL') makeModel = entity.mentionText || '';
      if (entity.type === 'MANUFACTURE_YEAR') year = parseInt(entity.mentionText || '2015', 10);
    }
  }

  // Fallback: регулярное выражение поиска VIN (17 символов, исключая I, O, Q)
  if (!vin && document?.text) {
    const vinMatch = document.text.match(/[A-HJ-NPR-Z0-9]{17}/);
    if (vinMatch) vin = vinMatch[0];
  }

  return { vin, makeModel, year };
}

/**
 * 2. Анализ сломанной детали и подбор артикулов через Gemini 2.5 Flash Vision
 */
async function matchBrokenPartWithGemini(
  brokenPartBase64: string,
  vehicleInfo: { vin: string; makeModel: string; year: number }
): Promise<AutoPartsMatchResult> {
  const systemInstruction = `Сен — Алматы мен Астананың ең білікті СТО бас шебері әрі автобөлшектер каталогының (EPC, Microcat, ETKA) бас сарапшысысың.
Міндетің:
1. Көлік мәліметтерін (VIN: ${vehicleInfo.vin}, Модель: ${vehicleInfo.makeModel}, Жыл: ${vehicleInfo.year}) ескере отырып, фотодағы сынған немесе тозған бөлшекті (помпа, сайлентблок, суппорт, генератор релесі) 100% дәл анықтау.
2. Бөлшектің визуалды зақымын сипаттау және нақты OEM каталогының артикулын шығару.
3. Қазақстан нарығындағы (Car City, Phaeton, Барахолка) 3 баға деңгейін (Оригинал, Корея/Жапония дубликат, Эконом) ұсыну.`;

  const response = await ai.models.generateContent({
    model: 'gemini-2.5-flash',
    contents: [
      {
        role: 'user',
        parts: [
          {
            inlineData: {
              data: brokenPartBase64,
              mimeType: 'image/jpeg'
            }
          },
          {
            text: `Автомобиль: ${vehicleInfo.makeModel} (${vehicleInfo.year} ж.), VIN: ${vehicleInfo.vin}.
Анықта: бұл қандай сынған бөлшек, оның OEM коды қандай және қандай дубликаттар сәйкес келеді?`
          }
        ]
      }
    ],
    config: {
      systemInstruction,
      responseMimeType: 'application/json',
      responseSchema: {
        type: Type.OBJECT,
        properties: {
          detectedPart: {
            type: Type.OBJECT,
            properties: {
              category: { type: Type.STRING },
              subsystem: { type: Type.STRING, enum: ['ENGINE', 'SUSPENSION', 'BRAKES', 'COOLING', 'TRANSMISSION'] },
              visualDamage: { type: Type.STRING },
              estimatedOemNumber: { type: Type.STRING }
            },
            required: ['category', 'subsystem', 'visualDamage']
          },
          recommendedMatches: {
            type: Type.ARRAY,
            items: {
              type: Type.OBJECT,
              properties: {
                oemCode: { type: Type.STRING },
                brand: { type: Type.STRING },
                titleRu: { type: Type.STRING },
                tier: { type: Type.STRING, enum: ['ORIGINAL', 'PREMIUM_AFTERMARKET', 'BUDGET'] },
                priceKztAvg: { type: Type.NUMBER },
                inStockAlmaty: { type: Type.BOOLEAN }
              },
              required: ['oemCode', 'brand', 'titleRu', 'tier', 'priceKztAvg']
            }
          },
          confidenceScore: { type: Type.NUMBER }
        },
        required: ['detectedPart', 'recommendedMatches', 'confidenceScore']
      },
      temperature: 0.1
    }
  });

  const parsed = JSON.parse(response.text || '{}');
  return {
    vin: vehicleInfo.vin,
    vehicle: {
      make: vehicleInfo.makeModel.split(' ')[0] || 'Unknown',
      model: vehicleInfo.makeModel,
      year: vehicleInfo.year,
      engineVolume: '2.5L'
    },
    ...parsed
  };
}

/**
 * Главный эндпоинт для СТО: принимает 2 фото за один запрос
 */
app.post('/api/v1/sto/diagnose-and-match', async (req: Request, res: Response) => {
  const { techPassportBase64, brokenPartBase64, workshopId, mechanicName } = req.body;

  if (!techPassportBase64 || !brokenPartBase64) {
    return res.status(400).json({ error: 'Both techPassportBase64 and brokenPartBase64 are required' });
  }

  try {
    const t0 = Date.now();

    // Параллельное / последовательное выполнение Document AI и Gemini 2.5 Flash
    const vehicleInfo = await parseVehicleRegistrationCertificate(techPassportBase64);
    const matchResult = await matchBrokenPartWithGemini(brokenPartBase64, vehicleInfo);

    const totalLatencyMs = Date.now() - t0;

    // Логирование диагностики в Firestore
    const logRef = db.collection('sto_diagnostics').doc();
    await logRef.set({
      workshopId: workshopId || 'STO_SEIFULLINA_BOX_4',
      mechanic: mechanicName || 'Серик шебер',
      matchResult,
      totalLatencyMs,
      timestamp: FieldValue.serverTimestamp()
    });

    return res.status(200).json({
      status: 'success',
      diagnosticId: logRef.id,
      totalLatencyMs,
      data: matchResult
    });
  } catch (error: any) {
    console.error('[STO AI Matcher Error]:', error);
    return res.status(500).json({ error: 'MATCHING_PIPELINE_FAILED', message: error.message });
  }
});

const PORT = process.env.PORT || 3000;
app.listen(PORT, () => console.log(`STO AI Parts Matcher active on port ${PORT}`));

/**
 * Валидатор контрольной суммы VIN-номера по стандарту ISO 3779 / US DOT
 * Предотвращает ошибки оптического распознавания (путаница 0 и O, 1 и I)
 * Источник: OZAT Engineering Blog (https://ozat.kz)
 */
export function validateAndCleanVin(rawVin: string): { isValid: boolean; cleanedVin: string; error?: string } {
  // 1. Очистка от пробелов, дефисов и перевод в верхний регистр
  const cleanedVin = rawVin.trim().toUpperCase().replace(/[^A-Z0-9]/g, '');

  // 2. Длина VIN обязана быть ровно 17 символов (для автомобилей с 1981 года)
  if (cleanedVin.length !== 17) {
    return { isValid: false, cleanedVin, error: `Неверная длина VIN (${cleanedVin.length} вместо 17)` };
  }

  // 3. Запрещенные символы в VIN: I (И), O (О), Q (Кью) — во избежание путаницы с 1 и 0
  if (/[IOQ]/.test(cleanedVin)) {
    return { isValid: false, cleanedVin, error: 'VIN не может содержать буквы I, O, Q' };
  }

  // 4. Таблица весовых коэффициентов позиций (1..17)
  const positionWeights = [8, 7, 6, 5, 4, 3, 2, 10, 0, 9, 8, 7, 6, 5, 4, 3, 2];

  // 5. Таблица численных значений буквенных символов
  const charValues: Record<string, number> = {
    A: 1, B: 2, C: 3, D: 4, E: 5, F: 6, G: 7, H: 8,
    J: 1, K: 2, L: 3, M: 4, N: 5, P: 7, R: 9,
    S: 2, T: 3, U: 4, V: 5, W: 6, X: 7, Y: 8, Z: 9,
    '0': 0, '1': 1, '2': 2, '3': 3, '4': 4,
    '5': 5, '6': 6, '7': 7, '8': 8, '9': 9
  };

  let weightedSum = 0;
  for (let i = 0; i < 17; i++) {
    const char = cleanedVin[i];
    const val = charValues[char];
    if (val === undefined) {
      return { isValid: false, cleanedVin, error: `Недопустимый символ ${char}` };
    }
    weightedSum += val * positionWeights[i];
  }

  // 6. Расчет контрольного знака (остаток от деления на 11)
  const remainder = weightedSum % 11;
  const expectedCheckDigit = remainder === 10 ? 'X' : remainder.toString();
  const actualCheckDigit = cleanedVin[8]; // 9-й символ (индекс 8) — контрольный знак

  // Для рынков Северной Америки и большинства мировых брендов проверка строгая
  const isChecksumValid = expectedCheckDigit === actualCheckDigit;

  return {
    isValid: isChecksumValid,
    cleanedVin,
    error: isChecksumValid ? undefined : `Ошибка контрольной суммы: ожидалось '${expectedCheckDigit}', получено '${actualCheckDigit}'`
  };
}