mirror of
https://github.com/believethehype/nostrdvm.git
synced 2025-10-11 00:12:28 +02:00
move functions into component modules
This commit is contained in:
23
main.py
23
main.py
@@ -4,35 +4,24 @@ from threading import Thread
|
||||
|
||||
import dotenv
|
||||
import utils.env as env
|
||||
from tasks.textextractionPDF import TextExtractionPDF
|
||||
from tasks.translation import Translation
|
||||
from utils.definitions import EventDefinitions
|
||||
|
||||
|
||||
def run_nostr_dvm_with_local_config():
|
||||
from dvm import dvm, DVMConfig
|
||||
from utils.nip89_utils import NIP89Announcement
|
||||
|
||||
dvmconfig = DVMConfig()
|
||||
dvmconfig.PRIVATE_KEY = os.getenv(env.NOSTR_PRIVATE_KEY)
|
||||
dvmconfig.SUPPORTED_TASKS = ["translation", "pdf-to-text"]
|
||||
dvmconfig.SUPPORTED_TASKS = [Translation.TASK, TextExtractionPDF.TASK]
|
||||
dvmconfig.LNBITS_INVOICE_KEY = os.getenv(env.LNBITS_INVOICE_KEY)
|
||||
dvmconfig.LNBITS_URL = os.getenv(env.LNBITS_HOST)
|
||||
|
||||
# In admin_utils, set rebroadcast_nip89 to true to (re)broadcast your DVM. You can create a valid dtag and the content on vendata.io
|
||||
# Add the dtag in your .env file so you can update your dvm later and change the content here as needed.
|
||||
|
||||
nip89extraction = NIP89Announcement()
|
||||
nip89extraction.kind = EventDefinitions.KIND_NIP90_EXTRACT_TEXT
|
||||
nip89extraction.dtag = os.getenv(env.TASK_TRANSLATION_NIP89_DTAG)
|
||||
nip89extraction.pk = os.getenv(env.NOSTR_PRIVATE_KEY)
|
||||
nip89extraction.content = "{\"name\":\"Pdf Extractor\",\"image\":\"https://image.nostr.build/c33ca6fc4cc038ca4adb46fdfdfda34951656f87ee364ef59095bae1495ce669.jpg\",\"about\":\"I extract Text from pdf documents\",\"nip90Params\":{}}"
|
||||
dvmconfig.NIP89s.append(nip89extraction)
|
||||
|
||||
nip89translation = NIP89Announcement()
|
||||
nip89translation.kind = EventDefinitions.KIND_NIP90_TRANSLATE_TEXT
|
||||
nip89translation.dtag = os.getenv(env.TASK_TRANSLATION_NIP89_DTAG)
|
||||
nip89translation.pk = os.getenv(env.NOSTR_PRIVATE_KEY)
|
||||
nip89translation.content = "{\"name\":\"Translator\",\"image\":\"https://image.nostr.build/c33ca6fc4cc038ca4adb46fdfdfda34951656f87ee364ef59095bae1495ce669.jpg\",\"about\":\"I translate text from given text/event/job, currently using Google Translation Services into language defined in param. \",\"nip90Params\":{\"language\":{\"required\":true,\"values\":[\"af\",\"am\",\"ar\",\"az\",\"be\",\"bg\",\"bn\",\"bs\",\"ca\",\"ceb\",\"co\",\"cs\",\"cy\",\"da\",\"de\",\"el\",\"eo\",\"es\",\"et\",\"eu\",\"fa\",\"fi\",\"fr\",\"fy\",\"ga\",\"gd\",\"gl\",\"gu\",\"ha\",\"haw\",\"hi\",\"hmn\",\"hr\",\"ht\",\"hu\",\"hy\",\"id\",\"ig\",\"is\",\"it\",\"he\",\"ja\",\"jv\",\"ka\",\"kk\",\"km\",\"kn\",\"ko\",\"ku\",\"ky\",\"la\",\"lb\",\"lo\",\"lt\",\"lv\",\"mg\",\"mi\",\"mk\",\"ml\",\"mn\",\"mr\",\"ms\",\"mt\",\"my\",\"ne\",\"nl\",\"no\",\"ny\",\"or\",\"pa\",\"pl\",\"ps\",\"pt\",\"ro\",\"ru\",\"sd\",\"si\",\"sk\",\"sl\",\"sm\",\"sn\",\"so\",\"sq\",\"sr\",\"st\",\"su\",\"sv\",\"sw\",\"ta\",\"te\",\"tg\",\"th\",\"tl\",\"tr\",\"ug\",\"uk\",\"ur\",\"uz\",\"vi\",\"xh\",\"yi\",\"yo\",\"zh\",\"zu\"]}}}"
|
||||
dvmconfig.NIP89s.append(nip89translation)
|
||||
# Add the dtag in your .env file so you can update your dvm later and change the content in the module file as needed.
|
||||
dvmconfig.NIP89s.append(TextExtractionPDF.NIP89_announcement())
|
||||
dvmconfig.NIP89s.append(Translation.NIP89_announcement())
|
||||
|
||||
nostr_dvm_thread = Thread(target=dvm, args=[dvmconfig])
|
||||
nostr_dvm_thread.start()
|
||||
|
Reference in New Issue
Block a user