Skip to content

Commit ca49b8e

Browse files
authored
Will remember previously used used huggingface model (#736)
1 parent 8990dcd commit ca49b8e

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

buzz/widgets/transcriber/hugging_face_search_line_edit.py

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,10 +14,12 @@
1414
QEvent,
1515
)
1616
from PyQt6.QtGui import QKeyEvent
17+
from PyQt6.QtCore import QSettings
1718
from PyQt6.QtNetwork import QNetworkAccessManager, QNetworkRequest, QNetworkReply
1819
from PyQt6.QtWidgets import QListWidget, QWidget, QAbstractItemView, QListWidgetItem
1920

2021
from buzz.widgets.line_edit import LineEdit
22+
from buzz.settings.settings import APP_NAME
2123

2224

2325
# Adapted from https://github.com/ismailsunni/scripts/blob/master/autocomplete_from_url.py
@@ -30,10 +32,12 @@ def __init__(
3032
network_access_manager: Optional[QNetworkAccessManager] = None,
3133
parent: Optional[QWidget] = None,
3234
):
33-
super().__init__("", parent)
35+
self.settings = QSettings(APP_NAME)
36+
default_value = self.settings.value("hugging_face_model_id", "openai/whisper-tiny")
37+
38+
super().__init__(default_value, parent)
3439

3540
self.setMinimumWidth(150)
36-
self.setPlaceholderText("openai/whisper-tiny")
3741

3842
self.timer = QTimer(self)
3943
self.timer.setSingleShot(True)
@@ -59,6 +63,10 @@ def __init__(
5963
self.popup.installEventFilter(self)
6064
self.popup.itemClicked.connect(self.on_select_item)
6165

66+
def focusInEvent(self, event):
67+
super().focusInEvent(event)
68+
self.clear()
69+
6270
def on_text_edited(self, text: str):
6371
self.model_selected.emit(text)
6472

@@ -68,6 +76,7 @@ def on_select_item(self):
6876

6977
item = self.popup.currentItem()
7078
self.setText(item.text())
79+
self.settings.setValue("hugging_face_model_id", item.text())
7180
QMetaObject.invokeMethod(self, "returnPressed")
7281
self.model_selected.emit(item.data(Qt.ItemDataRole.UserRole))
7382

0 commit comments

Comments
 (0)