Enha: popups from the main window
no longer user has to go to the props table to get a pleasant popup to choose an option
This commit is contained in:
Grail Finder
315
popups.go
Normal file
315
popups.go
Normal file
@@ -0,0 +1,315 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"slices"
|
||||
"strings"
|
||||
|
||||
"github.com/gdamore/tcell/v2"
|
||||
"github.com/rivo/tview"
|
||||
)
|
||||
|
||||
// showModelSelectionPopup creates a modal popup to select a model
|
||||
func showModelSelectionPopup() {
|
||||
// Helper function to get model list for a given API
|
||||
getModelListForAPI := func(api string) []string {
|
||||
if strings.Contains(api, "api.deepseek.com/") {
|
||||
return []string{"deepseek-chat", "deepseek-reasoner"}
|
||||
} else if strings.Contains(api, "openrouter.ai") {
|
||||
return ORFreeModels
|
||||
}
|
||||
// Assume local llama.cpp
|
||||
refreshLocalModelsIfEmpty()
|
||||
localModelsMu.RLock()
|
||||
defer localModelsMu.RUnlock()
|
||||
return LocalModels
|
||||
}
|
||||
// Get the current model list based on the API
|
||||
modelList := getModelListForAPI(cfg.CurrentAPI)
|
||||
// Check for empty options list
|
||||
if len(modelList) == 0 {
|
||||
logger.Warn("empty model list for", "api", cfg.CurrentAPI, "localModelsLen", len(LocalModels), "orModelsLen", len(ORFreeModels))
|
||||
message := "No models available for selection"
|
||||
if strings.Contains(cfg.CurrentAPI, "openrouter.ai") {
|
||||
message = "No OpenRouter models available. Check token and connection."
|
||||
} else if strings.Contains(cfg.CurrentAPI, "api.deepseek.com") {
|
||||
message = "DeepSeek models should be available. Please report bug."
|
||||
} else {
|
||||
message = "No llama.cpp models loaded. Ensure llama.cpp server is running with models."
|
||||
}
|
||||
if err := notifyUser("Empty list", message); err != nil {
|
||||
logger.Error("failed to send notification", "error", err)
|
||||
}
|
||||
return
|
||||
}
|
||||
// Create a list primitive
|
||||
modelListWidget := tview.NewList().ShowSecondaryText(false).
|
||||
SetSelectedBackgroundColor(tcell.ColorGray)
|
||||
modelListWidget.SetTitle("Select Model").SetBorder(true)
|
||||
// Find the current model index to set as selected
|
||||
currentModelIndex := -1
|
||||
for i, model := range modelList {
|
||||
if model == chatBody.Model {
|
||||
currentModelIndex = i
|
||||
}
|
||||
modelListWidget.AddItem(model, "", 0, nil)
|
||||
}
|
||||
// Set the current selection if found
|
||||
if currentModelIndex != -1 {
|
||||
modelListWidget.SetCurrentItem(currentModelIndex)
|
||||
}
|
||||
modelListWidget.SetSelectedFunc(func(index int, mainText string, secondaryText string, shortcut rune) {
|
||||
// Update the model in both chatBody and config
|
||||
chatBody.Model = mainText
|
||||
cfg.CurrentModel = chatBody.Model
|
||||
// Remove the popup page
|
||||
pages.RemovePage("modelSelectionPopup")
|
||||
// Update the status line to reflect the change
|
||||
updateStatusLine()
|
||||
})
|
||||
modelListWidget.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
|
||||
if event.Key() == tcell.KeyEscape {
|
||||
pages.RemovePage("modelSelectionPopup")
|
||||
return nil
|
||||
}
|
||||
return event
|
||||
})
|
||||
modal := func(p tview.Primitive, width, height int) tview.Primitive {
|
||||
return tview.NewFlex().
|
||||
AddItem(nil, 0, 1, false).
|
||||
AddItem(tview.NewFlex().SetDirection(tview.FlexRow).
|
||||
AddItem(nil, 0, 1, false).
|
||||
AddItem(p, height, 1, true).
|
||||
AddItem(nil, 0, 1, false), width, 1, true).
|
||||
AddItem(nil, 0, 1, false)
|
||||
}
|
||||
// Add modal page and make it visible
|
||||
pages.AddPage("modelSelectionPopup", modal(modelListWidget, 80, 20), true, true)
|
||||
app.SetFocus(modelListWidget)
|
||||
}
|
||||
|
||||
// showAPILinkSelectionPopup creates a modal popup to select an API link
|
||||
func showAPILinkSelectionPopup() {
|
||||
// Prepare API links dropdown - ensure current API is in the list, avoid duplicates
|
||||
apiLinks := make([]string, 0, len(cfg.ApiLinks)+1)
|
||||
// Add current API first if it's not already in ApiLinks
|
||||
foundCurrentAPI := false
|
||||
for _, api := range cfg.ApiLinks {
|
||||
if api == cfg.CurrentAPI {
|
||||
foundCurrentAPI = true
|
||||
}
|
||||
apiLinks = append(apiLinks, api)
|
||||
}
|
||||
// If current API is not in the list, add it at the beginning
|
||||
if !foundCurrentAPI {
|
||||
apiLinks = make([]string, 0, len(cfg.ApiLinks)+1)
|
||||
apiLinks = append(apiLinks, cfg.CurrentAPI)
|
||||
apiLinks = append(apiLinks, cfg.ApiLinks...)
|
||||
}
|
||||
// Check for empty options list
|
||||
if len(apiLinks) == 0 {
|
||||
logger.Warn("no API links available for selection")
|
||||
message := "No API links available. Please configure API links in your config file."
|
||||
if err := notifyUser("Empty list", message); err != nil {
|
||||
logger.Error("failed to send notification", "error", err)
|
||||
}
|
||||
return
|
||||
}
|
||||
// Create a list primitive
|
||||
apiListWidget := tview.NewList().ShowSecondaryText(false).
|
||||
SetSelectedBackgroundColor(tcell.ColorGray)
|
||||
apiListWidget.SetTitle("Select API Link").SetBorder(true)
|
||||
// Find the current API index to set as selected
|
||||
currentAPIIndex := -1
|
||||
for i, api := range apiLinks {
|
||||
if api == cfg.CurrentAPI {
|
||||
currentAPIIndex = i
|
||||
}
|
||||
apiListWidget.AddItem(api, "", 0, nil)
|
||||
}
|
||||
// Set the current selection if found
|
||||
if currentAPIIndex != -1 {
|
||||
apiListWidget.SetCurrentItem(currentAPIIndex)
|
||||
}
|
||||
apiListWidget.SetSelectedFunc(func(index int, mainText string, secondaryText string, shortcut rune) {
|
||||
// Update the API in config
|
||||
cfg.CurrentAPI = mainText
|
||||
// Update model list based on new API
|
||||
// Helper function to get model list for a given API (same as in props_table.go)
|
||||
getModelListForAPI := func(api string) []string {
|
||||
if strings.Contains(api, "api.deepseek.com/") {
|
||||
return []string{"deepseek-chat", "deepseek-reasoner"}
|
||||
} else if strings.Contains(api, "openrouter.ai") {
|
||||
return ORFreeModels
|
||||
}
|
||||
// Assume local llama.cpp
|
||||
refreshLocalModelsIfEmpty()
|
||||
localModelsMu.RLock()
|
||||
defer localModelsMu.RUnlock()
|
||||
return LocalModels
|
||||
}
|
||||
newModelList := getModelListForAPI(cfg.CurrentAPI)
|
||||
// Ensure chatBody.Model is in the new list; if not, set to first available model
|
||||
if len(newModelList) > 0 && !slices.Contains(newModelList, chatBody.Model) {
|
||||
chatBody.Model = newModelList[0]
|
||||
cfg.CurrentModel = chatBody.Model
|
||||
}
|
||||
// Remove the popup page
|
||||
pages.RemovePage("apiLinkSelectionPopup")
|
||||
// Update the parser and status line to reflect the change
|
||||
choseChunkParser()
|
||||
updateStatusLine()
|
||||
})
|
||||
apiListWidget.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
|
||||
if event.Key() == tcell.KeyEscape {
|
||||
pages.RemovePage("apiLinkSelectionPopup")
|
||||
return nil
|
||||
}
|
||||
return event
|
||||
})
|
||||
modal := func(p tview.Primitive, width, height int) tview.Primitive {
|
||||
return tview.NewFlex().
|
||||
AddItem(nil, 0, 1, false).
|
||||
AddItem(tview.NewFlex().SetDirection(tview.FlexRow).
|
||||
AddItem(nil, 0, 1, false).
|
||||
AddItem(p, height, 1, true).
|
||||
AddItem(nil, 0, 1, false), width, 1, true).
|
||||
AddItem(nil, 0, 1, false)
|
||||
}
|
||||
// Add modal page and make it visible
|
||||
pages.AddPage("apiLinkSelectionPopup", modal(apiListWidget, 80, 20), true, true)
|
||||
app.SetFocus(apiListWidget)
|
||||
}
|
||||
|
||||
// showUserRoleSelectionPopup creates a modal popup to select a user role
|
||||
func showUserRoleSelectionPopup() {
|
||||
// Get the list of available roles
|
||||
roles := listRolesWithUser()
|
||||
// Check for empty options list
|
||||
if len(roles) == 0 {
|
||||
logger.Warn("no roles available for selection")
|
||||
message := "No roles available for selection."
|
||||
if err := notifyUser("Empty list", message); err != nil {
|
||||
logger.Error("failed to send notification", "error", err)
|
||||
}
|
||||
return
|
||||
}
|
||||
// Create a list primitive
|
||||
roleListWidget := tview.NewList().ShowSecondaryText(false).
|
||||
SetSelectedBackgroundColor(tcell.ColorGray)
|
||||
roleListWidget.SetTitle("Select User Role").SetBorder(true)
|
||||
// Find the current role index to set as selected
|
||||
currentRole := cfg.UserRole
|
||||
if cfg.WriteNextMsgAs != "" {
|
||||
currentRole = cfg.WriteNextMsgAs
|
||||
}
|
||||
currentRoleIndex := -1
|
||||
for i, role := range roles {
|
||||
if strings.EqualFold(role, currentRole) {
|
||||
currentRoleIndex = i
|
||||
}
|
||||
roleListWidget.AddItem(role, "", 0, nil)
|
||||
}
|
||||
// Set the current selection if found
|
||||
if currentRoleIndex != -1 {
|
||||
roleListWidget.SetCurrentItem(currentRoleIndex)
|
||||
}
|
||||
roleListWidget.SetSelectedFunc(func(index int, mainText string, secondaryText string, shortcut rune) {
|
||||
// Update the user role in config
|
||||
cfg.WriteNextMsgAs = mainText
|
||||
// role got switch, update textview with character specific context for user
|
||||
filtered := filterMessagesForCharacter(chatBody.Messages, mainText)
|
||||
textView.SetText(chatToText(filtered, cfg.ShowSys))
|
||||
// Remove the popup page
|
||||
pages.RemovePage("userRoleSelectionPopup")
|
||||
// Update the status line to reflect the change
|
||||
updateStatusLine()
|
||||
colorText()
|
||||
})
|
||||
roleListWidget.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
|
||||
if event.Key() == tcell.KeyEscape {
|
||||
pages.RemovePage("userRoleSelectionPopup")
|
||||
return nil
|
||||
}
|
||||
return event
|
||||
})
|
||||
modal := func(p tview.Primitive, width, height int) tview.Primitive {
|
||||
return tview.NewFlex().
|
||||
AddItem(nil, 0, 1, false).
|
||||
AddItem(tview.NewFlex().SetDirection(tview.FlexRow).
|
||||
AddItem(nil, 0, 1, false).
|
||||
AddItem(p, height, 1, true).
|
||||
AddItem(nil, 0, 1, false), width, 1, true).
|
||||
AddItem(nil, 0, 1, false)
|
||||
}
|
||||
// Add modal page and make it visible
|
||||
pages.AddPage("userRoleSelectionPopup", modal(roleListWidget, 80, 20), true, true)
|
||||
app.SetFocus(roleListWidget)
|
||||
}
|
||||
|
||||
// showBotRoleSelectionPopup creates a modal popup to select a bot role
|
||||
func showBotRoleSelectionPopup() {
|
||||
// Get the list of available roles
|
||||
roles := listChatRoles()
|
||||
if len(roles) == 0 {
|
||||
logger.Warn("empty roles in chat")
|
||||
}
|
||||
if !strInSlice(cfg.AssistantRole, roles) {
|
||||
roles = append(roles, cfg.AssistantRole)
|
||||
}
|
||||
// Check for empty options list
|
||||
if len(roles) == 0 {
|
||||
logger.Warn("no roles available for selection")
|
||||
message := "No roles available for selection."
|
||||
if err := notifyUser("Empty list", message); err != nil {
|
||||
logger.Error("failed to send notification", "error", err)
|
||||
}
|
||||
return
|
||||
}
|
||||
// Create a list primitive
|
||||
roleListWidget := tview.NewList().ShowSecondaryText(false).
|
||||
SetSelectedBackgroundColor(tcell.ColorGray)
|
||||
roleListWidget.SetTitle("Select Bot Role").SetBorder(true)
|
||||
// Find the current role index to set as selected
|
||||
currentRole := cfg.AssistantRole
|
||||
if cfg.WriteNextMsgAsCompletionAgent != "" {
|
||||
currentRole = cfg.WriteNextMsgAsCompletionAgent
|
||||
}
|
||||
currentRoleIndex := -1
|
||||
for i, role := range roles {
|
||||
if strings.EqualFold(role, currentRole) {
|
||||
currentRoleIndex = i
|
||||
}
|
||||
roleListWidget.AddItem(role, "", 0, nil)
|
||||
}
|
||||
// Set the current selection if found
|
||||
if currentRoleIndex != -1 {
|
||||
roleListWidget.SetCurrentItem(currentRoleIndex)
|
||||
}
|
||||
roleListWidget.SetSelectedFunc(func(index int, mainText string, secondaryText string, shortcut rune) {
|
||||
// Update the bot role in config
|
||||
cfg.WriteNextMsgAsCompletionAgent = mainText
|
||||
// Remove the popup page
|
||||
pages.RemovePage("botRoleSelectionPopup")
|
||||
// Update the status line to reflect the change
|
||||
updateStatusLine()
|
||||
})
|
||||
roleListWidget.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
|
||||
if event.Key() == tcell.KeyEscape {
|
||||
pages.RemovePage("botRoleSelectionPopup")
|
||||
return nil
|
||||
}
|
||||
return event
|
||||
})
|
||||
modal := func(p tview.Primitive, width, height int) tview.Primitive {
|
||||
return tview.NewFlex().
|
||||
AddItem(nil, 0, 1, false).
|
||||
AddItem(tview.NewFlex().SetDirection(tview.FlexRow).
|
||||
AddItem(nil, 0, 1, false).
|
||||
AddItem(p, height, 1, true).
|
||||
AddItem(nil, 0, 1, false), width, 1, true).
|
||||
AddItem(nil, 0, 1, false)
|
||||
}
|
||||
// Add modal page and make it visible
|
||||
pages.AddPage("botRoleSelectionPopup", modal(roleListWidget, 80, 20), true, true)
|
||||
app.SetFocus(roleListWidget)
|
||||
}
|
||||
Reference in New Issue
Block a user