@using AIStudio.Chat
@inherits MSGComponentBase

<MudDialog>
    <DialogContent>
        <MudJustifiedText Class="mb-3" Typo="Typo.body1">
            @T("Create your custom chat template to tailor the LLM's behavior for specific tasks or domains. Define a custom system prompt and provide an example conversation to design an AI experience perfectly suited to your requirements.")
        </MudJustifiedText>

        <MudJustifiedText Class="mb-3" Typo="Typo.body1">
            @T("The name of the chat template is mandatory. Each chat template must have a unique name.")
        </MudJustifiedText>
        
        <MudForm @ref="@this.form" @bind-IsValid="@this.dataIsValid" @bind-Errors="@this.dataIssues">
            @* ReSharper disable once CSharpWarnings::CS8974 *@
            <MudTextField
                T="string"
                @bind-Text="@this.DataName"
                Label="@T("Chat Template Name")"
                Class="mb-3"
                Immediate="@true"
                MaxLength="40"
                Counter="40"
                Adornment="Adornment.Start"
                AdornmentIcon="@Icons.Material.Filled.Badge"
                AdornmentColor="Color.Info"
                Validation="@this.ValidateName"
                Variant="Variant.Outlined"
                UserAttributes="@SPELLCHECK_ATTRIBUTES"
            />
            
            <MudText Typo="Typo.h6" Class="mb-3 mt-3">
                @T("System Prompt")
            </MudText>

            <MudTextField
                T="string"
                @bind-Text="@this.DataSystemPrompt"
                AdornmentIcon="@Icons.Material.Filled.ListAlt"
                Adornment="Adornment.Start"
                Immediate="@true"
                Label="@T("What system prompt do you want to use?")"
                Variant="Variant.Outlined"
                Lines="6"
                MaxLines="12"
                AutoGrow="@true"
                Class="mb-3"
                UserAttributes="@SPELLCHECK_ATTRIBUTES"
                HelperText="@T("Tell the AI your system prompt.")"
            />
            
            <MudJustifiedText Class="mb-3" Typo="Typo.body1">
                @T("Are you unsure which system prompt to use? You might start with the default system prompt that AI Studio uses for all chats.")
            </MudJustifiedText>
            <MudButton Class="mb-3" Color="Color.Default" OnClick="@this.UseDefaultSystemPrompt" StartIcon="@Icons.Material.Filled.ListAlt" Variant="Variant.Filled">
                @T("Use the default system prompt")
            </MudButton>
            
            <MudText Typo="Typo.h6" Class="mb-3 mt-6">
                @T("Predefined User Input")
            </MudText>
            <MudJustifiedText Class="mb-3" Typo="Typo.body1">
                @T("You might want to predefine a first message that will be copied into the user prompt, when you use this chat template. This message could for example be a blueprint for a structured message that this chat template is defined to work with.")
            </MudJustifiedText>
            <MudTextField
                T="string"
                @bind-Text="@this.PredefinedUserPrompt"
                AdornmentIcon="@Icons.Material.Filled.ListAlt"
                Adornment="Adornment.Start"
                Immediate="@true"
                Label="@T("What predefined user input do you want to use?")"
                Variant="Variant.Outlined"
                Lines="4"
                MaxLines="12"
                AutoGrow="@true"
                Class="mb-3"
                UserAttributes="@SPELLCHECK_ATTRIBUTES"
                HelperText="@T("Tell the AI your predefined user input.")"
            />
            
            <MudText Typo="Typo.h6" Class="mb-3 mt-6">
                @T("Profile Usage")
            </MudText>
            <MudJustifiedText Class="mb-3" Typo="Typo.body1">
                @T("Using some chat templates in tandem with profiles might cause issues. Therefore, you might prohibit the usage of profiles here.")
            </MudJustifiedText>
            <MudTextSwitch @bind-Value="@this.AllowProfileUsage" Color="Color.Primary" Label="@T("Allow the use of profiles together with this chat template?")" LabelOn="@T("Yes, allow profiles when using this template")" LabelOff="@T("No, prohibit profile use for this template")" />
            
            <MudText Typo="Typo.h6" Class="mb-3 mt-6">
                @T("Example Conversation")
            </MudText>
            <MudJustifiedText Typo="Typo.body1" Class="mb-3">
                @T("Add messages of an example conversation (user prompt followed by assistant prompt) to demonstrate the desired interaction pattern. These examples help the AI understand your expectations by showing it the correct format, style, and content of responses before it receives actual user inputs.")
            </MudJustifiedText>
            <MudTable Items="@this.dataExampleConversation" FixedHeader="true" Hover="true" Class="mt-3 mb-3" CanCancelEdit="true" CancelEditTooltip="@T("Cancel")" CommitEditTooltip="@T("Commit Changes")" Outlined="true" RowEditCancel="@this.ResetItem" RowEditPreview="@this.BackupItem" EditTrigger="TableEditTrigger.RowClick" IsEditRowSwitchingBlocked="false" RowEditCommit="@this.CommitInlineEdit">
                <ColGroup>
                    <col style="width: 10em;" />
                    <col/>
                    <col style="width: 10em;" />
                </ColGroup>
                <HeaderContent>
                    <MudTh>@T("Role")</MudTh>
                    <MudTh>@T("Entry")</MudTh>
                    <MudTh>@T("Actions")</MudTh>
                </HeaderContent>
                <RowTemplate>
                    <MudTd>
                        @context.Role.ToChatTemplateName()
                    </MudTd>
                    <MudTd>
                        @switch(context.Content)
                        {
                            case ContentText textContent:
                                <MudTextField AutoGrow="true" Value="@textContent.Text" Placeholder="@T("Enter a message")" ReadOnly="true" Variant="Variant.Text" Validation="@this.ValidateExampleTextMessage"/>
                                break;
                                
                            case ContentImage { SourceType: ContentImageSource.URL or ContentImageSource.LOCAL_PATH } imageContent:
                                <MudImage Src="@imageContent.Source" Alt="@T("Image content")" Fluid="true" />
                                break;
                            
                            default:
                                @T("Unsupported content type")
                                break;
                        }
                    </MudTd>
                    <MudTd>
                        @if (!this.isInlineEditOnGoing)
                        {
                            <MudStack Row="true" Class="mb-2 mt-2" Wrap="Wrap.Wrap">
                                <MudTooltip Text="@T("Add a new message below")">
                                    <MudIconButton Color="Color.Primary" Icon="@Icons.Material.Filled.Add" OnClick="@(() => this.AddMessageBelow(context))"/>
                                </MudTooltip>
                                <MudIconButton Color="Color.Error" Icon="@Icons.Material.Filled.Delete" OnClick="@(() => this.RemoveMessage(context))"/>
                            </MudStack>
                        }
                    </MudTd>
                </RowTemplate>
                <RowEditingTemplate>
                    <MudTd>
                        <MudSelect Label="@T("Role")" @bind-Value="@context.Role" Required="true">
                            @foreach (var role in ChatRoles.ChatTemplateRoles())
                            {
                                <MudSelectItem Value="@role">
                                    @role.ToChatTemplateName()
                                </MudSelectItem>
                            }
                        </MudSelect>
                    </MudTd>
                    <MudTd>
                        @switch(context.Content)
                        {
                            case ContentText textContent:
                                <MudTextField AutoGrow="true" @bind-Value="@textContent.Text" Label="@T("The message")" Required="true" Immediate="true" Placeholder="@T("Enter a message")"/>
                                break;
                                
                            default:
                                <MudText Typo="Typo.body2">
                                    @T("Only text content is supported in the editing mode yet.")
                                </MudText>
                                break;
                        }
                    </MudTd>
                </RowEditingTemplate>
                <PagerContent>
                    <MudTablePager PageSizeOptions="[10,20,50,100]" RowsPerPageString="@T("Messages per page")" />
                </PagerContent>
            </MudTable>
        </MudForm>
        
        @if (!this.isInlineEditOnGoing)
        {
            <MudButton Class="mb-6" Color="Color.Primary" OnClick="@this.AddMessageToEnd" StartIcon="@Icons.Material.Filled.Add" Variant="Variant.Filled">
                @T("Add a message")
            </MudButton>
        }

        <Issues IssuesData="@this.dataIssues"/>
    </DialogContent>
    <DialogActions>
        <MudButton OnClick="@this.Cancel" Variant="Variant.Filled">
            @T("Cancel")
        </MudButton>
        
        @if (!this.isInlineEditOnGoing)
        {
            <MudButton OnClick="@this.Store" Variant="Variant.Filled" Color="Color.Primary">
                @if (this.IsEditing)
                {
                    @T("Update")
                }
                else
                {
                    @T("Add")
                }
            </MudButton>
        }
    </DialogActions>
</MudDialog>