aljaz/sciencestr
1
0
Fork 0
mirror of https://github.com/aljazceru/sciencestr.git synced 2025-12-17 05:54:24 +01:00
Code Issues Packages Projects Releases Wiki Activity

slight improvements, moving to short answer, adding docker compose

Browse Source
This commit is contained in:
Aljaz Ceru
2023-10-28 20:27:32 +02:00
parent 4030f2075d
commit 439c069b86
3 changed files with 72 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
science-bot.py
View File

@@ -2,7 +2,7 @@ from nostr_sdk import Keys, Client, Event, EventBuilder, Filter, HandleNotificat
import time
import requests
from settings import *
import sqlite3
init_logger(LogLevel.DEBUG)
@@ -17,6 +17,12 @@ client = Client(keys)
client.add_relay("wss://relay.damus.io")
client.add_relay("wss://nostr.mom")
client.add_relay("wss://nostr.oxtr.dev")
client.add_relay("wss://purplepag.es")
client.add_relay("wss://a.nos.lol")
client.add_relay("wss://nostr-01.bolt.observer")
client.add_relay("wss://e.nos.lol")
client.add_relay("wss://nostr-pub.semisol.dev")
client.add_relay("wss://relay.snort.social")
client.connect()
filter = Filter().pubkey(pk).kind(4).since(Timestamp.now())
@@ -29,25 +35,37 @@ def format_answer(response):
formatted_response = "Short answer:\n" + response["short_answer"] + "\n\nExcerpts from articles:\n" + response["chunks"]
return formatted_response
# function to interact with local database
def interact_with_db(event_id, action):
conn = sqlite3.connect('events.db')
c = conn.cursor()
if action == "check":
c.execute("SELECT * FROM events WHERE id=?", (str(event_id),))
return c.fetchone() is not None
elif action == "insert":
c.execute("INSERT INTO events VALUES (?)", (str(event_id),))
conn.commit()
conn.close()
class NotificationHandler(HandleNotification):
def handle(self, relay_url, event):
print(f"Received new event from {relay_url}: {event.as_json()}")
if event.kind() == 4:
print("Decrypting event")
try:
msg = nip04_decrypt(sk, event.pubkey(), event.content())
print(f"Received new msg: {msg}")
print(f"Received new question: {msg}")
params = {'query': str(msg)}
try:
resp = requests.get('http://127.0.0.1:6000/ask', params=params)
if resp.status_code == 200:
print("Sending answer")
event = EventBuilder.new_encrypted_direct_msg(keys, event.pubkey(), f"{resp.text}", event.id()).to_event(keys)
client.send_event(event)
else:
print(f"Error querying API: status code: {resp.status_code}, response: {resp.t}")
except Exception as e:
print(f"Error during request: {e}")
if resp.status_code == 200:
print("Sending answer")
event = EventBuilder.new_encrypted_direct_msg(keys, event.pubkey(), f"{resp.text}", event.id()).to_event(keys)
client.send_event(event)
else:
print(f"Error during content decryption: {resp.text}")
except Exception as e:
print(f"Error during content decryption: {e}")