working note generator

This commit is contained in:
Tomasz Frątczak 2024-02-13 23:04:36 +01:00
parent f6bab3e7eb
commit 0d0750fa96
6 changed files with 31 additions and 11 deletions

0
notescriber/config.py Normal file
View File

View File

@ -17,16 +17,13 @@ class Dispatcher:
self.noteGenerator = noteGenerator self.noteGenerator = noteGenerator
def process_audio_file(self, audio_file): def process_audio_file(self, audio_file):
self.threads.append(threading.Thread(target=self._execute_thread, args=(audio_file.pathname,))) self.threads.append(threading.Thread(target=self._execute_thread, args=(audio_file,)))
self.threads[-1].start() self.threads[-1].start()
# TODO: we need to have a way to wipe old threads, otherwise we will have a memory leak # TODO: we need to have a way to wipe old threads, otherwise we will have a memory leak
def generate_note(self, note):
pass
def _execute_thread(self, audio_file): def _execute_thread(self, audio_file):
note = self.audioProcessor.execute(audio_file) note = self.audioProcessor.execute(audio_file.pathname)
self.noteGenerator.generate_note(note) self.noteGenerator.generate_note(audio_file.pathname, note)
def halt(self): def halt(self):
for thread in self.threads: for thread in self.threads:

View File

@ -1,7 +1,15 @@
from nextcloud_notes_api import NotesApi, Note
class Notetaker: class Notetaker:
def __init__(self): username = None
pass password = None
def generate_note(self, note): def __init__(self, username, password):
print("Received note: " + note) self.username = username
self.password = password
def generate_note(self, filename, note):
api = NotesApi(self.username, self.password, 'cloud.octopusx.de')
ncnote = Note(title=filename, content=note, category="Voicenotes")
api.create_note(ncnote)

View File

@ -1,2 +1,3 @@
openai-whisper==20231117 openai-whisper==20231117
pyinotify==0.9.6 pyinotify==0.9.6
nextcloud_notes_api==1.0.0

0
tests/__init__.py Normal file
View File

14
tests/notetaker_test.py Normal file
View File

@ -0,0 +1,14 @@
import unittest
from notescriber.notetaker import Notetaker
class TranscriptorTest(unittest.TestCase):
def setUp(self):
pass
def tearDown(self):
pass
def test_run(self):
ntk = Notetaker()
ntk.generate_note("Test Note Title","This is my note")
self.assertEqual(True, True)