VideoDetect/aiworker/run.pl

#!/usr/bin/perl
use strict;
use MIME::Base64 qw(encode_base64);
use LWP::UserAgent;
use JSON;
use Data::Dumper;

my $seconds = $ENV{'SECONDS_PER_FRAME'} || 30;
my $ollama = $ENV{'OLLAMA_URL'} || 'http://10.0.1.7:11434';
my $api = $ENV{'API_BASE_URL'} || 'http://10.0.1.7:3000';
my $task_count = $ENV{'TASK_COUNT'} || 1;
my $sensitivity = 20;
my $resolution = "672:672";
my $model = "hf.co/Jommarn/UNSEEN_Gemma_4_26B_NSFW-GGUF:Q8_0";

chdir("/tmp") || die "Unable to change directory to /tmp: $!\n";

my $ua = LWP::UserAgent->new;
$ua->timeout(60);

for(my $i=0; $i<$task_count; $i++) {
    my $ref = GetNext();
    my $task = $ref->{'task'};
    my $video = $ref->{'video'};

    my $task_id = $task->{'id'};
    my $video_id = $video->{'id'};
    my $path = $video->{'file_path'};
    my $result = CheckPath($path);

    next unless $result;
    
    my $response = PostResponse($task_id, $ref->{'assign_key'}, $result);
}

print "Done\n";
exit(0);

sub CheckPath {
    my ($path) = @_;
    die "File path not set\n" unless $path;
    die "File does not exist: $path\n" unless -e $path;

    unlink(<*.png>);
    system("ffmpeg -i \"$path\" -vf fps=1/$seconds,scale=$resolution output_%04d.png");


    my $total=0;
    my $yes=0;
    $/ = undef;
    foreach my $file (<*.png>) {
        print "$file...\n";
        open(IN, '<:raw', $file) || die "Unable to read $file: $!\n";
        my $data = <IN>;
        close(IN);

        my $base64 = encode_base64($data);
        $data=undef;

        my $ref = {
            'model'     => $model,
            'prompt'    => "Are there any african american men in this photo, whether they are shown completely or not?  Judge based on skin tone, accounting for shadows.  Answer yes for very dark skin.Please answer 'yes' or 'no' only.",
            'images'    => [$base64],
        };

        my $url = "$ollama/api/generate";
        my $res = GetRest($url, $ref);
        print "$file\t$res\n";
        $total++;
        if($res=~/yes$/i) {
            $yes++;
        }
        unlink($file);
    }

    if($total == 0) {
        warn "No images were processed for $path\n";
        return undef;
    }

    my $percentage = ($yes*100)/$total;
    print "$yes of $total images gave yes - $percentage\n";
    my $verdict = 0;
    if($percentage >= $sensitivity) {
        print "$path: Probably a black dude, $percentage\n";
        $verdict = 1;
    } else {
        print "$path: Probably not, $percentage\n";
        $verdict = 0;
    }
    return {
        'images'    => $total,
        'yes'       => $yes,
        'percentage'=> $percentage,
        'verdict'   => $verdict,
        'model'     => $model,
        'path'      => $path,

    };
}

sub GetRest {
    my($url, $ref)=@_;
    my $payload = encode_json($ref);

    my $response = $ua->post($url, 'Content_Type' => 'application/json', 'Content' => $payload);
    if($response->is_success) {
        my $content = $response->decoded_content;
        my @parts = split(/\n/, $content);
        my @refs = map { decode_json($_) } @parts;
        return join(' ', map { $_->{'response'} } @refs);
    } else {
        die "Error: " . $response->status_line;
    }
}

sub PostResponse {
    my ($task_id, $assign_key, $results) = @_;
    my $url = "$api/api/v1/task/$task_id/complete";
    my $ref = {
        'assign_key' => $assign_key,
        'results'    => $results,
    };
    my $payload = encode_json($ref);
    my $response = $ua->post($url, 'Content_Type' => 'application/json', 'Content' => $payload);
    if($response->is_success) {
        return $response->decoded_content;
    } else {
        die "Error posting response ($url): " . $response->status_line;
    }
}

sub GetNext {
    my $url = "$api/api/v1/nexttask/AISCAN";
    my $task;
    my $video;

    my $response = $ua->get($url);
    if($response->is_success) {
        $task=decode_json($response->decoded_content);
    } else {
        die "Error fetching next task ($url): " . $response->status_line;
    }

    my $video_id = $task->{'task'}{'video_id'};
    $url = "$api/api/v1/video/$video_id";
    $response = $ua->get($url);
    if($response->is_success) {
        $video=decode_json($response->decoded_content);
        $task->{'video'} = $video;
    } else {
        die "Error fetching video ($url): " . $response->status_line;
    }

    return $task;
}