From e07f507db1c6864745bd65ac5b4849a18f329e6f Mon Sep 17 00:00:00 2001
From: tux <t@tux.rs>
Date: Wed, 2 Jul 2025 23:50:31 +0530
Subject: [PATCH] feat: add custom prompt box for screenshots

---
 ui/controls/ChatUserControl.axaml.cs     | 4 +---
 ui/controls/SettingsUserControl.axaml    | 7 ++++++-
 ui/controls/SettingsUserControl.axaml.cs | 4 +++-
 utils/InMemoryDB.cs                      | 4 +---
 utils/SetttingsManager.cs                | 3 ++-
 5 files changed, 13 insertions(+), 9 deletions(-)

diff --git a/ui/controls/ChatUserControl.axaml.cs b/ui/controls/ChatUserControl.axaml.cs
index 7b8bf52..371da33 100644
--- a/ui/controls/ChatUserControl.axaml.cs
+++ b/ui/controls/ChatUserControl.axaml.cs
@@ -27,7 +27,6 @@ public partial class ChatUserControl : UserControl
     {
         try
         {
-
             var timestamp = DateTime.Now.ToString("yyyyMMdd_HHmmss");
             var fileName = $"screenshot_{timestamp}.png";
             var filePath = Path.Combine(Environment.CurrentDirectory, fileName);
@@ -41,8 +40,7 @@ public partial class ChatUserControl : UserControl
             List<ChatMessage> messages =
             [
                 new UserChatMessage(
-                    ChatMessageContentPart.CreateTextPart(
-                        "I'm attaching a screenshot of a problem. I want you to read it and give me the appropriate answer."),
+                    ChatMessageContentPart.CreateTextPart(InMemoryDb.Obj.SettingsManager.Settings.ScreenshotPrompt),
                     ChatMessageContentPart.CreateImagePart(imageBytes, "image/png")
                 )
             ];
diff --git a/ui/controls/SettingsUserControl.axaml b/ui/controls/SettingsUserControl.axaml
index 6f60ca2..f6bb01c 100644
--- a/ui/controls/SettingsUserControl.axaml
+++ b/ui/controls/SettingsUserControl.axaml
@@ -5,7 +5,7 @@
              mc:Ignorable="d" d:DesignWidth="800" d:DesignHeight="450"
              x:Class="highminded.ui.controls.SettingsUserControl">
 
-    <Grid Margin="15" RowSpacing="10" ColumnSpacing="10" RowDefinitions="Auto,Auto,Auto" ColumnDefinitions="*,*">
+    <Grid Margin="15" RowSpacing="10" ColumnSpacing="10" RowDefinitions="Auto,Auto,Auto,Auto" ColumnDefinitions="*,*">
         
         <StackPanel Grid.Row="0" Grid.Column="0">
             <TextBlock Text="Model" />
@@ -23,6 +23,11 @@
         </StackPanel>
         
         <StackPanel Grid.Row="2" Grid.Column="0" Grid.ColumnSpan="2">
+            <TextBlock Text="Screenshot Prompt" />
+            <TextBox Name="ScreenshotPromptTextBox" TextWrapping="Wrap"/>
+        </StackPanel>
+        
+        <StackPanel Grid.Row="3" Grid.Column="0" Grid.ColumnSpan="2">
             <Button Name="SaveSettingsBtn" Content="Save" Click="SaveSettingsBtn_OnClick"/>
         </StackPanel> 
         
diff --git a/ui/controls/SettingsUserControl.axaml.cs b/ui/controls/SettingsUserControl.axaml.cs
index a7fbf89..98b92d8 100644
--- a/ui/controls/SettingsUserControl.axaml.cs
+++ b/ui/controls/SettingsUserControl.axaml.cs
@@ -13,13 +13,15 @@ public partial class SettingsUserControl : UserControl
         ModelTextBox.Text = InMemoryDb.Obj.SettingsManager.Settings.Model;
         ApiUrlTextBox.Text = InMemoryDb.Obj.SettingsManager.Settings.ApiURL;
         ApiKeyTextBox.Text = InMemoryDb.Obj.SettingsManager.Settings.ApiKey;
+        ScreenshotPromptTextBox.Text = InMemoryDb.Obj.SettingsManager.Settings.ScreenshotPrompt;
     }
 
     private void SaveSettingsBtn_OnClick(object? sender, RoutedEventArgs e)
     {
         InMemoryDb.Obj.SaveSettings(new AppSettings()
         {
-            ApiKey = ApiKeyTextBox.Text, ApiURL = ApiUrlTextBox.Text, Model = ModelTextBox.Text
+            ApiKey = ApiKeyTextBox.Text, ApiURL = ApiUrlTextBox.Text, Model = ModelTextBox.Text,
+            ScreenshotPrompt = ScreenshotPromptTextBox.Text
         });
     }
 }
\ No newline at end of file
diff --git a/utils/InMemoryDB.cs b/utils/InMemoryDB.cs
index f859d7a..6964685 100644
--- a/utils/InMemoryDB.cs
+++ b/utils/InMemoryDB.cs
@@ -34,9 +34,7 @@ public class InMemoryDb
 
     public void SaveSettings(AppSettings settings)
     {
-        SettingsManager.Settings.ApiKey = settings.ApiKey;
-        SettingsManager.Settings.ApiURL = settings.ApiURL;
-        SettingsManager.Settings.Model = settings.Model;
+        SettingsManager.Settings = settings;
         SettingsManager.Save();
         InitOpenAIClient();
     }
diff --git a/utils/SetttingsManager.cs b/utils/SetttingsManager.cs
index 66826ab..6928f2b 100644
--- a/utils/SetttingsManager.cs
+++ b/utils/SetttingsManager.cs
@@ -9,13 +9,14 @@ public class AppSettings
     public string Model { get; set; }
     public string ApiURL { get; set; }
     public string ApiKey { get; set; }
+    public string ScreenshotPrompt { get; set; }
     
 }
 
 public class SettingsManager<T> where T : class, new()
 {
     private readonly string _settingsPath;
-    public T Settings { get; private set; }
+    public T Settings { get; internal set; }
 
     public SettingsManager(string appName = "highminded")
     {