From 43b0fe3739b27deae8193daf1145f42fd616b1f1 Mon Sep 17 00:00:00 2001
From: Grail Finder <wohilas@gmail.com>
Date: Mon, 16 Feb 2026 19:08:16 +0300
Subject: Feat: image preview for filepicker

---
 config/config.go |  1 +
 props_table.go   |  3 +++
 tables.go        | 66 ++++++++++++++++++++++++++++++++++++++++++++++++++++++--
 3 files changed, 68 insertions(+), 2 deletions(-)

diff --git a/config/config.go b/config/config.go
index bed24f6..d5c059f 100644
--- a/config/config.go
+++ b/config/config.go
@@ -30,6 +30,7 @@ type Config struct {
 	DBPATH                        string `toml:"DBPATH"`
 	FilePickerDir                 string `toml:"FilePickerDir"`
 	FilePickerExts                string `toml:"FilePickerExts"`
+	ImagePreview                  bool   `toml:"ImagePreview"`
 	EnableMouse                   bool   `toml:"EnableMouse"`
 	// embeddings
 	RAGEnabled bool   `toml:"RAGEnabled"`
diff --git a/props_table.go b/props_table.go
index a7ad067..ac47f49 100644
--- a/props_table.go
+++ b/props_table.go
@@ -135,6 +135,9 @@ func makePropsTable(props map[string]float32) *tview.Table {
 		// Reconfigure the app's mouse setting
 		app.EnableMouse(cfg.EnableMouse)
 	})
+	addCheckboxRow("Image Preview (file picker)", cfg.ImagePreview, func(checked bool) {
+		cfg.ImagePreview = checked
+	})
 	addCheckboxRow("Auto turn (for cards with many chars)", cfg.AutoTurn, func(checked bool) {
 		cfg.AutoTurn = checked
 	})
diff --git a/tables.go b/tables.go
index eb3d315..e4066c8 100644
--- a/tables.go
+++ b/tables.go
@@ -2,6 +2,7 @@ package main
 
 import (
 	"fmt"
+	"image"
 	"os"
 	"path"
 	"strings"
@@ -823,9 +824,25 @@ func makeFilePicker() *tview.Flex {
 	statusView := tview.NewTextView()
 	statusView.SetBorder(true).SetTitle("Selected File").SetTitleAlign(tview.AlignLeft)
 	statusView.SetTextColor(tcell.ColorYellow)
-	// Layout - only include list view and status view
+	// Image preview pane
+	var imgPreview *tview.Image
+	if cfg.ImagePreview {
+		imgPreview = tview.NewImage()
+		imgPreview.SetBorder(true).SetTitle("Preview").SetTitleAlign(tview.AlignLeft)
+	}
+	// Horizontal flex for list + preview
+	var hFlex *tview.Flex
+	if cfg.ImagePreview && imgPreview != nil {
+		hFlex = tview.NewFlex().SetDirection(tview.FlexColumn).
+			AddItem(listView, 0, 3, true).
+			AddItem(imgPreview, 0, 2, false)
+	} else {
+		hFlex = tview.NewFlex().SetDirection(tview.FlexColumn).
+			AddItem(listView, 0, 1, true)
+	}
+	// Main vertical flex
 	flex := tview.NewFlex().SetDirection(tview.FlexRow)
-	flex.AddItem(listView, 0, 3, true)
+	flex.AddItem(hFlex, 0, 3, true)
 	flex.AddItem(statusView, 3, 0, false)
 	// Refresh the file list
 	var refreshList func(string)
@@ -846,6 +863,7 @@ func makeFilePicker() *tview.Flex {
 				// We're at the root ("/") and trying to go up, just don't add the parent item
 			} else {
 				listView.AddItem("../ [gray](Parent Directory)[-]", "", 'p', func() {
+					imgPreview.SetImage(nil)
 					refreshList(parentDir)
 					dirStack = append(dirStack, parentDir)
 					currentStackPos = len(dirStack) - 1
@@ -869,6 +887,7 @@ func makeFilePicker() *tview.Flex {
 				// Capture the directory name for the closure to avoid loop variable issues
 				dirName := name
 				listView.AddItem(dirName+"/ [gray](Directory)[-]", "", 0, func() {
+					imgPreview.SetImage(nil)
 					newDir := path.Join(dir, dirName)
 					refreshList(newDir)
 					dirStack = append(dirStack, newDir)
@@ -898,6 +917,43 @@ func makeFilePicker() *tview.Flex {
 	}
 	// Initialize the file list
 	refreshList(startDir)
+	// Update image preview when selection changes
+	if cfg.ImagePreview && imgPreview != nil {
+		listView.SetChangedFunc(func(index int, mainText, secondaryText string, rune rune) {
+			itemText, _ := listView.GetItemText(index)
+			if strings.HasPrefix(itemText, "Exit file picker") || strings.HasPrefix(itemText, "../") {
+				imgPreview.SetImage(nil)
+				return
+			}
+			actualItemName := itemText
+			if bracketPos := strings.Index(itemText, " ["); bracketPos != -1 {
+				actualItemName = itemText[:bracketPos]
+			}
+			if strings.HasSuffix(actualItemName, "/") {
+				imgPreview.SetImage(nil)
+				return
+			}
+			if !isImageFile(actualItemName) {
+				imgPreview.SetImage(nil)
+				return
+			}
+			filePath := path.Join(currentDisplayDir, actualItemName)
+			go func() {
+				file, err := os.Open(filePath)
+				if err != nil {
+					app.QueueUpdate(func() { imgPreview.SetImage(nil) })
+					return
+				}
+				defer file.Close()
+				img, _, err := image.Decode(file)
+				if err != nil {
+					app.QueueUpdate(func() { imgPreview.SetImage(nil) })
+					return
+				}
+				app.QueueUpdate(func() { imgPreview.SetImage(img) })
+			}()
+		})
+	}
 	// Set up keyboard navigation
 	flex.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
 		switch event.Key() {
@@ -905,6 +961,9 @@ func makeFilePicker() *tview.Flex {
 			pages.RemovePage(filePickerPage)
 			return nil
 		case tcell.KeyBackspace2: // Backspace to go to parent directory
+			if cfg.ImagePreview && imgPreview != nil {
+				imgPreview.SetImage(nil)
+			}
 			if currentStackPos > 0 {
 				currentStackPos--
 				prevDir := dirStack[currentStackPos]
@@ -954,6 +1013,9 @@ func makeFilePicker() *tview.Flex {
 					}
 					// Navigate to the selected directory
 					logger.Info("going to the dir", "dir", targetDir)
+					if cfg.ImagePreview && imgPreview != nil {
+						imgPreview.SetImage(nil)
+					}
 					refreshList(targetDir)
 					dirStack = append(dirStack, targetDir)
 					currentStackPos = len(dirStack) - 1
-- 
cgit v1.2.3