Feature: Display GPT token count in file preview panel.

Author: deanthecoder

CodeIngest.Desktop/MainViewModel.cs

@@ -151,9 +151,15 @@ public int? PreviewFileCount
     public long? PreviewFileSize
     {
         get => m_previewFileSize;
-        set => SetField(ref m_previewFileSize, value);
+        set
+        {
+            if (SetField(ref m_previewFileSize, value))
+                OnPropertyChanged(nameof(PreviewTokenCount));
+        }
     }
 
+    public int PreviewTokenCount => (int)((PreviewFileSize ?? 0) / 3.8);
+
     public bool IsGeneratingPreview
     {
         get => m_isGeneratingPreview;
@@ -243,47 +249,51 @@ private IngestOptions GetIngestOptions()
     private void OnFolderSelectionChanged(object sender, EventArgs e) =>
         InvalidatePreviewStats();
 
-    private void InvalidatePreviewStats() =>
+    private void InvalidatePreviewStats()
+    {
+        IsGeneratingPreview = true;
+        PreviewFileCount = null;
+        PreviewFileSize = null;
         m_folderSelectionConsolidator.Invoke();
+    }
 
     private void RefreshPredictedSize()
     {
-        m_backgroundRefreshProgress?.Cancel();
-        
-        var selectedFolders = Root.GetSelectedItems().ToArray();
-        if (selectedFolders.Length == 0)
+        try
         {
-            PreviewFileCount = 0;
-            PreviewFileSize = 0;
-            return; // Nothing to do.
-        }
+            m_backgroundRefreshProgress?.Cancel();
 
-        var options = GetIngestOptions();
-        if (options.FilePatterns.Count == 0)
-        {
-            PreviewFileCount = 0;
-            PreviewFileSize = 0;
-            return; // Nothing search.
-        }
+            var selectedFolders = Root.GetSelectedItems().ToArray();
+            if (selectedFolders.Length == 0)
+            {
+                PreviewFileCount = 0;
+                PreviewFileSize = 0;
+                return; // Nothing to do.
+            }
 
-        var ingester = new Ingester(options);
-        m_backgroundRefreshProgress = new ProgressToken(true);
+            var options = GetIngestOptions();
+            if (options.FilePatterns.Count == 0)
+            {
+                PreviewFileCount = 0;
+                PreviewFileSize = 0;
+                return; // Nothing search.
+            }
 
-        Task.Run(() =>
-        {
-            IsGeneratingPreview = true;
-            try
+            var ingester = new Ingester(options);
+            m_backgroundRefreshProgress = new ProgressToken(isCancelSupported: true);
+        
+            Task.Run(() =>
             {
                 var result = ingester.Run(selectedFolders, progress: m_backgroundRefreshProgress);
                 if (!result.HasValue || m_backgroundRefreshProgress.CancelRequested)
                     return;
                 PreviewFileCount = result.Value.FileCount;
                 PreviewFileSize = result.Value.OutputBytes;
-            }
-            finally
-            {
-                IsGeneratingPreview = false;
-            }
-        });
+            });
+        }
+        finally
+        {
+            IsGeneratingPreview = false;
+        }
     }
 }

CodeIngest.Desktop/Views/MainWindow.axaml

@@ -49,22 +49,33 @@
                         </Style>
                     </Border.Styles>
 
-                    <Grid RowDefinitions="Auto,Auto" ColumnDefinitions="Auto,*,Auto"
+                    <Grid RowDefinitions="Auto,Auto" ColumnDefinitions="Auto,Auto,24,Auto,Auto,*,Auto"
                           Margin="4">
                         <TextBlock Grid.Column="0" Grid.Row="0" Text="Source files:"/>
                         <TextBlock Grid.Column="1" Grid.Row="0" Text="N/A"
-                                   IsVisible="{Binding PreviewFileCount, Converter={x:Static ObjectConverters.IsNull}}"/>
-                        <TextBlock Grid.Column="1" Grid.Row="0" Text="{Binding PreviewFileCount, StringFormat=N0}" 
-                                   IsVisible="{Binding PreviewFileCount, Converter={x:Static ObjectConverters.IsNotNull}}"/>
+                                   IsVisible="{Binding PreviewFileCount, Converter={x:Static ObjectConverters.IsNull}}"
+                                   MinWidth="80" />
+                        <TextBlock Grid.Column="1" Grid.Row="0" Text="{Binding PreviewFileCount, StringFormat=N0}"
+                                   IsVisible="{Binding PreviewFileCount, Converter={x:Static ObjectConverters.IsNotNull}}"
+                                   MinWidth="80" />
 
                         <TextBlock Grid.Column="0" Grid.Row="1" Text="Output size:"/>
                         <TextBlock Grid.Column="1" Grid.Row="1" Text="N/A"
                                    IsVisible="{Binding PreviewFileSize, Converter={x:Static ObjectConverters.IsNull}}"/>
                         <TextBlock Grid.Column="1" Grid.Row="1" Text="{Binding PreviewFileSize, Converter={StaticResource BytesToUiStringConverter}}"
                                    IsVisible="{Binding PreviewFileSize, Converter={x:Static ObjectConverters.IsNotNull}}"/>
 
+                        <TextBlock Grid.Column="3" Grid.Row="0" Text="GPT Context Usage:"/>
+                        <TextBlock Grid.Column="4" Grid.Row="0" Text="N/A"
+                                   IsVisible="{Binding PreviewFileSize, Converter={x:Static ObjectConverters.IsNull}}" />
+                        <StackPanel Grid.Column="4" Grid.Row="0" Orientation="Horizontal"
+                                    IsVisible="{Binding PreviewFileSize, Converter={x:Static ObjectConverters.IsNotNull}}">
+                            <TextBlock Text="{Binding PreviewTokenCount, StringFormat=N0}"/>
+                            <TextBlock Text="tokens"/>
+                        </StackPanel>
+
                         <!-- ReSharper disable once Xaml.StyleClassNotFound -->
-                        <ProgressBar Grid.Column="2" Grid.Row="0" Grid.RowSpan="2"
+                        <ProgressBar Grid.Column="6" Grid.Row="0" Grid.RowSpan="2"
                                      Classes="circular" IsIndeterminate="True"
                                      VerticalAlignment="Center"
                                      Width="36" Height="36" Margin="8,0"
@@ -73,6 +84,7 @@
                 </Border>
             </Grid>
 
+            <!-- Language/selection options -->
             <Grid Grid.Column="1" Margin="0,8,8,8"
                   RowDefinitions="Auto,Auto,*">
                 <Button Command="{Binding SelectRoot}">