aicheck 0.2.0

Detect AI-generated content via provenance signals (C2PA, XMP/IPTC, EXIF)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103

/// Known AI content generation tool name patterns (case-insensitive matching).
pub const AI_TOOL_PATTERNS: &[&str] = &[
    "dall-e",
    "dall·e",
    "dalle",
    "midjourney",
    "stable diffusion",
    "stablediffusion",
    "comfyui",
    "automatic1111",
    "a1111",
    "novelai",
    "adobe firefly",
    "firefly",
    "bing image creator",
    "copilot designer",
    "microsoft designer",
    "imagen",
    "ideogram",
    "flux",
    "leonardo.ai",
    "leonardo ai",
    "runway",
    "pika",
    "sora",
    "google veo",
    "veo",
    "kling",
    "vidu",
    "meta ai",
    "canva ai",
    "stability ai",
    "invoke ai",
    "invokeai",
    "dreamstudio",
    "nightcafe",
    "craiyon",
    "glide",
    "parti",
    "muse",
    "deepai",
    "foocus",
    "fooocus",
    "gpt-4o",
    "gpt-4",
    "gpt image",
    "openai",
    "chatgpt",
    "seedream",
    "recraft",
    "elevenlabs",
    "eleven labs",
    "suno",
    "udio",
    "soundraw",
    "aiva",
    "boomy",
    "mubert",
    "loudly.com",
    "beatoven",
    "soundful",
    // New image generation tools
    "grok",
    "gemini",
    "jimeng",
    "即梦",
    "dreamina",
    // New video generation tools
    "luma",
    "hailuo",
    "海螺",
    "pixverse",
    "genmo",
    "haiper",
    // New audio generation tools
    "hume",
    "fish audio",
];

/// Check if a string contains any known AI tool pattern (case-insensitive).
/// Returns the matched tool name if found.
pub fn match_ai_tool(text: &str) -> Option<&'static str> {
    let lower = text.to_lowercase();
    AI_TOOL_PATTERNS
        .iter()
        .find(|&&pattern| lower.contains(pattern))
        .copied()
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_match_known_tools() {
        assert_eq!(match_ai_tool("DALL-E 3/OpenAI"), Some("dall-e"));
        assert_eq!(match_ai_tool("Adobe Firefly 2.0"), Some("adobe firefly"));
        assert_eq!(match_ai_tool("Canon EOS R5"), None);
        assert_eq!(match_ai_tool("ComfyUI v1.2"), Some("comfyui"));
        assert_eq!(match_ai_tool("Midjourney v6"), Some("midjourney"));
        assert_eq!(match_ai_tool("Dreamina by ByteDance"), Some("dreamina"));
    }
}