add file icon for each folder and tokens count for each file in view

Author: yuchenliu15

internal/model/view.go

@@ -4,9 +4,11 @@ import (
 	"fmt"
 	"path/filepath"
 	"strings"
+	"os"
 
 	"github.com/charmbracelet/lipgloss"
 	"github.com/epilande/codegrab/internal/ui"
+	"github.com/epilande/codegrab/internal/utils"
 )
 
 func (m Model) View() string {
@@ -227,12 +229,20 @@ func (m *Model) refreshViewportContent() {
 		rawSuffix := ""
 		if node.IsDir {
 			if count := dirSelectedCounts[node.Path]; count > 0 {
-				rawSuffix = fmt.Sprintf(" [%d]", count)
+				rawSuffix = fmt.Sprintf(" [%d 📄]", count)
+			}
+		} else {
+			if node.IsDependency {
+				rawSuffix += " [dep]"
+			}
+			if ok, err := utils.IsTextFile(node.Path); ok && err == nil {
+				if contentBytes, err := os.ReadFile(node.Path); err == nil {
+					content := string(contentBytes)
+					tokensEstimate := utils.EstimateTokens(content)
+					rawSuffix += fmt.Sprintf(" [%d tokens]", tokensEstimate)
+				}
 			}
-		} else if node.IsDependency {
-			rawSuffix = " [dep]"
 		}
-
 		isCursorLine := i == m.cursor
 
 		rendered := ui.StyleFileLine(