diff --git a/.coveragerc b/.coveragerc index 4cdb448..9843aa0 100644 --- a/.coveragerc +++ b/.coveragerc @@ -1,6 +1,6 @@ [run] omit = - whisper_cpp.py + buzz/whisper_cpp.py *_test.py [html] diff --git a/Makefile b/Makefile index 9ad05c1..c397fdf 100644 --- a/Makefile +++ b/Makefile @@ -43,18 +43,18 @@ endif clean: rm -f $(LIBWHISPER) - rm -f whisper_cpp.py + rm -f buzz/whisper_cpp.py rm -rf dist/* || true -test: whisper_cpp.py +test: buzz/whisper_cpp.py pytest --cov --cov-report=html -dist/Buzz dist/Buzz.app: whisper_cpp.py +dist/Buzz dist/Buzz.app: buzz/whisper_cpp.py pyinstaller --noconfirm Buzz.spec version: poetry version ${version} - echo "VERSION = \"${version}\"" > __version__.py + echo "VERSION = \"${version}\"" > buzz/__version__.py sed -i "s/version=.*,/version=\'${version_escaped}\',/" Buzz.spec CMAKE_FLAGS= @@ -79,8 +79,8 @@ $(LIBWHISPER): cp whisper.cpp/build/$(LIBWHISPER) . || true cp whisper.cpp/build/bin/Debug/$(LIBWHISPER) . || true -whisper_cpp.py: $(LIBWHISPER) - ctypesgen ./whisper.cpp/whisper.h -l$(LIBWHISPER) -o whisper_cpp.py +buzz/whisper_cpp.py: $(LIBWHISPER) + ctypesgen ./whisper.cpp/whisper.h -l$(LIBWHISPER) -o buzz/whisper_cpp.py staple_app_mac: xcrun stapler staple ${mac_app_path} diff --git a/buzz/__init__.py b/buzz/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/__version__.py b/buzz/__version__.py similarity index 100% rename from __version__.py rename to buzz/__version__.py diff --git a/conn.py b/buzz/conn.py similarity index 100% rename from conn.py rename to buzz/conn.py diff --git a/gui.py b/buzz/gui.py similarity index 99% rename from gui.py rename to buzz/gui.py index 18eb08e..8d3cc6d 100644 --- a/gui.py +++ b/buzz/gui.py @@ -20,10 +20,10 @@ from PyQt6.QtWidgets import (QApplication, QCheckBox, QComboBox, QDialog, from requests import get from whisper import tokenizer -from __version__ import VERSION -from transcriber import FileTranscriber, OutputFormat, RecordingTranscriber -from whispr import LOADED_WHISPER_DLL, Task -from model_loader import ModelLoader +from .__version__ import VERSION +from .transcriber import FileTranscriber, OutputFormat, RecordingTranscriber +from .whispr import LOADED_WHISPER_DLL, Task +from .model_loader import ModelLoader APP_NAME = 'Buzz' diff --git a/model_loader.py b/buzz/model_loader.py similarity index 100% rename from model_loader.py rename to buzz/model_loader.py diff --git a/transcriber.py b/buzz/transcriber.py similarity index 98% rename from transcriber.py rename to buzz/transcriber.py index 3bcbd64..62827ad 100644 --- a/transcriber.py +++ b/buzz/transcriber.py @@ -18,8 +18,9 @@ import stable_whisper import whisper from sounddevice import PortAudioError -from conn import pipe_stderr, pipe_stdout -from whispr import Segment, Task, WhisperCpp, read_progress, whisper_cpp_params +from .conn import pipe_stderr, pipe_stdout +from .whispr import (Segment, Task, WhisperCpp, read_progress, + whisper_cpp_params) class RecordingTranscriber: diff --git a/whispr.py b/buzz/whispr.py similarity index 98% rename from whispr.py rename to buzz/whispr.py index 3f221ea..d353022 100644 --- a/whispr.py +++ b/buzz/whispr.py @@ -14,7 +14,7 @@ import whisper # the DLL in some envs is fixed. LOADED_WHISPER_DLL = False try: - import whisper_cpp + import buzz.whisper_cpp as whisper_cpp LOADED_WHISPER_DLL = True except ImportError: logging.exception('') diff --git a/main.py b/main.py index cf73788..b780888 100644 --- a/main.py +++ b/main.py @@ -34,6 +34,6 @@ if __name__ == "__main__": level=logging.DEBUG, format="[%(asctime)s] %(module)s.%(funcName)s:%(lineno)d %(levelname)s -> %(message)s") - from gui import Application + from buzz.gui import Application app = Application() sys.exit(app.exec()) diff --git a/tests/__init__.py b/tests/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/gui_test.py b/tests/gui_test.py similarity index 96% rename from gui_test.py rename to tests/gui_test.py index 2e78be8..8ad59be 100644 --- a/gui_test.py +++ b/tests/gui_test.py @@ -5,11 +5,12 @@ from unittest.mock import patch import sounddevice from PyQt6.QtCore import Qt -from gui import (Application, AudioDevicesComboBox, - DownloadModelProgressDialog, FileTranscriberWidget, - LanguagesComboBox, MainWindow, OutputFormatsComboBox, Quality, Settings, AboutDialog, - QualityComboBox, TranscriberProgressDialog) -from transcriber import OutputFormat +from buzz.gui import (AboutDialog, Application, AudioDevicesComboBox, + DownloadModelProgressDialog, FileTranscriberWidget, + LanguagesComboBox, MainWindow, OutputFormatsComboBox, + Quality, QualityComboBox, Settings, + TranscriberProgressDialog) +from buzz.transcriber import OutputFormat class TestApplication: diff --git a/transcriber_test.py b/tests/transcriber_test.py similarity index 96% rename from transcriber_test.py rename to tests/transcriber_test.py index ac34420..cc2d54d 100644 --- a/transcriber_test.py +++ b/tests/transcriber_test.py @@ -1,4 +1,3 @@ -import logging import os import pathlib import tempfile @@ -6,10 +5,10 @@ import time import pytest -from model_loader import ModelLoader -from transcriber import (FileTranscriber, OutputFormat, RecordingTranscriber, - to_timestamp) -from whispr import Task +from buzz.model_loader import ModelLoader +from buzz.transcriber import (FileTranscriber, OutputFormat, + RecordingTranscriber, to_timestamp) +from buzz.whispr import Task def get_model_path(model_name: str, use_whisper_cpp: bool) -> str: