UE5-cyberHuman/Plugins/OpenAi/Source/OpenAIEditor/Public/UI/ChatGPTWidget.h

// OpenAI, Copyright LifeEXE. All Rights Reserved.

#pragma once

#include "CoreMinimal.h"
#include "EditorUtilityWidget.h"
#include "UI/ChatMessageWidget.h"
#include "Provider/ResponseTypes.h"
#include "Provider/RequestTypes.h"
#include "Provider/CommonTypes.h"
#include "FuncLib/ModelTypes.h"
#include "ChatGPT/BaseService.h"
#include "ChatGPTWidget.generated.h"

class UVerticalBox;
class UScrollBox;
class UEditableText;
class UButton;
class UChatMessageWidget;
class UComboBoxString;
class UChatGPT;
class UServiceWidget;
class UGridPanel;
class USaveSettings;
class UTextBlock;
class UAttachedFilesContainerWidget;

UCLASS(BlueprintType)
class OPENAIEDITOR_API UChatGPTWidget : public UEditorUtilityWidget
{
    GENERATED_BODY()

protected:
    UPROPERTY(EditDefaultsOnly, BlueprintReadWrite, Category = "OpenAI")
    int32 MaxTokens{2000};

protected:
    UPROPERTY(meta = (BindWidget))
    TObjectPtr<UVerticalBox> StartVerticalBox;

    UPROPERTY(meta = (BindWidget))
    TObjectPtr<UComboBoxString> ChatGPTModelComboBox;

    UPROPERTY(meta = (BindWidget))
    TObjectPtr<UVerticalBox> MessagesVerticalBox;

    UPROPERTY(meta = (BindWidget))
    TObjectPtr<UScrollBox> ChatScrollBox;

    UPROPERTY(meta = (BindWidget), BlueprintReadWrite, Category = "OpenAI")
    TObjectPtr<UEditableText> UserInputTextBox;

    UPROPERTY(meta = (BindWidget))
    TObjectPtr<UButton> SendMessageButton;

    UPROPERTY(meta = (BindWidget))
    TObjectPtr<UButton> DumpChatButton;

    UPROPERTY(meta = (BindWidget))
    TObjectPtr<UButton> ClearChatButton;

    UPROPERTY(meta = (BindWidget))
    TObjectPtr<UButton> AttachImageButton;

    UPROPERTY(EditDefaultsOnly, BlueprintReadOnly, Category = "UI")
    TSubclassOf<UChatMessageWidget> ChatMessageWidgetClass;

    UPROPERTY(meta = (BindWidget))
    TObjectPtr<UGridPanel> ServiceContainer;

    UPROPERTY(EditDefaultsOnly, BlueprintReadOnly, Category = "UI")
    TSubclassOf<UServiceWidget> ServiceWidgetClass;

    UPROPERTY(EditDefaultsOnly, BlueprintReadOnly, Category = "OpenAI")
    TArray<TSubclassOf<UBaseService>> Services;

    UPROPERTY(meta = (BindWidget))
    TObjectPtr<UTextBlock> ServicesWarningTextBlock;

    UPROPERTY(meta = (BindWidget))
    TObjectPtr<UAttachedFilesContainerWidget> AttachedFilesContainer;

    virtual void NativeConstruct() override;

private:
    UPROPERTY()
    TObjectPtr<UChatGPT> ChatGPT;

    UPROPERTY()
    TArray<TObjectPtr<UChatMessageWidget>> ChatWidgets;

    UPROPERTY()
    TObjectPtr<USaveSettings> SaveSettings;

    UPROPERTY()
    TArray<TObjectPtr<UServiceWidget>> ServiceWidgets;

private:
    UFUNCTION()
    void OnSendMessage();

    UFUNCTION()
    void OnDumpChat();

    UFUNCTION()
    void OnClearChat();

    UFUNCTION()
    void OnTextCommitted(const FText& Text, ETextCommit::Type CommitMethod);

    UFUNCTION()
    void OnTextChanged(const FText& Text);

    UFUNCTION()
    void OnModelSelectionChanged(FString SelectedItem, ESelectInfo::Type SelectionType);

    UFUNCTION()
    void OnImageSelected();

    void InitChatGPT();
    void InitModelsComboBox();

    void OnRequestCompleted();
    void OnRequestUpdated(const FMessage& Message, bool WasError);

    void CreateMessageWidget(const FMessage& Message);
    void EnableControls(bool Enabled);

    FString GenerateFilePath() const;

    void UpdateVisionElements();
    void UpdateSendMessageButton();
    void OnFileRemoved();

private:
    OpenAI::ServiceSecrets ServiceSecrets;

    void CreateServiceWidgets();
    void OnServiceEnabled(bool IsEnabled, const TSubclassOf<UBaseService>& ServiceClass, const TObjectPtr<UServiceWidget>& ServiceWidget);
};