vtvz · vtvz · Mar 14, 2026 · Feb 25, 2026 · Feb 26, 2026 · Feb 26, 2026
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -46,6 +46,8 @@ uuid = "1.20.0"
 whatlang = { version = "0.18.0", features = ["serde"] }
 redis = "0.32.7"
 prometheus = { version = "0.14.0", features = ["push", "process"] }
+csv = "1.4.0"
+async-trait = "0.1.89"
 
 [dependencies.isolang]
 features = ["serde"]

diff --git a/README.md b/README.md
@@ -7,7 +7,7 @@
 
 > **📢 Subscribe to the author's telegram channel for updates and more projects:** [**@vtvz_dev**](https://t.me/vtvz_dev)
 
-> A Telegram bot that monitors your Spotify playback, detects profane lyrics, integrates with AI for text analysis, and automatically skips tracks you've disliked
+> A Telegram bot that monitors your Spotify playback, detects profane lyrics and AI-generated music, integrates with AI for text analysis, and automatically skips tracks you've disliked
 
 [![Rust](https://img.shields.io/badge/rust-nightly-orange.svg)](https://www.rust-lang.org/)
 [![License](https://img.shields.io/badge/license-MIT-blue.svg)](LICENSE)
@@ -23,13 +23,14 @@
 
 ---
 
-Rustify is an intelligent Telegram bot that integrates with Spotify to provide real-time profanity detection and track management. It continuously monitors what you're listening to, analyzes lyrics for inappropriate content, and automatically skips tracks you've marked with the dislike button.
+Rustify is an intelligent Telegram bot that integrates with Spotify to provide real-time profanity detection, AI-generated music detection, and track management. It continuously monitors what you're listening to, analyzes lyrics for inappropriate content, detects AI-generated tracks, and automatically skips tracks you've marked with the dislike button.
 
 ## ✨ Features
 
 ### 🎯 Core Features
 
 - **🔍 Real-time Profanity Detection** - Automatically analyzes song lyrics as you listen using advanced profanity detection algorithms
+- **🤖 AI-Generated Music Detection** - Identifies AI-generated tracks using multiple detection providers and shows notifications with attribution
 - **⏭️ Auto-Skip** - Instantly skips tracks you've marked with dislike
 - **📊 Multi-Provider Lyrics** - Fetches lyrics from multiple sources (Musixmatch, Genius, LrcLib) for maximum coverage
 - **🤖 AI-Powered Analysis** - Optional OpenAI-compatible API integration for analyzing song lyrics meaning, storyline, and content themes, plus individual word analysis

diff --git a/locales/actions.yml b/locales/actions.yml
@@ -52,26 +52,26 @@ actions.dislike:
   en: |-
     👎 Disliked %{track_link}
 
-    If you change your mind, press 'Ignore text 🙈'
+    If you change your mind, press '%{ignore_button_label}'
 
     💡 Do not forget you can send a link to a song to find current status
   ru: |-
     👎 Дизлайкнул %{track_link}
 
-    Если передумаете, нажмите 'Игнорировать текст 🙈'
+    Если передумаете, нажмите '%{ignore_button_label}'
 
     💡 Не забывайте, что можете отправить ссылку на песню, чтобы узнать текущий статус
 
 actions.ignore:
   en: |-
     🙈 Bad words of %{track_link} will be forever ignored
 
-    If you change your mind, press 'Dislike 👎'
+    If you change your mind, press '%{dislike_button_label}'
 
     💡 Do not forget you can send a link to a song to find current status
   ru: |-
     🙈 Плохие слова из %{track_link} будут навсегда проигнорированы
 
-    Если передумаете, нажмите 'Не нравится 👎'
+    Если передумаете, нажмите '%{dislike_button_label}'
 
     💡 Не забывайте, что можете отправить ссылку на песню, чтобы узнать текущий статус
diff --git a/locales/ai_slop.yml b/locales/ai_slop.yml
@@ -0,0 +1,75 @@
+_version: 2
+
+ai-slop.alert:
+  en: |-
+    🎵 %{track_name}
+    Album: %{album_name}
+
+    <tg-emoji emoji-id="5429334864010681294">💩</tg-emoji> This track is most likely <b>%{prediction}</b>. You can block track or ignore similar notifications for this track
+
+    To permanently block this artist from recommendations:
+    <tg-emoji emoji-id="6028171274939797252">🔗</tg-emoji> Open the artist's page in Spotify
+    <tg-emoji emoji-id="6032924188828767321">➕</tg-emoji> Tap the three dots near the <i>"Follow"</i> button
+    <tg-emoji emoji-id="5774077015388852135">❌</tg-emoji> Select <i>"Don't play this artist"</i>
+
+    <i>Information provided by "%{ai_check_provider}"</i>
+
+    <b>Support real organic hand-made music instead of AI-slop!</b>
+
+    <blockquote>Want to disable these alerts or auto-skip AI tracks? Try /%{config_command}</blockquote>
+  ru: |-
+    🎵 %{track_name}
+    Альбом: %{album_name}
+
+    <tg-emoji emoji-id="5429334864010681294">💩</tg-emoji> Этот трек скорее всего <b>%{prediction}</b>. Вы можете заблокировать трек или отключить подобные уведомления для этого трека
+
+    Чтобы навсегда скрыть этого исполнителя из рекомендаций:
+    <tg-emoji emoji-id="6028171274939797252">🔗</tg-emoji> Откройте страницу исполнителя в Spotify
+    <tg-emoji emoji-id="6032924188828767321">➕</tg-emoji> Нажмите на три точки рядом с кнопкой <i>«Подписаться»</i>
+    <tg-emoji emoji-id="5774077015388852135">❌</tg-emoji> Выберите <i>«Не включать треки этого исполнителя»</i>
+
+    <i>Информация предоставлена "%{ai_check_provider}"</i>
+
+    <b>Поддерживайте настоящую живую музыку вместо ИИ-шлака!</b>
+
+    <blockquote>Хотите отключить уведомления или настроить автопропуск AI-треков? Попробуйте /%{config_command}</blockquote>
+
+ai-slop.alert-pure-ai:
+  en: fully AI-generated
+  ru: полностью сгенерирован ИИ
+
+ai-slop.alert-processed-ai:
+  en: partially generated or processed by AI
+  ru: частично сгенерирован или обработан ИИ
+
+ai-slop.setting-description:
+  en: |-
+    <tg-emoji emoji-id="5429334864010681294">💩</tg-emoji> AI Music Detection
+
+    Bot automatically checks tracks for AI generation
+
+    Configure bot's reaction to such tracks:
+  ru: |-
+    <tg-emoji emoji-id="5429334864010681294">💩</tg-emoji> Обнаружение ИИ-музыки
+
+    Бот автоматически проверяет треки на ИИ-генерацию
+
+    Настройте реакцию бота на такие треки:
+
+ai-slop.button-notify:
+  en: |-
+    %{mark}Notify
+  ru: |-
+    %{mark}Уведомлять
+
+ai-slop.button-ignore:
+  en: |-
+    %{mark}Don't notify
+  ru: |-
+    %{mark}Не уведомлять
+
+ai-slop.button-skip:
+  en: |-
+    %{mark}Skip automatically
+  ru: |-
+    %{mark}Пропускать автоматически
diff --git a/locales/inline_buttons.yml b/locales/inline_buttons.yml
@@ -2,22 +2,28 @@ _version: 2
 
 inline-buttons.dislike:
   en: |-
-    Dislike 👎
+    Block track 👎
   ru: |-
-    Не нравится 👎
+    Заблокировать трек 👎
 
 inline-buttons.ignore:
   en: |-
-    Ignore text 🙈
+    Hide alerts for this track 🙈
   ru: |-
-    Игнорировать текст 🙈
+    Скрыть уведомления для этого трека 🙈
 
 inline-buttons.analyze:
   en: |-
     Analyze text 🔍
   ru: |-
     Анализировать текст 🔍
 
+inline-buttons.artist-page:
+  en: |-
+    Go to artist's page 🔗
+  ru: |-
+    Перейти на страницу исполнителя 🔗
+
 inline-buttons.alert-login:
   en: |-
     You need to login first

diff --git a/locales/login.yml b/locales/login.yml
@@ -33,6 +33,7 @@ login.invite:
     <b>Features:</b>
     • Block unwanted songs ("Dislike")
     • Notifications for profanity in lyrics (English only)
+    • Detect AI-generated music (notify/ignore/auto-skip)
     • Get song lyrics and AI-powered analysis
     • Create <i>Magic Playlist™</i>✨ from your favorite tracks
     • <i>Skippage™</i> function will diverse your listening experience
@@ -48,6 +49,7 @@ login.invite:
     <b>Возможности:</b>
     • Блокировка нежелательных песен ("Дизлайк")
     • Уведомление на нецензурную лексику в тексте (только английский)
+    • Обнаружение ИИ-музыки (уведомление/игнорирование/авто-пропуск)
     • Получение текстов песен и их анализ с помощью ИИ
     • Создание <i>Magic Playlist™</i>✨ из любимых треков
     • Функция <i>Skippage™</i> разнообразит ваши впечатления от прослушивания

diff --git a/locales/profanity_check.yml b/locales/profanity_check.yml
@@ -8,12 +8,12 @@ profanity-check.message:
 
     <a href="%{lyrics_link}">%{lyrics_link_text}</a>
 
-    Press 'Ignore text 🙈' to never see this notification for <b>this song</b> again
+    Press '%{ignore_button_label}' to never see this notification for <b>this song</b> again
   ru: |-
     🚨 Текущая песня (%{track_name}) <b>вероятно</b> содержит нецензурные слова:
 
     %{bad_lines}
 
     <a href="%{lyrics_link}">%{lyrics_link_text}</a>
 
-    Нажмите 'Игнорировать текст 🙈', чтобы больше не видеть это уведомление для <b>этой песни</b>
+    Нажмите '%{ignore_button_label}', чтобы больше не видеть это уведомление для <b>этой песни</b>
diff --git a/migrations/20260302172253_add_user_cfg_ai_slop_detection_field.sql b/migrations/20260302172253_add_user_cfg_ai_slop_detection_field.sql
@@ -0,0 +1,2 @@
+alter table "user"
+    add cfg_ai_slop_detection text default 'notify' not null;
diff --git a/src/app.rs b/src/app.rs
@@ -16,7 +16,7 @@ use teloxide::requests::RequesterExt as _;
 use crate::metrics::influx::InfluxClient;
 use crate::metrics::prometheus::PrometheusClient;
 use crate::queue::QueueManager;
-use crate::services::{SongLinkService, UserService};
+use crate::services::{AISlopDetectionService, SongLinkService, UserService};
 use crate::user::UserState;
 use crate::{lyrics, profanity, spotify};
 
@@ -32,6 +32,7 @@ pub struct App {
     dialogue_storage: Arc<TeloxideRedisStorage<Bincode>>,
     server_http_address: String,
     song_link: SongLinkService,
+    ai_slop_detection: AISlopDetectionService,
     queue_manager: QueueManager,
 }
 
@@ -67,6 +68,8 @@ struct EnvConfig {
     genius_service_url: String,
     lyrics_cache_ttl: Option<u64>,
 
+    shlabs_api_key: Option<String>,
+
     censor_blacklist: Option<String>,
     censor_whitelist: Option<String>,
 
@@ -137,6 +140,10 @@ impl App {
     pub fn queue_manager(&self) -> &QueueManager {
         &self.queue_manager
     }
+
+    pub fn ai_slop_detection(&self) -> &AISlopDetectionService {
+        &self.ai_slop_detection
+    }
 }
 
 fn init_influx(env: &EnvConfig) -> anyhow::Result<Option<InfluxClient>> {
@@ -338,6 +345,8 @@ impl App {
 
         let queue_manager = QueueManager::new(redis_url).await?;
 
+        let shlabs_api_key = env.shlabs_api_key.filter(|key| !key.trim().is_empty());
+
         // Make global static variable to prevent hassle with Arc
         let app = Box::new(Self {
             bot,
@@ -354,6 +363,7 @@ impl App {
                 .server_http_address
                 .unwrap_or_else(|| "0.0.0.0:3000".into()),
             queue_manager,
+            ai_slop_detection: AISlopDetectionService::new(shlabs_api_key),
         });
 
         let app = &*Box::leak(app);

diff --git a/src/entity/prelude.rs b/src/entity/prelude.rs
@@ -19,6 +19,7 @@ pub use super::track_status::{
 };
 #[allow(unused_imports)]
 pub use super::user::{
+    AISlopDetection as UserAISlopDetection,
     ActiveModel as UserActiveModel,
     Column as UserColumn,
     Entity as UserEntity,

diff --git a/src/entity/user.rs b/src/entity/user.rs
@@ -38,6 +38,8 @@ pub struct Model {
     pub cfg_skip_tracks: bool,
     pub cfg_skippage_secs: i64,
     pub cfg_skippage_enabled: bool,
+    #[sea_orm(enum_name = "CfgAISlopDetection")]
+    pub cfg_ai_slop_detection: AISlopDetection,
     pub magic_playlist: Option<String>,
     pub spotify_state: Uuid,
     pub ref_code: Option<String>,
@@ -84,6 +86,8 @@ pub enum Column {
     CfgSkipTracks,
     CfgSkippageSecs,
     CfgSkippageEnabled,
+    #[sea_orm(column_name = "cfg_ai_slop_detection")]
+    CfgAISlopDetection,
     MagicPlaylist,
     SpotifyState,
     RefCode,
@@ -126,6 +130,7 @@ impl ColumnTrait for Column {
             Self::CfgSkipTracks => ColumnType::Boolean.def(),
             Self::CfgSkippageSecs => ColumnType::BigInteger.def(),
             Self::CfgSkippageEnabled => ColumnType::Boolean.def(),
+            Self::CfgAISlopDetection => AISlopDetection::db_type(),
             Self::MagicPlaylist => ColumnType::Text.def().null(),
             Self::SpotifyState => ColumnType::Uuid.def(),
             Self::RefCode => ColumnType::Text.def().null(),
@@ -302,3 +307,50 @@ impl Role {
         matches!(self, Self::Admin)
     }
 }
+
+#[derive(
+    Debug, Copy, Clone, EnumIter, DeriveActiveEnum, PartialEq, Eq, Default, Serialize, Deserialize,
+)]
+#[sea_orm(rs_type = "String", db_type = "Text")]
+pub enum AISlopDetection {
+    #[sea_orm(string_value = "skip")]
+    Skip,
+    #[sea_orm(string_value = "notify")]
+    #[default]
+    Notify,
+    #[sea_orm(string_value = "ignore")]
+    Ignore,
+}
+
+impl FromStr for AISlopDetection {
+    type Err = sea_orm::DbErr;
+
+    fn from_str(s: &str) -> Result<Self, Self::Err> {
+        Self::try_from(s)
+    }
+}
+
+impl TryFrom<&str> for AISlopDetection {
+    type Error = sea_orm::DbErr;
+
+    fn try_from(value: &str) -> Result<Self, Self::Error> {
+        Self::try_from_value(&value.to_owned())
+    }
+}
+
+impl AISlopDetection {
+    #[must_use]
+    pub fn is_skip(&self) -> bool {
+        matches!(self, Self::Skip)
+    }
+
+    #[must_use]
+    pub fn is_notify(&self) -> bool {
+        matches!(self, Self::Notify)
+    }
+
+    #[must_use]
+    pub fn is_ignore(&self) -> bool {
+        matches!(self, Self::Ignore)
+    }
+}
diff --git a/src/lyrics/mod.rs b/src/lyrics/mod.rs
@@ -3,7 +3,6 @@ use genius::GeniusLocal;
 use isolang::Language;
 use lrclib::LrcLib;
 use musixmatch::Musixmatch;
-use serde::Serialize;
 use strum_macros::Display;
 
 use crate::spotify::ShortTrack;
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		alter table "user"
		add cfg_ai_slop_detection text default 'notify' not null;
Comment thread vtvz marked this conversation as resolved.