KI-Kunst-Kirsten-Kloeckner/KIKunstKirstenKlöckner/Pages/AiArt.razor

@page "/aiart"
@using OpenAI_API
@using OpenAI_API.Chat
@using OpenAI_API.Models
@using System.Diagnostics

@inject TooltipService TooltipService
@inject DialogService DialogService

<PageTitle>AiArt</PageTitle>

<h1>Wunschbilder von KI nur für dich</h1>

@*<EditForm Model="@exampleModel" OnSubmit="@DoStuff">
    <InputText @bind-Value="exampleModel.Name" />

    <button type="submit">Submit</button>
</EditForm>*@

<!-- Dies ist das Textfeld -->
@*<input type="text" @bind="request" placeholder="Text eingeben..." />
*@
<!-- Dies ist der Button -->
@*<button @onclick="DoStuff">Klick mich</button>*@

<RadzenStack Orientation="Orientation.Vertical" AlignItems="AlignItems.Center">
    
    <RadzenText TextStyle="TextStyle.H2">Nenne uns deinen Wunsch:</RadzenText>
    <RadzenTextBox @bind-Value=@request Placeholder="Dein Wunsch"/>

    <RadzenPanel AllowCollapse="true" Style="width: 500px;" Text="Mehr Optionen">
        <ChildContent>
            <RadzenCard class="rz-mt-4">
                <RadzenStack Orientation="Orientation.Horizontal"
                             MouseEnter="@(args => ShowTemperatureTooltip(args))"
                             MouseLeave="TooltipService.Close"
                             AlignItems="AlignItems.Center" Wrap="FlexWrap.Wrap">
                    <RadzenText>Temperature:</RadzenText>
                    <RadzenSlider @bind-Value=@temperature TValue="float"
                                  Step="0.0001" Min="0.0m" Max="2.0m">
                    </RadzenSlider>
                    <RadzenText>@temperature</RadzenText>
                </RadzenStack>
            </RadzenCard>
        </ChildContent>
    </RadzenPanel>

    <RadzenButton Visible=@_buttonVisible Click=@DoStuff>Generate</RadzenButton>

    <RadzenPanel AllowCollapse="true" Style="width: 500px;" Text="Zeige Prompt">
        <ChildContent>
            <RadzenCard class="rz-mt-4">
                <RadzenStack Orientation="Orientation.Horizontal"
                             MouseEnter="@(args => ShowTemperatureTooltip(args))"
                             MouseLeave="TooltipService.Close"
                             AlignItems="AlignItems.Center" Wrap="FlexWrap.Wrap">
                    <RadzenText>@_imagePrompt</RadzenText>
                </RadzenStack>
            </RadzenCard>
        </ChildContent>
    </RadzenPanel>

    <RadzenText TextStyle="TextStyle.H4">Die Idee, die gemalt wird:</RadzenText>
    <RadzenText>@_imageDescription</RadzenText>

    <RadzenProgressBarCircular Visible=@_progressVisible ProgressBarStyle="ProgressBarStyle.Secondary" Value="100" ShowValue="false" Mode="ProgressBarMode.Indeterminate" />
    <RadzenText Visible=@_progressVisible TextStyle="TextStyle.H6" Text=@BusyMessage></RadzenText>
    
    <RadzenImage Path=@_imageUrl></RadzenImage>

</RadzenStack>

