diff --git a/nostr_dvm/tasks/imagegeneration_replicate.py b/nostr_dvm/tasks/imagegeneration_replicate.py index 4a8746d..6f6e902 100644 --- a/nostr_dvm/tasks/imagegeneration_replicate.py +++ b/nostr_dvm/tasks/imagegeneration_replicate.py @@ -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": diff --git a/nostr_dvm/tasks/imagegeneration_replicate_fluxpro.py b/nostr_dvm/tasks/imagegeneration_replicate_fluxpro.py index 96d817a..f5a8702 100644 --- a/nostr_dvm/tasks/imagegeneration_replicate_fluxpro.py +++ b/nostr_dvm/tasks/imagegeneration_replicate_fluxpro.py @@ -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": diff --git a/nostr_dvm/utils/output_utils.py b/nostr_dvm/utils/output_utils.py index c4f2f83..6a2f9dc 100644 --- a/nostr_dvm/utils/output_utils.py +++ b/nostr_dvm/utils/output_utils.py @@ -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 diff --git a/tests/otherstuff.py b/tests/otherstuff.py index b7d1491..459f096 100644 --- a/tests/otherstuff.py +++ b/tests/otherstuff.py @@ -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()