Skip to content

feat: add history manager to track AI responses and transformations #273

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
saurondark22 wants to merge 2 commits into
base: main
Choose a base branch
from
Open

feat: add history manager to track AI responses and transformations #273

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
0d4c7ca
feat: add history manager to track AI responses and transformations
saurondark22 Jun 4, 2026
d30a6f6
feat(history): add delete button per entry and search filter
saurondark22 Jun 6, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
47 changes: 45 additions & 2 deletions Windows_and_Linux/WritingToolApp.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@
import ui.ResponseWindow
import ui.SettingsWindow
from aiprovider import GeminiProvider, OllamaProvider, OpenAICompatibleProvider, obfuscate_api_key
from history_manager import HistoryManager
from pynput import keyboard as pykeyboard
from PySide6 import QtCore, QtGui, QtWidgets
from PySide6.QtCore import QLocale, Signal, Slot
Expand Down Expand Up @@ -46,6 +47,7 @@ class WritingToolApp(QtWidgets.QApplication):
show_message_signal = Signal(str, str) # a signal for showing message boxes
hotkey_triggered_signal = Signal()
followup_response_signal = Signal(str)
history_updated_signal = Signal()


def __init__(self, argv):
Expand All @@ -65,6 +67,11 @@ def __init__(self, argv):
self.options = None
self.options_path = None
self.load_options()
self.history_manager = HistoryManager(
os.path.dirname(sys.argv[0]),
on_updated=self.history_updated_signal.emit
)
self.history_updated_signal.connect(self.history_manager.refresh_window)
self.onboarding_window = None
self.popup_window = None
self.tray_icon = None
Expand Down Expand Up @@ -149,6 +156,7 @@ def setup_translations(self, lang=None):
ui.ResponseWindow._ = self._
ui.OnboardingWindow._ = self._
ui.CustomPopupWindow._ = self._
self.history_manager.set_translation_function(self._)

def retranslate_ui(self):
self.update_tray_menu()
Expand Down Expand Up @@ -663,6 +671,8 @@ def process_option(self, option, custom_change=None):
if hasattr(self, 'current_response_window'):
delattr(self, 'current_response_window')

self.history_manager.clear_pending_inline_history()

threading.Thread(
target=self.process_option_thread,
args=(option, custom_change),
Expand Down Expand Up @@ -710,14 +720,17 @@ def process_option_thread(self, option, custom_change=None):
self.show_message_signal.emit('Error', 'Please select text to use this option.')
return

if self.options[option]['open_in_window']:
open_in_window = self.options[option]['open_in_window']
if open_in_window:
QtCore.QMetaObject.invokeMethod(
self,
'_setup_response_window',
QtCore.Qt.ConnectionType.BlockingQueuedConnection,
QtCore.Q_ARG(str, option),
QtCore.Q_ARG(str, selected_text)
)
else:
self.history_manager.set_pending_inline_history(option, selected_text)

try:
selected_prompt = self.options.get(option, ('', ''))
Expand All @@ -732,11 +745,28 @@ def process_option_thread(self, option, custom_change=None):

logging.debug(f'Getting response from provider for option: {option}')

if self.options[option]['open_in_window']:
if open_in_window:
logging.debug('Getting response for window display')
response = self.current_provider.get_response(system_instruction, prompt, return_response=True)
response = response or ''
logging.debug(f'Got response of length: {len(response) if response else 0}')

cleaned_response = response.rstrip('\n')
if cleaned_response.strip():
history_entry_id = self.history_manager.record_entry(
option=option,
input_text=selected_text,
output_text=cleaned_response,
conversation=[
{'role': 'user', 'content': selected_text},
{'role': 'assistant', 'content': cleaned_response}
]
)
self.history_manager.attach_entry_to_response_window(
getattr(self, 'current_response_window', None),
history_entry_id
)

if hasattr(self, 'current_response_window'):
# noinspection PyTypeChecker
QtCore.QMetaObject.invokeMethod(
Expand All @@ -753,6 +783,7 @@ def process_option_thread(self, option, custom_change=None):

except Exception as e:
logging.error(f'An error occurred: {e}', exc_info=True)
self.history_manager.clear_pending_inline_history()

if "Resource has been exhausted" in str(e):
self.show_message_signal.emit('Error - Rate Limit Hit', 'Whoops! You\'ve hit the per-minute rate limit of the Gemini API. Please try again in a few moments.\n\nIf this happens often, simply switch to a Gemini model with a higher usage limit in Settings.')
Expand Down Expand Up @@ -788,6 +819,7 @@ def replace_text(self, new_text):

# If the new text is the error message, show a message box
if current_output == error_message:
self.history_manager.clear_pending_inline_history()
self.show_message_signal.emit('Error', 'The text is incompatible with the requested change.')
return

Expand Down Expand Up @@ -826,12 +858,14 @@ def press_ctrl_v():
press_ctrl_v()
time.sleep(0.2)
pyperclip.copy(clipboard_backup)
self.history_manager.consume_pending_inline_history(cleaned_text)

if not hasattr(self, 'current_response_window'):
self.output_queue = ""

except Exception as e:
logging.error(f'Error processing output: {e}')
self.history_manager.clear_pending_inline_history()
else:
logging.debug('No new text to process')

Expand Down Expand Up @@ -870,6 +904,10 @@ def update_tray_menu(self):
# Apply dark mode styles using darkdetect
self.apply_dark_mode_styles(self.tray_menu)

# History menu item
history_action = self.tray_menu.addAction(self._('History'))
history_action.triggered.connect(self.history_manager.show_window)

# Settings menu item
settings_action = self.tray_menu.addAction(self._('Settings'))
settings_action.triggered.connect(self.show_settings)
Expand Down Expand Up @@ -975,6 +1013,9 @@ def process_thread():
"role": "user",
"content": question
})
history_entry_id = getattr(response_window, 'history_entry_id', None)
if history_entry_id:
self.history_manager.append_turn(history_entry_id, 'user', question)

# Get chat history
history = response_window.chat_history.copy()
Expand Down Expand Up @@ -1037,6 +1078,8 @@ def process_thread():
"role": "assistant",
"content": response_text
})
if history_entry_id and response_text:
self.history_manager.append_turn(history_entry_id, 'assistant', response_text)

# Emit response via signal
self.followup_response_signal.emit(response_text)
Expand Down
Loading