Add API endpoint for generating music prompts from Songsterr URLs and update GeneratorForm for Songsterr integration

GlebkaF · GlebkaF · commit 2a67eacb26e5 · 2025-12-26T18:54:56.000+07:00
diff --git a/app/api/generate-from-songsterr/route.ts b/app/api/generate-from-songsterr/route.ts
@@ -3,7 +3,7 @@ import { createGenerator } from "../../../lib/ai-generator/create-generator";
 import {
   extractSongsterrId,
   fetchSongsterrData,
-  buildPromptFromSongsterr,
+  buildPromptWithMetadata,
 } from "../../../lib/utils/songsterr";
 
 // Исключаем этот API роут из статической генерации
@@ -84,21 +84,26 @@ export async function POST(
       );
     }
 
-    // Шаг 3: Формируем промпт с учетом конкретного trackId из URL
-    const prompt = buildPromptFromSongsterr(songData, trackType, trackId);
-    console.log(`💡 Generated prompt: "${prompt}"`);
+    // Шаг 3: Формируем промпт с метаданными с учетом конкретного trackId из URL
+    const promptResult = buildPromptWithMetadata(songData, trackType, trackId);
+    console.log(`💡 Generated prompt: "${promptResult.prompt}"`);
+    console.log("📊 Metadata:", promptResult.metadata);
 
     // Шаг 4: Запускаем генератор с сформированным промптом
     const generator = await createGenerator();
