highperfocused/nostrdvm
1
0
Fork 0
mirror of https://github.com/believethehype/nostrdvm.git synced 2025-11-19 22:36:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

rename request function

Browse Source
This commit is contained in:
Believethehype
2023-12-03 13:57:41 +01:00
parent 4410be7239
commit 26dab119c3
13 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
core/dvm.py
View File

@@ -451,8 +451,8 @@ class DVM:
for dvm in self.dvm_config.SUPPORTED_DVMS: for dvm in self.dvm_config.SUPPORTED_DVMS:
try: try:
if task == dvm.TASK: if task == dvm.TASK:
request_form = dvm.create_request_form_from_nostr_event(job_event, self.client, request_form = dvm.create_request_from_nostr_event(job_event, self.client,
self.dvm_config) self.dvm_config)
result = dvm.process(request_form) result = dvm.process(request_form)
try: try:
post_processed = dvm.post_process(result, job_event) post_processed = dvm.post_process(result, job_event)

2
interfaces/dvmtaskinterface.py
View File

@@ -67,7 +67,7 @@ class DVMTaskInterface:
"""Check if input is supported for current Task.""" """Check if input is supported for current Task."""
pass pass
def create_request_form_from_nostr_event(self, event, client=None, dvm_config=None) -> dict: def create_request_from_nostr_event(self, event, client=None, dvm_config=None) -> dict:
"""Parse input into a request form that will be given to the process method""" """Parse input into a request form that will be given to the process method"""
pass pass

2
tasks/advanced_search.py
View File

@@ -40,7 +40,7 @@ class AdvancedSearch(DVMTaskInterface):
# no input required # no input required
return True return True
def create_request_form_from_nostr_event(self, event, client=None, dvm_config=None): def create_request_from_nostr_event(self, event, client=None, dvm_config=None):
self.dvm_config = dvm_config self.dvm_config = dvm_config
print(self.dvm_config.PRIVATE_KEY) print(self.dvm_config.PRIVATE_KEY)

2
tasks/convert_media.py
View File

@@ -44,7 +44,7 @@ class MediaConverter(DVMTaskInterface):
return False return False
return True return True
def create_request_form_from_nostr_event(self, event, client=None, dvm_config=None): def create_request_from_nostr_event(self, event, client=None, dvm_config=None):
request_form = {"jobID": event.id().to_hex()} request_form = {"jobID": event.id().to_hex()}
url = "" url = ""
media_format = "video/mp4" media_format = "video/mp4"

2
tasks/discovery_inactive_follows.py
View File

@@ -42,7 +42,7 @@ class DiscoverInactiveFollows(DVMTaskInterface):
# no input required # no input required
return True return True
def create_request_form_from_nostr_event(self, event, client=None, dvm_config=None): def create_request_from_nostr_event(self, event, client=None, dvm_config=None):
self.client = client self.client = client
self.dvm_config = dvm_config self.dvm_config = dvm_config

2
tasks/imagegeneration_openai_dalle.py
View File

@@ -54,7 +54,7 @@ class ImageGenerationDALLE(DVMTaskInterface):
return True return True
def create_request_form_from_nostr_event(self, event, client=None, dvm_config=None): def create_request_from_nostr_event(self, event, client=None, dvm_config=None):
request_form = {"jobID": event.id().to_hex() + "_" + self.NAME.replace(" ", "")} request_form = {"jobID": event.id().to_hex() + "_" + self.NAME.replace(" ", "")}
prompt = "" prompt = ""
width = "1024" width = "1024"

2
tasks/imagegeneration_sdxl.py
View File

@@ -51,7 +51,7 @@ class ImageGenerationSDXL(DVMTaskInterface):
return True return True
def create_request_form_from_nostr_event(self, event, client=None, dvm_config=None): def create_request_from_nostr_event(self, event, client=None, dvm_config=None):
request_form = {"jobID": event.id().to_hex() + "_" + self.NAME.replace(" ", "")} request_form = {"jobID": event.id().to_hex() + "_" + self.NAME.replace(" ", "")}
request_form["trainerFilePath"] = 'modules\\stablediffusionxl\\stablediffusionxl.trainer' request_form["trainerFilePath"] = 'modules\\stablediffusionxl\\stablediffusionxl.trainer'

2
tasks/imagegeneration_sdxlimg2img.py
View File

@@ -58,7 +58,7 @@ class ImageGenerationSDXLIMG2IMG(DVMTaskInterface):
return True return True
def create_request_form_from_nostr_event(self, event, client=None, dvm_config=None): def create_request_from_nostr_event(self, event, client=None, dvm_config=None):
request_form = {"jobID": event.id().to_hex() + "_" + self.NAME.replace(" ", "")} request_form = {"jobID": event.id().to_hex() + "_" + self.NAME.replace(" ", "")}
request_form["trainerFilePath"] = r'modules\stablediffusionxl\stablediffusionxl-img2img.trainer' request_form["trainerFilePath"] = r'modules\stablediffusionxl\stablediffusionxl-img2img.trainer'

2
tasks/textextraction_google.py
View File

@@ -53,7 +53,7 @@ class SpeechToTextGoogle(DVMTaskInterface):
return True return True
def create_request_form_from_nostr_event(self, event, client=None, dvm_config=None): def create_request_from_nostr_event(self, event, client=None, dvm_config=None):
request_form = {"jobID": event.id().to_hex() + "_" + self.NAME.replace(" ", "")} request_form = {"jobID": event.id().to_hex() + "_" + self.NAME.replace(" ", "")}
url = "" url = ""

2
tasks/textextraction_pdf.py
View File

@@ -40,7 +40,7 @@ class TextExtractionPDF(DVMTaskInterface):
return False return False
return True return True
def create_request_form_from_nostr_event(self, event, client=None, dvm_config=None): def create_request_from_nostr_event(self, event, client=None, dvm_config=None):
request_form = {"jobID": event.id().to_hex()} request_form = {"jobID": event.id().to_hex()}
# default values # default values

2
tasks/textextraction_whisperx.py
View File

@@ -51,7 +51,7 @@ class SpeechToTextWhisperX(DVMTaskInterface):
return True return True
def create_request_form_from_nostr_event(self, event, client=None, dvm_config=None): def create_request_from_nostr_event(self, event, client=None, dvm_config=None):
request_form = {"jobID": event.id().to_hex() + "_" + self.NAME.replace(" ", ""), request_form = {"jobID": event.id().to_hex() + "_" + self.NAME.replace(" ", ""),
"trainerFilePath": 'modules\\whisperx\\whisperx_transcript.trainer'} "trainerFilePath": 'modules\\whisperx\\whisperx_transcript.trainer'}

2
tasks/translation_google.py
View File

@@ -41,7 +41,7 @@ class TranslationGoogle(DVMTaskInterface):
return False return False
return True return True
def create_request_form_from_nostr_event(self, event, client=None, dvm_config=None): def create_request_from_nostr_event(self, event, client=None, dvm_config=None):
request_form = {"jobID": event.id().to_hex()} request_form = {"jobID": event.id().to_hex()}
text = "" text = ""
translation_lang = "en" translation_lang = "en"

2
tasks/translation_libretranslate.py
View File

@@ -44,7 +44,7 @@ class TranslationLibre(DVMTaskInterface):
return False return False
return True return True
def create_request_form_from_nostr_event(self, event, client=None, dvm_config=None): def create_request_from_nostr_event(self, event, client=None, dvm_config=None):
request_form = {"jobID": event.id().to_hex()} request_form = {"jobID": event.id().to_hex()}
text = "" text = ""
translation_lang = "en" translation_lang = "en"