RAGParameterDialog.py

from Settings import settings
import wx


class RAGParameterDialog(wx.Dialog):
	def __init__(self, parent, title):
		super().__init__(parent, title=title, size=(400, 400))

		self.panel = wx.Panel(self)
		self.grid_sizer = wx.GridBagSizer(
			5, 5
		)  # Specify vertical and horizontal gap between widgets

		# Define parameters and their tooltips
		self.parameters = {
			"chunk_size": {
				"label": "Chunk Size",
				"control": "SpinCtrl",
				"min": 1,
				"max": int(settings.parameters["num_ctx"]["value"] / 2),
				"initial": settings.chunk_size,
				"tooltip": "Defines the size of text chunks for indexing. Smaller sizes may improve search granularity.",
			},
			"chunk_overlap": {
				"label": "Chunk Overlap",
				"control": "SpinCtrl",
				"min": 0,
				"max": 100,
				"initial": settings.chunk_overlap,
				"tooltip": "Overlap between chunks to ensure continuity in search results.",
			},
			"similarity_top_k": {
				"label": "Similarity Top K",
				"control": "SpinCtrl",
				"min": 1,
				"max": 100,
				"initial": settings.similarity_top_k,
				"tooltip": "The number of top results to consider when evaluating similarity.",
			},
			"similarity_cutoff": {
				"label": "Similarity Cutoff",
				"control": "SpinCtrlDouble",
				"min": 0.0,
				"max": 1.0,
				"inc": 0.01,
				"initial": settings.similarity_cutoff,
				"tooltip": "The minimum similarity score for a result to be considered relevant.",
			},
			"response_mode": {
				"label": "Response Mode",
				"control": "Choice",
				"choices": [
					"refine",
					"compact",
					"tree_summarize",
					"simple_summarize",
					"accumulate",
					"compact_accumulate",
				],
				"initial": settings.response_mode,
				"tooltip": "Determines how the indexed results are processed and presented.",
			},
			"show_context": {
				"label": "Show Context",
				"control": "CheckBox",
				"initial": settings.show_context,
				"tooltip": "Toggle to show or hide additional context related to your query.",
			},
		}

		self.controls = {}
		row_index = 0
		for key, value in self.parameters.items():
			self.add_parameter_control(row_index, key, **value)
			row_index += 1

		# Add OK and Cancel buttons at the bottom
		ok_button = wx.Button(self.panel, label="OK", id=wx.ID_OK)
		ok_button.Bind(wx.EVT_BUTTON, self.on_ok)
		cancel_button = wx.Button(self.panel, label="Cancel", id=wx.ID_CANCEL)
		self.grid_sizer.Add(ok_button, pos=(row_index, 0), flag=wx.ALL, border=5)
		self.grid_sizer.Add(cancel_button, pos=(row_index, 1), flag=wx.ALL, border=5)

		self.panel.SetSizer(self.grid_sizer)

	def add_parameter_control(self, row, name, label, control, **kwargs):
		label_widget = wx.StaticText(self.panel, label=label)
		self.grid_sizer.Add(
			label_widget, pos=(row, 0), flag=wx.ALL | wx.ALIGN_CENTER_VERTICAL, border=5
		)

		if control == "SpinCtrl":
			widget = wx.SpinCtrl(
				self.panel,
				value=str(kwargs["initial"]),
				min=kwargs["min"],
				max=kwargs["max"],
			)
		elif control == "CheckBox":
			widget = wx.CheckBox(self.panel)
			widget.SetValue(kwargs["initial"])
		elif control == "SpinCtrlDouble":
			widget = wx.SpinCtrlDouble(
				self.panel,
				value=str(kwargs["initial"]),
				min=kwargs["min"],
				max=kwargs["max"],
				inc=kwargs["inc"],
			)
			widget.SetValue(kwargs["initial"])
		elif control == "Choice":
			widget = wx.Choice(self.panel, choices=kwargs["choices"])
			widget.SetStringSelection(kwargs["initial"])
		else:
			raise ValueError("Unsupported control type")

		# Set tooltip
		widget.SetToolTip(kwargs["tooltip"])
		widget.SetName(label)
		self.controls[name] = widget
		self.grid_sizer.Add(widget, pos=(row, 1), flag=wx.EXPAND | wx.ALL, border=5)

	def on_ok(self, event):
		for key, widget in self.controls.items():
			if isinstance(widget, wx.Choice):
				value = widget.GetStringSelection()
			else:
				value = widget.GetValue()
			setattr(settings, key, value)
		self.Close()