highperfocused/nostrdvm
1
0
Fork 0
mirror of https://github.com/believethehype/nostrdvm.git synced 2025-03-17 21:31:52 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' into 036

Browse Source
This commit is contained in:
dbth 2024-11-06 11:14:59 +01:00 committed by GitHub
commit e96b736e3a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 16 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
nostr_dvm/tasks/imagegeneration_replicate.py
View File

@ -72,6 +72,7 @@ class ImageGenerationReplicate(DVMTaskInterface):
self.model = "stability-ai/stable-diffusion-3",
for tag in event.tags().to_vec():
if tag.as_vec()[0] == 'i':
input_type = tag.as_vec()[2]
if input_type == "text":

1
nostr_dvm/tasks/imagegeneration_replicate_fluxpro.py
View File

@ -70,6 +70,7 @@ class ImageGenerationReplicateFluxPro(DVMTaskInterface):
height = "1024"
for tag in event.tags().to_vec():
if tag.as_vec()[0] == 'i':
input_type = tag.as_vec()[2]
if input_type == "text":

18
nostr_dvm/utils/output_utils.py
View File

@ -105,9 +105,12 @@ def post_process_list_to_events(result):
if len(result_list) == 0:
return "No results found"
for tag in result_list:
e_tag = Tag.parse(tag)
result_str = result_str + "nostr:" + EventId.from_hex(
e_tag.as_vec()[1]).to_bech32() + "\n"
try:
e_tag = Tag.parse(tag)
id = EventId.from_hex(e_tag.as_vec()[1]).to_bech32()
result_str = result_str + "nostr:" + id + "\n"
except Exception as e:
print(e)
return result_str
@ -117,9 +120,12 @@ def post_process_list_to_users(result):
if len(result_list) == 0:
return "No results found"
for tag in result_list:
p_tag = Tag.parse(tag)
result_str = result_str + "nostr:" + PublicKey.parse(
p_tag.as_vec()[1]).to_bech32() + "\n"
try:
p_tag = Tag.parse(tag)
key = PublicKey.parse(p_tag.as_vec()[1]).to_bech32()
result_str = result_str + "nostr:" + key + "\n"
except Exception as e:
print(e)
return result_str

4
tests/otherstuff.py
View File

@ -418,9 +418,9 @@ def playground(announce=False):
#discovery_bots = build_botfarms("Bot Hunter", "discovery_botfarms", announce)
discovery_bots = build_botfarms("Bot Hunter", "discovery_botfarms", announce)
#bot_config.SUPPORTED_DVMS.append(discovery_bots)
#discovery_bots.run()
discovery_bots.run()
admin_config = AdminConfig()