<?php

/**
 * @copyright  (C) 2025 Open Source Matters, Inc. <https://www.joomla.org>
 * @license    GNU General Public License version 2 or later; see LICENSE
 */

namespace Joomla\AI\Tests;

use Joomla\AI\Provider\OpenAIProvider;
use PHPUnit\Framework\TestCase;

/**
 * Test class for Joomla\AI\Provider\OpenAIProvider.
 */
class VisionTest extends TestCase
{
    protected $config;

    protected $api_key;

    protected $provider;

    /**
     * Sets up the fixture, for example, opens a network connection.
     * This method is called before a test is executed.
     *
     * @return  void
     *
     * @since   1.0
     */
    protected function setUp(): void
    {
        parent::setUp();

        $configFile    = __DIR__ . '/../config.json';
        $this->config  = json_decode(file_get_contents($configFile), true);
        $this->api_key = $config['openai_api_key'] ?? null;
        // Create provider with your API key
        $this->provider = new OpenAIProvider([
            'api_key' => $this->api_key
        ]);
    }

    public function testVisionWithImageURL()
    {
        $imageUrl = "https://upload.wikimedia.org/wikipedia/commons/e/eb/Ash_Tree_-_geograph.org.uk_-_590710.jpg";

        $response = $this->provider->vision("What do you see in this image?", $imageUrl);

        $this->assertEquals(200, $response->getStatusCode());
        $this->assertTrue(strlen($response->getBody()) > 10);
        $this->assertEquals('OpenAI', $response->getProvider());
        $this->assertArrayHasKey('model', $response->getMetadata());
        $this->assertArrayHasKey('usage', $response->getMetadata());
    }

    public function testVisionWithSpecificModel()
    {
        $imageUrl = "https://upload.wikimedia.org/wikipedia/commons/e/eb/Ash_Tree_-_geograph.org.uk_-_590710.jpg";

        $response = $this->provider->vision(
            "Describe the colors and mood of this image.",
            $imageUrl,
            ['model' => 'gpt-4o', 'max_tokens' => 100]
        );

        echo "Vision API call successful!\n";
        echo "Response: " . $response->getContent() . "\n";
        echo "Provider: " . $response->getProvider() . "\n";

        $metadata = $response->getMetadata();
        if (!empty($metadata)) {
            echo "Model used: " . ($metadata['model']) . "\n";
            if (isset($metadata['usage'])) {
                echo "Tokens used: " . ($metadata['usage']['total_tokens']) . "\n";
            }
        }
    }
}