<?php

namespace common\components;

use common\models\Meal;
use common\models\MealForm;
use Exception;
use Yii;
use yii\helpers\FileHelper;
use yii\httpclient\Client;

class GeminiApiComponent extends \yii\base\Component
{
    public Client $client;
    public string $apiKey;
    public string $baseUrl;
    public string $model;

    public function init()
    {
        parent::init();
        $this->client = new Client([
            'baseUrl' => $this->baseUrl,
            'requestConfig' => [
                'format' => Client::FORMAT_JSON
            ],
            'responseConfig' => [
                'format' => Client::FORMAT_JSON
            ],
        ]);
    }

    public function mealInquiry(MealForm $model)
    {
        $data = [
            "contents" => [
                [
                    "role" => "user",
                    "parts" => [
                        [
                            "inline_data" => [
                                "data" => base64_encode(file_get_contents($model->filepath)),
                                "mimeType" => FileHelper::getMimeType($model->filepath)
                            ]
                        ]
                    ]
                ]
            ],
            "systemInstruction" => [
                "role" => "user",
                "parts" => [
                    [
                        "text" => "Provide a caloric and macro estimate for pictures I provide to you. Try to be as 
                        accurate as possible and always calculate the everything you see in the picture. Provide a 
                        3 or 7 word `food_name` with no special characters. If the user provides context pay attention 
                        as it may contain details about the calories or specifics details about the picture."
                    ]
                ]
            ],
            "generationConfig" => [
                "temperature" => 1,
                "topK" => 40,
                "topP" => 0.95,
                "maxOutputTokens" => 1000,
                "responseMimeType" => "application/json",
                "responseSchema" => [
                    "type" => "object",
                    "properties" => [
                        "food_name" => [
                            "type" => "string"
                        ],
                        "calories" => [
                            "type" => "integer"
                        ],
                        "protein" => [
                            "type" => "integer"
                        ],
                        "fat" => [
                            "type" => "integer"
                        ],
                        "carbohydrates" => [
                            "type" => "integer"
                        ],
                        "fiber" => [
                            "type" => "integer"
                        ]
                    ],
                    "required" => [
                        "food_name",
                        "calories",
                        "protein",
                        "fat",
                        "carbohydrates",
                        "fiber"
                    ]
                ]
            ]
        ];

        $data['contents'][0]['parts'][] = [
                'text' => !empty($model->context) ? 'Context: ' . $model->context : 'INSERT_TEXT_HERE'
        ];

        $response = $this->client
            ->post([$this->model, 'key' => $this->apiKey])
            ->setData($data)
            ->send();

        if ($response->statusCode != 200) {
            throw new Exception('There was an issue with the AI side of things - sorry! It is a MVP after all :/');
        }

        $meal = new Meal();
        $gemini = json_decode($response->getContent(), true);
        $geminiMeal = json_decode($gemini['candidates'][0]['content']['parts'][0]['text'], true);
        $meal->context = $model->context;
        $meal->date = $model->date->format('Y-m-d H:i:s');
        $meal->type = $model->type;
        $meal->protein = $geminiMeal['protein'];
        $meal->calories = $geminiMeal['calories'];
        $meal->carbohydrates = $geminiMeal['carbohydrates'];
        $meal->fat = $geminiMeal['fat'];
        $meal->fiber = $geminiMeal['fiber'];
        $meal->food_name = $geminiMeal['food_name'];
        // @TODO if moved a job queue then this must be an object otherwise the queue is NOT aware of the user
        $meal->user_id = Yii::$app->user->id;
        $meal->file_name = $model->filepath;
        Yii::debug($meal);
        $meal->validate();
        $errors = $meal->getErrors();
        $meal->save();
        // @TODO catch unidentified pictures?

        return $meal->id;
    }

}