From 795cc64097df35e92a70cf4ea30a11b6fb2c8bcf Mon Sep 17 00:00:00 2001
From: Aljaz Ceru <aljaz@ceru.si>
Date: Thu, 12 Oct 2023 11:33:51 +0200
Subject: [PATCH] proof of concept - horrible code

---
 example_settings.py |  1 +
 science-bot.py      | 60 +++++++++++++++++++++++++++++++++++++++++++++
 science.py          | 43 ++++++++++++++++++++++++++++++++
 3 files changed, 104 insertions(+)
 create mode 100644 example_settings.py
 create mode 100644 science-bot.py
 create mode 100644 science.py

diff --git a/example_settings.py b/example_settings.py
new file mode 100644
index 0000000..2ce45b8
--- /dev/null
+++ b/example_settings.py
@@ -0,0 +1 @@
+NOSTR_KEY="<hex priv key"
\ No newline at end of file
diff --git a/science-bot.py b/science-bot.py
new file mode 100644
index 0000000..85b0f0d
--- /dev/null
+++ b/science-bot.py
@@ -0,0 +1,60 @@
+from nostr_sdk import Keys, Client, Event, EventBuilder, Filter, HandleNotification, Timestamp, nip04_decrypt, SecretKey, init_logger, LogLevel
+import time
+import requests
+from settings import *
+
+
+init_logger(LogLevel.DEBUG)
+
+sk = SecretKey.from_hex(NOSTR_KEY)
+keys = Keys(sk)
+sk = keys.secret_key()
+pk = keys.public_key()
+print(f"Bot public key: {pk.to_bech32()}")
+
+client = Client(keys)
+
+client.add_relay("wss://relay.damus.io")
+client.add_relay("wss://nostr.mom")
+client.add_relay("wss://nostr.oxtr.dev")
+client.connect()
+
+filter = Filter().pubkey(pk).kind(4).since(Timestamp.now())
+client.subscribe([filter])
+
+# function to format results into title and url
+
+def format_answer(response):
+    # Extract the answer
+    formatted_response = "Short answer:\n" + response["short_answer"] + "\n\nExcerpts from articles:\n" + response["chunks"]
+    return formatted_response
+
+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}")
+                params = {'query': str(msg)}
+                try:
+                    resp = requests.get('http://127.0.0.1:6000/ask', params=params)
+                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}")
+
+    def handle_msg(self, relay_url, msg):
+        None
+        
+client.handle_notifications(NotificationHandler())
+
+while True:
+    time.sleep(5.0)
\ No newline at end of file
diff --git a/science.py b/science.py
new file mode 100644
index 0000000..eea567c
--- /dev/null
+++ b/science.py
@@ -0,0 +1,43 @@
+import logging
+import sys
+from cybrex.cybrex_ai import CybrexAI
+from quart import Quart, request, jsonify
+
+logging.basicConfig(stream=sys.stdout, level=logging.INFO)
+
+
+
+app = Quart(__name__)
+
+@app.route('/ask')
+async def ask():
+    cybrex = CybrexAI()
+    query = request.args.get('query')
+    if not query:
+        return {'error': 'Please provide a query parameter'}, 400
+
+    # Only await if cybrex.start() needs to be called for each request
+    # Otherwise, consider initializing cybrex outside the route
+    await cybrex.start()
+    
+    # Assuming chat_science is an asynchronous method, it should be awaited
+    answer = await cybrex.chat_science(query=query, n_chunks=4, n_documents=10)
+
+    payload = format_cybrex_response(answer)
+    return payload
+
+def format_cybrex_response(answer):
+    # Extract the answer
+    chunks = answer.chunks
+    reply = answer.answer
+    formatted_response = "Short answer:\n" + reply + "\n\nExcerpts from articles:\n"
+    
+    # Extract and format each chunk
+    for chunk in chunks:
+        formatted_response += "\n" + chunk.title + "\n" + chunk.text + "\n"
+    
+    return formatted_response
+
+
+if __name__ == '__main__':
+    app.run(debug=True, port=6000)