@code {

    private bool _progressVisible = false;
    private bool _buttonVisible = true;

    public string BusyMessage { get; set; } = "Initial Message";

    // Busy dialog from markup
    async Task ShowBusyDialog()
    {
        await DialogService.OpenAsync("", ds =>
            @<RadzenStack AlignItems="AlignItems.Center" Gap="2rem" Class="rz-p-12">
                <RadzenProgressBarCircular ProgressBarStyle="ProgressBarStyle.Secondary" Value="100" ShowValue="false" Mode="ProgressBarMode.Indeterminate"/>
        <RadzenText TextStyle="TextStyle.H6" Text=@BusyMessage></RadzenText>
            </RadzenStack>, new DialogOptions() { ShowTitle = false, Style = "min-height:auto;min-width:auto;width:auto", CloseDialogOnEsc = false });
    }

    // Busy dialog from string
    async Task BusyDialog(string message)
    {
        await DialogService.OpenAsync("", ds =>
        {
            RenderFragment content = b =>
            {
                b.OpenElement(0, "RadzenRow");

                b.OpenElement(1, "RadzenColumn");
                b.AddAttribute(2, "Size", "12");

                b.AddContent(3, message);

                b.CloseElement();
                b.CloseElement();
            };
            return content;
        }, new DialogOptions() { ShowTitle = false, Style = "min-height:auto;min-width:auto;width:auto", CloseDialogOnEsc = false });
    }

    void ShowTooltip(ElementReference elementReference, string text, TooltipOptions? options = null) => TooltipService.Open(elementReference, text, options);

    void ShowTemperatureTooltip(ElementReference elementReference) => TooltipService.Open(elementReference, ds =>
        @<div>
            Gibt an, wie <em>kreativ</em> ChatGPT sein soll.<br />
            Ich glaube, eigentlich bedeutet es eher, wie <em>deterministisch</em> die Ausgabe ist.<br />
            Bei 0.0 kommt immer fast die selbe Antwort. Bei zu hohen Werten kommt nur noch Schwachsinn.<br />
            OpenAI empfielt einen Wert von ca. 0.9 für kreative Anwendungen.
        </div>,
        new() { Position = TooltipPosition.Bottom, Duration = null});

    private string _imagePrompt = "\"Painting of a soccer match with the Hamburger SV team scoring multiple goals, vibrant colors and dynamic brushstrokes.\" Watercolor + ink on paper, Pen drawing, wet-on-wet technique, dry-on-dry technique, dabbing technique.";
    private string _imageDescription = "Ich habe mich für diese Interpretation des Titels entschieden, um die Dynamik und den Erfolg des Hamburger SV Fußballvereins darzustellen. Durch die Darstellung eines Spiels, in dem der HSV viele Tore schießt, wird die Leidenschaft und das Streben nach Erfolg hervorgehoben. Die lebendigen Farben und die dynamischen Pinselstriche sollen die Energie und Aufregung des Spiels wiedergeben.";

    private float temperature = 0.9f; 
    private string request = "";

    private string _basePrompt;

    private string _imageUrl;

    //protected override async Task OnInitializedAsync()
    //{
    //    _basePrompt = await File.ReadAllTextAsync($"{Directory.GetCurrentDirectory()}{@"\wwwroot\prompt.txt"}");
    //}

    private async Task FunnyMessageSwitcher_ImageGen(CancellationToken cancellationToken)
    {
        Stopwatch sw = Stopwatch.StartNew();

        await Task.Delay(1000, cancellationToken);

        await UpdateBusyMessage("Dauert noch eine Weile...");

        await Task.Delay(1000, cancellationToken);

        await UpdateBusyMessage("Gut Ding hat Weil...");

        await Task.Delay(1000, cancellationToken);

        await UpdateBusyMessage("Sach ma, was issn da l... achso, er ist auf Klo gegangen... Er ist bestimmt gleich wieder da...");

        await Task.Delay(1000, cancellationToken);

        await UpdateBusyMessage("Na, also langsam verlier ich hier die Geduld, ich dachte KI soll alles schneller machen...");
        
        while (!cancellationToken.IsCancellationRequested)
        {
            await Task.Delay(1000, cancellationToken);

            await UpdateBusyMessage($"Keine Sorge, er arbeitet noch. Die Bilder werden gemalt... ({sw.Elapsed.Seconds}s)");
        }
    }

    async Task UpdateBusyMessage(string newMessage)
    {
        BusyMessage = newMessage;
        await InvokeAsync(StateHasChanged);
    }

    private readonly HttpClient _client = new();

    protected override Task OnInitializedAsync()
    {
        var inferenceModelUrl = "https://api-inference.huggingface.co/models/Nacken/ki-kunst-kirsten-kloeckner-colab";

        // TODO: nicht im repo speichern
        var API_TOKEN = "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";

        //_client.BaseAddress = new Uri(inferenceModelUrl);

        _client.DefaultRequestHeaders.Clear();
        _client.DefaultRequestHeaders.Add("Authorization", $"Bearer {API_TOKEN}");
        
        return base.OnInitializedAsync();
    }

    /// <summary>
    /// Geneiert das Bild für den aktuellen <see cref="_imagePrompt"/>
    /// </summary>
    public async Task GenerateImageAsync()
    {
        var postData = new
        {
            inputs = _imagePrompt
        };

        JsonContent content = JsonContent.Create(postData);

        async Task FailedToDrawImage()
        {
            bool? retry = await DialogService.Confirm("Leider konnte das Bild nicht gemalt werden. Möchtest du es noch eimal versuchen?", "Ups, ein Fehler ist aufgetreten...",
                new ConfirmOptions { OkButtonText = "Ja", CancelButtonText = "Nein" });

            if (retry == true)
                await GenerateImageAsync();
        }

        try
        {
            var inferenceModelUrl = "https://api-inference.huggingface.co/models/Nacken/ki-kunst-kirsten-kloeckner-colab";

            var response = await _client.PostAsync(inferenceModelUrl, content);

            if (response?.IsSuccessStatusCode == true)
            {
                await using Stream imageStream = await response.Content.ReadAsStreamAsync();

                using Image image = await Image.LoadAsync(imageStream);

                string imgUrl = $"generated_images/{DateTime.Now:dd_MM_yyyy_hh_mm_s}.jpg";

                string mapPath = $"./wwwroot/{imgUrl}";
                await image.SaveAsJpegAsync(mapPath);

                _imageUrl = imgUrl;
            }
            else
            {
                Console.WriteLine($"Image conversion failed: {response}");

                if (Debugger.IsAttached)
                    Debugger.Break();

                await FailedToDrawImage();
            }
        }
        catch (Exception exception)
        {
            Console.WriteLine($"Image request failed: {exception}");

            if (Debugger.IsAttached)
                Debugger.Break();
                
            await FailedToDrawImage();
        }
    }

    private async Task DoStuff()
    {
        // Der Dialog blokiert so lange, wie der er offen ist, deshalb dürfen wir hier nicht warten, da wir sonst nie mit der Arbeit anfangen...
        //Task busyDialog = ShowBusyDialog();
        _progressVisible = true;
        _buttonVisible = false;

        await UpdateBusyMessage("Kirstens Assistent zerbricht sich über deine Idee den Kopf...");

        _basePrompt = await File.ReadAllTextAsync($"{Directory.GetCurrentDirectory()}{@"/wwwroot/prompt.txt"}");
        
        OpenAIAPI api = new OpenAIAPI("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");

        ChatRequest chatRequest = new ChatRequest
        {
            Temperature = temperature,
            Model = Model.ChatGPTTurbo,
        };

        Conversation converse = api.Chat.CreateConversation(chatRequest);

        converse.AppendUserInput(_basePrompt + " " + request);
        
        string response = await converse.GetResponseFromChatbotAsync();

        // Der Propt sollte dafür sorgen, dass ChatGPT einen Absatz zwischen Bild-Promt und der Erklärung macht.
        int lineBreak = response.IndexOf('\n');

        if (lineBreak == -1)
        {
            // TODO: Error
        }

        _imagePrompt = response.Substring(0, lineBreak);
        _imageDescription = response.Substring(lineBreak);

        await UpdateBusyMessage("Kirstens Assistent hat eine Idee! Er wird sie nun malen...");
        
        Task makeImagesTask = GenerateImageAsync();

        //CancellationTokenSource cancelFunnyMessages = new CancellationTokenSource();

        //Task funnyMessagesTask = FunnyMessageSwitcher_ImageGen(cancelFunnyMessages.Token);

        // Wichtig, erst Bilder awaiten, dann die lustingen Sprüche, sonst warten wir ewig...
        await makeImagesTask;

        //cancelFunnyMessages.Cancel();

        //await funnyMessagesTask;

        _progressVisible = false;
        _buttonVisible = true;
        await InvokeAsync(StateHasChanged);
    }
}