feat(mistralai_engine): migrate to mistralai module, add support for OCR and Document Understanding

basilisk/gui/account_dialog.py

@@ -658,11 +658,13 @@ def _update_base_url_fields(self, provider: Provider) -> None:
 		self.custom_base_url_text_ctrl.Enable(provider.allow_custom_base_url)
 		default_base_url = provider.base_url
 		if default_base_url:
+			# Translators: A label in account dialog
 			self.custom_base_url_label.SetLabel(
 				_("Custom &base URL (default: {})").format(default_base_url)
 			)
 		else:
-			self.custom_base_url_label.SetLabel(_("Custom &base URL"))
+			# Translators: A label in account dialog
+			self.custom_base_url_label.SetLabel(_("Custom &base URL:"))
 
 	def on_ok(self, event: wx.CommandEvent) -> None:
 		"""Handle the OK button click event.

basilisk/gui/conversation_tab.py

@@ -20,6 +20,7 @@
 import re
 import threading
 import time
+from multiprocessing import Process
 from typing import TYPE_CHECKING, Any, Optional
 
 import wx
@@ -51,6 +52,7 @@
 
 from .base_conversation import BaseConversation
 from .history_msg_text_ctrl import HistoryMsgTextCtrl
+from .ocr_handler import OCRHandler
 from .read_only_message_dialog import ReadOnlyMessageDialog
 
 if TYPE_CHECKING:
@@ -62,6 +64,8 @@
 log = logging.getLogger(__name__)
 accessible_output = get_accessible_output()
 
+CHECK_TASK_DELAY = 100  # ms
+
 
 class ConversationTab(wx.Panel, BaseConversation):
 	"""A tab panel that manages a single conversation with an AI assistant.
@@ -164,7 +168,9 @@ def __init__(
 		self.last_time = 0
 		self.recording_thread: Optional[RecordingThread] = None
 		self.task = None
+		self.process: Optional[Process] = None
 		self._stop_completion = False
+		self.ocr_handler = OCRHandler(self)
 		self.init_ui()
 		self.init_data(profile)
 		self.adjust_advanced_mode_setting()
@@ -251,6 +257,10 @@ def init_ui(self):
 		self.attachments_list.SetColumnWidth(1, 100)
 		self.attachments_list.SetColumnWidth(2, 500)
 		sizer.Add(self.attachments_list, proportion=0, flag=wx.ALL | wx.EXPAND)
+
+		self.ocr_button = self.ocr_handler.create_ocr_widget(self)
+		sizer.Add(self.ocr_button, proportion=0, flag=wx.EXPAND)
+
 		label = self.create_model_widget()
 		sizer.Add(label, proportion=0, flag=wx.EXPAND)
 		sizer.Add(self.model_list, proportion=0, flag=wx.ALL | wx.EXPAND)
@@ -364,6 +374,9 @@ def on_account_change(self, event: wx.CommandEvent | None):
 		self.toggle_record_btn.Enable(
 			ProviderCapability.STT in account.provider.engine_cls.capabilities
 		)
+		self.ocr_button.Enable(
+			ProviderCapability.OCR in account.provider.engine_cls.capabilities
+		)
 		self.web_search_mode.Enable(
 			ProviderCapability.WEB_SEARCH
 			in account.provider.engine_cls.capabilities
@@ -708,10 +721,12 @@ def refresh_attachments_list(self):
 		if not self.attachment_files:
 			self.attachments_list_label.Hide()
 			self.attachments_list.Hide()
+			self.ocr_button.Hide()
 			self.Layout()
 			return
 		self.attachments_list_label.Show()
 		self.attachments_list.Show()
+		self.ocr_button.Show()
 		for attachment in self.attachment_files:
 			self.attachments_list.Append(attachment.get_display_info())
 		last_index = len(self.attachment_files) - 1
@@ -1138,6 +1153,11 @@ def get_completion_args(self) -> dict[str, Any] | None:
 		}
 
 	def _check_attachments_valid(self) -> bool:
+		"""Check if all attachments are valid and supported by the current provider.
+
+		Returns:
+			True if all attachments are valid, False otherwise
+		"""
 		supported_attachment_formats = (
 			self.current_engine.supported_attachment_formats
 		)
@@ -1158,7 +1178,6 @@ def _check_attachments_valid(self) -> bool:
 				invalid_found = True
 		return not invalid_found
 
-	@ensure_no_task_running
 	def on_submit(self, event: wx.CommandEvent):
 		"""Handle the submission of a new message block for completion.