mirror of
https://github.com/believethehype/nostrdvm.git
synced 2025-03-17 21:31:52 +01:00
104 lines
3.8 KiB
Python
104 lines
3.8 KiB
Python
import json
|
|
import time
|
|
from pathlib import Path
|
|
from threading import Thread
|
|
|
|
import dotenv
|
|
from nostr_sdk import Keys, Client, Tag, EventBuilder, Filter, HandleNotification, Timestamp, nip04_decrypt, \
|
|
NostrSigner
|
|
|
|
from nostr_dvm.utils.dvmconfig import DVMConfig
|
|
from nostr_dvm.utils.nostr_utils import send_event, check_and_set_private_key
|
|
from nostr_dvm.utils.definitions import EventDefinitions
|
|
|
|
|
|
def nostr_client_test_tts(prompt):
|
|
keys = Keys.parse(check_and_set_private_key("test_client"))
|
|
|
|
iTag = Tag.parse(["i", prompt, "text"])
|
|
paramTag1 = Tag.parse(["param", "language", "en"])
|
|
|
|
|
|
bidTag = Tag.parse(['bid', str(1000 * 1000), str(1000 * 1000)])
|
|
relaysTag = Tag.parse(['relays', "wss://relay.damus.io", "wss://blastr.f7z.xyz", "wss://relayable.org",
|
|
"wss://nostr-pub.wellorder.net"])
|
|
alttag = Tag.parse(["alt", "This is a NIP90 DVM AI task to generate TTS"])
|
|
event = EventBuilder(EventDefinitions.KIND_NIP90_TEXT_TO_SPEECH, str("Generate an Audio File."),
|
|
[iTag, paramTag1, bidTag, relaysTag, alttag]).to_event(keys)
|
|
|
|
relay_list = ["wss://relay.damus.io", "wss://blastr.f7z.xyz", "wss://relayable.org",
|
|
"wss://nostr-pub.wellorder.net"]
|
|
|
|
|
|
signer = NostrSigner.keys(keys)
|
|
client = Client(signer)
|
|
for relay in relay_list:
|
|
client.add_relay(relay)
|
|
client.connect()
|
|
config = DVMConfig
|
|
send_event(event, client=client, dvm_config=config)
|
|
return event.as_json()
|
|
|
|
def nostr_client():
|
|
keys = Keys.parse(check_and_set_private_key("test_client"))
|
|
sk = keys.secret_key()
|
|
pk = keys.public_key()
|
|
print(f"Nostr Test Client public key: {pk.to_bech32()}, Hex: {pk.to_hex()} ")
|
|
signer = NostrSigner.keys(keys)
|
|
client = Client(signer)
|
|
|
|
dvmconfig = DVMConfig()
|
|
for relay in dvmconfig.RELAY_LIST:
|
|
client.add_relay(relay)
|
|
client.connect()
|
|
|
|
dm_zap_filter = Filter().pubkey(pk).kinds([EventDefinitions.KIND_DM,
|
|
EventDefinitions.KIND_ZAP]).since(
|
|
Timestamp.now()) # events to us specific
|
|
dvm_filter = (Filter().kinds([EventDefinitions.KIND_NIP90_RESULT_TEXT_TO_SPEECH,
|
|
EventDefinitions.KIND_FEEDBACK]).since(Timestamp.now())) # public events
|
|
client.subscribe([dm_zap_filter, dvm_filter])
|
|
|
|
|
|
nostr_client_test_tts("Hello, this is a test. Mic check one, two.")
|
|
print("Sending Job Request")
|
|
|
|
|
|
#nostr_client_test_image_private("a beautiful ostrich watching the sunset")
|
|
class NotificationHandler(HandleNotification):
|
|
def handle(self, relay_url, event):
|
|
print(f"Received new event from {relay_url}: {event.as_json()}")
|
|
if event.kind() == 7000:
|
|
print("[Nostr Client]: " + event.as_json())
|
|
elif 6000 < event.kind() < 6999:
|
|
print("[Nostr Client]: " + event.as_json())
|
|
print("[Nostr Client]: " + event.content())
|
|
|
|
elif event.kind() == 4:
|
|
dec_text = nip04_decrypt(sk, event.author(), event.content())
|
|
print("[Nostr Client]: " + f"Received new msg: {dec_text}")
|
|
|
|
elif event.kind() == 9735:
|
|
print("[Nostr Client]: " + f"Received new zap:")
|
|
print(event.as_json())
|
|
|
|
def handle_msg(self, relay_url, msg):
|
|
return
|
|
|
|
client.handle_notifications(NotificationHandler())
|
|
while True:
|
|
time.sleep(5.0)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
env_path = Path('.env')
|
|
if env_path.is_file():
|
|
print(f'loading environment from {env_path.resolve()}')
|
|
dotenv.load_dotenv(env_path, verbose=True, override=True)
|
|
else:
|
|
raise FileNotFoundError(f'.env file not found at {env_path} ')
|
|
|
|
nostr_dvm_thread = Thread(target=nostr_client())
|
|
nostr_dvm_thread.start()
|