64 lines
1.8 KiB
C
64 lines
1.8 KiB
C
|
// OpenAI, Copyright LifeEXE. All Rights Reserved.
|
||
|
|
|
||
|
|
#pragma once
|
||
|
|
|
||
|
|
#include "CoreMinimal.h"
|
||
|
|
#include "EditorUtilityWidget.h"
|
||
|
|
#include "Provider/CommonTypes.h"
|
||
|
|
#include "FuncLib/ModelTypes.h"
|
||
|
|
#include "ChatMessageWidget.generated.h"
|
||
|
|
|
||
|
|
class UBorder;
|
||
|
|
class UMultiLineEditableText;
|
||
|
|
class UTextBlock;
|
||
|
|
class UWidgetSwitcher;
|
||
|
|
class UHorizontalBox;
|
||
|
|
class UImagePreviewWidget;
|
||
|
|
|
||
|
|
UCLASS()
|
||
|
|
class OPENAIEDITOR_API UChatMessageWidget : public UEditorUtilityWidget
|
||
|
|
{
|
||
|
|
GENERATED_BODY()
|
||
|
|
|
||
|
|
protected:
|
||
|
|
UPROPERTY(meta = (BindWidget))
|
||
|
|
TObjectPtr<UBorder> MainBorder;
|
||
|
|
|
||
|
|
UPROPERTY(meta = (BindWidget))
|
||
|
|
TObjectPtr<UTextBlock> RoleTextBox;
|
||
|
|
|
||
|
|
UPROPERTY(meta = (BindWidget))
|
||
|
|
TObjectPtr<UMultiLineEditableText> MessageText;
|
||
|
|
|
||
|
|
UPROPERTY(meta = (BindWidget))
|
||
|
|
TObjectPtr<UTextBlock> TimeTextBox;
|
||
|
|
|
||
|
|
UPROPERTY(meta = (BindWidget))
|
||
|
|
TObjectPtr<UWidgetSwitcher> WidgetSwitcher; // preloader -> text;
|
||
|
|
|
||
|
|
UPROPERTY(meta = (BindWidget))
|
||
|
|
TObjectPtr<UHorizontalBox> ImageHorizontalBox;
|
||
|
|
|
||
|
|
UPROPERTY(EditDefaultsOnly, BlueprintReadWrite, Category = "UI")
|
||
|
|
TMap<ERole, FLinearColor> ColorMap{
|
||
|
|
{ERole::Assistant, FColor::FromHex("444654FF")},
|
||
|
|
{ERole::User, FColor::FromHex("343541FF")},
|
||
|
|
{ERole::System, FColor::FromHex("595a68FF")},
|
||
|
|
};
|
||
|
|
|
||
|
|
UPROPERTY(EditDefaultsOnly, BlueprintReadWrite, Category = "UI")
|
||
|
|
FLinearColor DefaultColor{FColor::FromHex("0e0e0eFF")};
|
||
|
|
|
||
|
|
UPROPERTY(EditDefaultsOnly, BlueprintReadWrite, Category = "UI")
|
||
|
|
FLinearColor ErrorColor{FColor::FromHex("41343dFF")};
|
||
|
|
|
||
|
|
UPROPERTY(EditDefaultsOnly, BlueprintReadOnly, Category = "UI")
|
||
|
|
TSubclassOf<UImagePreviewWidget> ImagePreviewWidgetClass;
|
||
|
|
|
||
|
|
public:
|
||
|
|
void SetMessage(const FMessage& Message, bool WasError = false);
|
||
|
|
|
||
|
|
private:
|
||
|
|
TArray<TObjectPtr<UImagePreviewWidget>> ImageWidgets;
|
||
|
|
};
|