-    const generationId = await generator.generate(prompt);
+    const generationId: string = await generator.generate(
+      promptResult.prompt,
+      songsterrUrl,
+      promptResult.metadata,
+    );
 
     console.log(`✅ Generation created with ID: ${generationId}`);
 
     // Формируем ответ
     const response: GenerateFromSongsterrResponse = {
       generationId,
       message: "Generation created successfully from Songsterr URL",
-      prompt,
+      prompt: promptResult.prompt,
       songData: {
         artist: songData.artist,
         title: songData.title,
diff --git a/components/PresetCreateForm.tsx b/components/PresetCreateForm.tsx
@@ -1,10 +1,11 @@
 "use client";
 
-import { useState, useMemo } from "react";
+import { useState, useMemo, useEffect } from "react";
 import { useRouter } from "next/navigation";
 import Image from "next/image";
 import { GenerationRecord } from "../lib/jsondb/types";
 import { ValidatedArtist } from "../lib/public/schemas/artist";
+import { smartMapToPart } from "../lib/utils/track-mapping";
 
 interface PresetCreateFormProps {
   generation: GenerationRecord;
@@ -50,6 +51,53 @@ export function PresetCreateForm({
     },
   });
 
+  // Автозаполнение формы из songsterrData
+  useEffect(() => {
+    if (!generation.songsterrData) return;
+
+    const { artist, title, trackType, trackName, url } =
+      generation.songsterrData;
+
+    // 1. Поиск существующего артиста (fuzzy match)
+    const foundArtist = artists.find(
+      (a) =>
+        a.title.toLowerCase() === artist.toLowerCase() ||
+        a.title.toLowerCase().includes(artist.toLowerCase()) ||
+        artist.toLowerCase().includes(a.title.toLowerCase()),
+    );
+
+    if (foundArtist) {
+      // Артист найден - выбираем его
+      setIsNewArtist(false);
+      setFormData((prev) => ({
+        ...prev,
+        artistId: foundArtist.id,
+        song: title,
+        part: smartMapToPart(trackType, trackName),
+        tabsUrl: url,
+      }));
+      console.log(
+        `✅ Артист найден: ${foundArtist.title} (ID: ${String(foundArtist.id)})`,
+      );
+    } else {
+      // Артист не найден - создаем нового
+      setIsNewArtist(true);
+      setFormData((prev) => ({
+        ...prev,
+        newArtistTitle: artist,
+        newArtistDescription: `Описание для ${artist}`,
+        song: title,
+        part: smartMapToPart(trackType, trackName),
+        tabsUrl: url,
+      }));
+      console.log(`📝 Создание нового артиста: ${artist}`);
+    }
+
+    console.log(
+      `🎵 Автозаполнение: ${artist} - ${title} (${smartMapToPart(trackType, trackName)})`,
+    );
+  }, [generation.songsterrData, artists]);
+
   // Получаем последнюю версию chain из генерации (не используется в UI, но может понадобиться)
   const latestChain = useMemo(() => {
     if (generation.versions.length === 0) return null;
@@ -163,6 +211,33 @@ export function PresetCreateForm({
         }}
         className="space-y-6"
       >
+        {/* Автозаполнение из Songsterr */}
+        {generation.songsterrData && (
+          <div className="bg-green-50 border border-green-200 rounded-lg p-4 mb-6">
+            <h3 className="text-lg font-semibold text-green-900 mb-2">
+              ✅ Форма автозаполнена из Songsterr
+            </h3>
+            <div className="text-sm text-green-700 space-y-1">
+              <p>
+                <span className="font-medium">Артист:</span>{" "}
+                {generation.songsterrData.artist}
+              </p>
+              <p>
+                <span className="font-medium">Песня:</span>{" "}
+                {generation.songsterrData.title}
+              </p>
+              <p>
+                <span className="font-medium">Тип трека:</span>{" "}
+                {generation.songsterrData.trackType}
+              </p>
+              <p className="text-xs text-green-600 mt-2">
+                💡 Проверьте автозаполненные поля ниже и внесите правки при
+                необходимости
+              </p>
+            </div>
+          </div>
+        )}
+
         {/* Информация о генерации */}
         <div className="bg-gray-50 rounded-lg p-4">
           <h3 className="text-lg font-semibold text-gray-900 mb-3">
diff --git a/lib/ai-generator/create-generator.ts b/lib/ai-generator/create-generator.ts
@@ -35,7 +35,16 @@ interface RealRig {
 class ChainGenerator {
   constructor(private openai: OpenAI) {}
 
-  async generate(prompt: string): Promise<string> {
+  async generate(
+    prompt: string,
+    songsterrUrl?: string,
+    songsterrMetadata?: {
+      artist: string;
+      title: string;
+      trackType: string;
+      trackName?: string;
+    },
+  ): Promise<string> {
     const proDescription = await this.createProDescription(prompt);
     const realRig = await this.createRealRig(proDescription);
     const emptyChain = await this.createEmptyChain(realRig);
@@ -46,7 +55,22 @@ class ChainGenerator {
 
     // Сохраняем результаты в базу данных
     const now = new Date().toISOString();
-    const generationId = await this.saveToDatabase({
+
+    const dataToSave: {
+      timestamp: string;
+      originalPrompt: string;
+      proDescription: ProDescription;
+      realRig: RealRig;
+      finalChain: Chain;
+      versions: Array<{ chain: Chain; prompt: string; timestamp: string }>;
+      songsterrData?: {
+        url: string;
+        artist: string;
+        title: string;
+        trackType: string;
+        trackName?: string;
+      };
+    } = {
       timestamp: now,
       originalPrompt: prompt,
       proDescription,
@@ -59,7 +83,21 @@ class ChainGenerator {
           timestamp: now,
         },
       ],
-    });
+    };
+
+    if (songsterrUrl && songsterrMetadata) {
+      dataToSave.songsterrData = {
+        url: songsterrUrl,
+        artist: songsterrMetadata.artist,
+        title: songsterrMetadata.title,
+        trackType: songsterrMetadata.trackType,
+        ...(songsterrMetadata.trackName
+          ? { trackName: songsterrMetadata.trackName }
+          : {}),
+      };
+    }
+
+    const generationId = await this.saveToDatabase(dataToSave);
 
     return generationId;
   }
@@ -149,6 +187,13 @@ class ChainGenerator {
     realRig: RealRig;
     finalChain: Chain;
     versions: Array<{ chain: Chain; prompt: string; timestamp: string }>;
+    songsterrData?: {
+      url: string;
+      artist: string;
+      title: string;
+      trackType: string;
+      trackName?: string;
+    };
   }): Promise<string> {
     try {
       const generationId = await generationDb.addGeneration(result);
diff --git a/lib/jsondb/types.ts b/lib/jsondb/types.ts
@@ -30,6 +30,13 @@ export interface GenerationRecord {
     settings: unknown;
   };
   versions: GenerationVersion[];
+  songsterrData?: {
+    url: string;
+    artist: string;
+    title: string;
+    trackType: string;
+    trackName?: string;
+  };
 }
 
 export interface JsonDatabase {
diff --git a/lib/utils/songsterr.ts b/lib/utils/songsterr.ts
@@ -147,6 +147,16 @@ function detectGuitarType(instrument: string): string | null {
   return null;
 }
 
+export interface SongsterrPromptResult {
+  prompt: string;
+  metadata: {
+    artist: string;
+    title: string;
+    trackType: string;
+    trackName?: string;
+  };
+}
+
 /**
  * Формирует промпт для генератора из данных Songsterr
  * @param songData Данные о песне из Songsterr
@@ -236,6 +246,92 @@ export function buildPromptFromSongsterr(
   return prompt;
 }
 
+/**
+ * Формирует промпт и метаданные для генератора из данных Songsterr
+ * @param songData Данные о песне из Songsterr
+ * @param trackType Опциональный тип трека (Rhythm/Solo/Lead)
+ * @param specificTrackId Опциональный ID конкретного трека (partId из URL)
+ * @returns Объект с промптом и метаданными
+ */
+export function buildPromptWithMetadata(
+  songData: SongsterrSongData,
+  trackType?: string,
+  specificTrackId?: number | null,
+): SongsterrPromptResult {
+  const artist = songData.artist;
+  const title = songData.title;
+
+  // Если trackType не указан, определяем автоматически
+  let guitarType = trackType;
+  let selectedTrackName: string | undefined;
+
+  if (!guitarType && songData.tracks && songData.tracks.length > 0) {
+    // Фильтруем гитарные треки
+    const guitarTracks = songData.tracks.filter((t) => t.isGuitar);
+
+    if (guitarTracks.length > 0) {
+      let selectedTrack: SongsterrTrack | undefined;
+
+      // Если указан конкретный trackId из URL, используем его
+      if (specificTrackId !== undefined && specificTrackId !== null) {
+        selectedTrack = guitarTracks.find((t) => t.partId === specificTrackId);
+      }
+
+      // Если трек не найден или не указан, берем самый популярный
+      if (!selectedTrack) {
+        const popularPartId =
+          songData.popularTrackGuitar !== undefined
+            ? songData.popularTrackGuitar
+            : (guitarTracks[0]?.partId ?? 0);
+
+        selectedTrack = guitarTracks.find((t) => t.partId === popularPartId);
+
+        if (!selectedTrack) {
+          selectedTrack = guitarTracks.reduce((prev, current) =>
+            current.views > prev.views ? current : prev,
+          );
+        }
+      }
+
+      // Сохраняем название трека
+      selectedTrackName = selectedTrack.title;
+
+      // Определяем тип трека по его названию
+      const trackName = selectedTrack.title.toLowerCase();
+
+      if (trackName.includes("lead")) {
+        guitarType = "Lead";
+      } else if (trackName.includes("solo")) {
+        guitarType = "Solo";
+      } else if (trackName.includes("rhythm")) {
+        guitarType = "Rhythm";
+      } else if (trackName.includes("background")) {
+        guitarType = "Rhythm";
+      } else {
+        guitarType = selectedTrack.partId <= 1 ? "Rhythm" : "Lead";
+      }
+    }
+  }
+
+  // По умолчанию используем Rhythm
+  if (!guitarType) {
+    guitarType = "Rhythm";
+  }
+
+  // Формируем промпт
+  const prompt = `${artist} ${title} ${guitarType} Guitar Main Riff`;
+
+  return {
+    prompt,
+    metadata: {
+      artist,
+      title,
+      trackType: guitarType,
+      ...(selectedTrackName ? { trackName: selectedTrackName } : {}),
+    },
+  };
+}
+
 /**
  * Получает список доступных гитарных треков из данных Songsterr
  * @param songData Данные о песне
diff --git a/lib/utils/track-mapping.ts b/lib/utils/track-mapping.ts
