Skip to content

Commit

Permalink
Will remember previously used used huggingface model (#736)
Browse files Browse the repository at this point in the history
  • Loading branch information
raivisdejus committed May 14, 2024
1 parent 8990dcd commit ca49b8e
Showing 1 changed file with 11 additions and 2 deletions.
13 changes: 11 additions & 2 deletions buzz/widgets/transcriber/hugging_face_search_line_edit.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,10 +14,12 @@
QEvent,
)
from PyQt6.QtGui import QKeyEvent
from PyQt6.QtCore import QSettings
from PyQt6.QtNetwork import QNetworkAccessManager, QNetworkRequest, QNetworkReply
from PyQt6.QtWidgets import QListWidget, QWidget, QAbstractItemView, QListWidgetItem

from buzz.widgets.line_edit import LineEdit
from buzz.settings.settings import APP_NAME


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

super().__init__(default_value, parent)

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

self.timer = QTimer(self)
self.timer.setSingleShot(True)
Expand All @@ -59,6 +63,10 @@ def __init__(
self.popup.installEventFilter(self)
self.popup.itemClicked.connect(self.on_select_item)

def focusInEvent(self, event):
super().focusInEvent(event)
self.clear()

def on_text_edited(self, text: str):
self.model_selected.emit(text)

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

item = self.popup.currentItem()
self.setText(item.text())
self.settings.setValue("hugging_face_model_id", item.text())
QMetaObject.invokeMethod(self, "returnPressed")
self.model_selected.emit(item.data(Qt.ItemDataRole.UserRole))

Expand Down

0 comments on commit ca49b8e

Please sign in to comment.