#!/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'} || die "OLLAMA_URL environment variable not set\n";
my $api = $ENV{'API_BASE_URL'} || die "API_BASE_URL environment variable not set\n";
my $task_count = $ENV{'TASK_COUNT'} || 1;
my $sensitivity = 20;
my $resolution = "672:672";

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);
    
    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'     => 'srizon/pixie:latest',
            'prompt'    => "Are there any african american people or in this photo, whether they are shown completely or not?  Please answer 'yes' or 'no' only.",
            'images'    => [$base64],
        };

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

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

    };
}

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;
}