CultureQuestORG/SDP2023

package ch.epfl.culturequest.backend.artprocessing.apis;

import android.annotation.SuppressLint;

import com.theokanning.openai.service.OpenAiService;

import java.time.Duration;
import java.util.concurrent.CompletableFuture;

import ch.epfl.culturequest.BuildConfig;
import ch.epfl.culturequest.backend.artprocessing.processingobjects.BasicArtDescription;
import ch.epfl.culturequest.database.Database;


/**
 * Combines the results of the RecognitionApi and the WikipediaDescriptionApi
 * This is the main entry point for the art scanning functionality
 * After the user has taken a picture, the image is uploaded, its URL retrieved, and then passed to this class
 */

public class ProcessingApi {


    /** Returns an art description object (as a future) given the URL of the image associated to the scanned piece of art */
    @SuppressLint("NewApi")
    public CompletableFuture<BasicArtDescription> getArtDescriptionFromUrl(String imageUrl){

        RecognitionApi recognitionApi = new RecognitionApi("https://lens.google.com/uploadbyurl");


        OpenAiService openAiService = new OpenAiService(BuildConfig.OPEN_AI_API_KEY, Duration.ofSeconds(40));
        WikipediaDescriptionApi wikipediaDescriptionApi = new WikipediaDescriptionApi("https://en.wikipedia.org/wiki/Special:Search?search=");
        GeneralDescriptionApi descriptionApi = new GeneralDescriptionApi(wikipediaDescriptionApi, openAiService);

        return recognitionApi.getArtName(imageUrl)
                .thenCompose((artRecognition) -> {
                    CompletableFuture<BasicArtDescription> basicArtDescription = new CompletableFuture<>();
                    CompletableFuture<BasicArtDescription> api = descriptionApi.getArtDescription(artRecognition);
                    api.acceptEither(Database.getArtworkScan(artRecognition.getArtName()), (description) -> {
                        basicArtDescription.complete(description);

                        // Cancel the API call if a result is found (has no effect if the API call has already completed)
                        api.cancel(true);
                    });
                    return basicArtDescription;
                });
    }

}