import json
import time
from datetime import timedelta
from nicegui import run, ui
from nostr_sdk import Keys, Client, Tag, EventBuilder, Filter, \
Options, Timestamp, ClientSigner, EventId, Nip19Event, PublicKey
from nostr_dvm.utils import dvmconfig
from nostr_dvm.utils.dvmconfig import DVMConfig
from nostr_dvm.utils.nostr_utils import send_event, check_and_set_private_key, get_events_by_id
from nostr_dvm.utils.definitions import EventDefinitions
@ui.page('/', dark=True)
def init():
keys = Keys.from_sk_str(check_and_set_private_key("test_client"))
opts = (Options().wait_for_send(False).send_timeout(timedelta(seconds=2))
.skip_disconnected_relays(True))
signer = ClientSigner.keys(keys)
client = Client.with_opts(signer, opts)
relay_list = dvmconfig.DVMConfig.RELAY_LIST
for relay in relay_list:
client.add_relay(relay)
client.connect()
data = []
def nostr_client_test_search(prompt, users=None, since="", until=""):
if users is None:
users = []
iTag = Tag.parse(["i", prompt, "text"])
userTag = Tag.parse(['param', 'users', json.dumps(users)])
sinceTag = Tag.parse(['param', 'since', since])
untilTag = Tag.parse(['param', 'until', until])
maxResultsTag = Tag.parse(['param', 'max_results', "100"])
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 search content"])
tags = [iTag, relaysTag, alttag, maxResultsTag]
if users:
tags.append(userTag)
if since != "":
tags.append(sinceTag)
if until != "":
tags.append(untilTag)
event = EventBuilder(EventDefinitions.KIND_NIP90_CONTENT_SEARCH, str("Search.."),
tags).to_event(keys)
config = DVMConfig
config.RELAY_LIST = relay_list
send_event(event, client=client, dvm_config=config)
return event
def handledvm(now, eventid):
response = False
feedbackfilter = Filter().pubkey(keys.public_key()).kinds(
[EventDefinitions.KIND_NIP90_RESULTS_CONTENT_SEARCH]).since(now).event(eventid)
feedbackfilter2 = Filter().pubkey(keys.public_key()).kinds(
[EventDefinitions.KIND_FEEDBACK]).since(now).event(eventid)
lastfeedback = ""
while not response:
events = client.get_events_of([feedbackfilter], timedelta(seconds=3))
fevents = client.get_events_of([feedbackfilter2], timedelta(seconds=3))
if len(fevents) > 0:
if lastfeedback != fevents[0].content():
for tag in fevents[0].tags():
if tag.as_vec()[0] == "status":
if tag.as_vec()[1] == "error":
with mainrow:
with maincolumn:
ui.notify(fevents[0].content(), type="negative")
lastfeedback = fevents[0].content()
break
else:
with mainrow:
with maincolumn:
ui.notify(fevents[0].content(), type="info")
lastfeedback = fevents[0].content()
break
if len(events) == 0:
response = False
time.sleep(1.0)
continue
else:
if events[0].content() == "[]":
return []
event_etags = json.loads(events[0].content())
event_ids = []
for etag in event_etags:
eventidob = EventId.from_hex(etag[1])
event_ids.append(eventidob)
config = DVMConfig()
events = get_events_by_id(event_ids, client, config)
if events is None:
return []
listui = []
for event in events:
nip19event = Nip19Event(event.id(), event.author(), dvmconfig.DVMConfig.RELAY_LIST)
nip19eventid = nip19event.to_bech32()
new = {'result': event.content(), 'author': event.author().to_hex(),
'eventid': str(event.id().to_hex()),
'time': str(event.created_at().to_human_datetime()),
'njump': "https://njump.me/" + nip19eventid,
'highlighter': "https://highlighter.com/a/" + nip19eventid,
'nostrudel': "https://nostrudel.ninja/#/n/" + nip19eventid
}
listui.append(new)
print(event.as_json())
return listui
async def search():
table.visible = False
now = Timestamp.now()
taggedusersfrom = [str(word).lstrip('from:') for word in prompt.value.split() if
word.startswith('from:')]
taggedusersto = [str(word).lstrip('to:') for word in prompt.value.split() if word.startswith('to:')]
search = prompt.value
tags = []
for word in taggedusersfrom:
search = str(search).replace(word, "")
user_pubkey = PublicKey.from_bech32(word.replace("@", "")).to_hex()
pTag = Tag.parse(["p", user_pubkey])
tags.append(pTag.as_vec())
search = str(search).replace("from:", "").replace("to:", "").replace("@", "").lstrip().rstrip()
print(search)
ev = nostr_client_test_search(search, tags)
ui.notify('Request sent to DVM, awaiting results..')
print("Sent: " + ev.as_json())
print(str(now.to_human_datetime()))
data.clear()
# table.clear()
listui = await run.io_bound(handledvm, now, ev.id())
ui.notify("Received results from DVM")
table.clear()
for element in listui:
table.add_rows(element)
table.visible = True
ui.update(table)
return
mainrow = ui.row().style('gap:10em').classes("row-1")
maincolumn = ui.column().classes("col-1")
with mainrow:
with maincolumn:
ui.label('Noogle Search').classes('text-2xl')
prompt = ui.input('Search').style('width: 20em')
ui.button('Search', on_click=search).style('width: 15em')
columns = [
{'name': 'result', 'label': 'result', 'field': 'result', 'sortable': True, 'align': 'left', },
{'name': 'time', 'label': 'time', 'field': 'time', 'sortable': True, 'align': 'left'},
]
table = ui.table(columns=columns, rows=data, row_key='result',
pagination={'rowsPerPage': 10, 'sortBy': 'time', 'descending': True, 'page': 1}).style(
'width: 50em')
table.add_slot('header', r'''
{{ col.label }}
''')
table.add_slot('body', r'''
{{ col.value }}
Njump
Highlighter
NoStrudel
''')
table.on('action', lambda msg: print(msg))
table.visible = False
ui.run(reload=True, port=1234)