Adding logging of input/output to Notion

app.py

@@ -3,6 +3,7 @@ from scipy.special import expit
 import numpy as np
 import os
 import gradio as gr
+import requests
 
 # set up model
 authtoken = os.environ.get("TOKEN") or True
@@ -27,9 +28,50 @@ def predict(text):
     harm = sorted(harm, key=lambda x: x[1])[::-1][:1]
     # top is the combo of these
     top = issues + feelings + harm
+
+    logToNotion(text, top)
+    
     d = {i: j for i, j in top}
     return d
 
+def logToNotion(text, top):
+    url = "https://api.notion.com/v1/pages"
+
+    payload = {
+        "properties": {
+            "title": {
+                "title": [{
+                    "text": {
+                        "content": "."
+                    }
+                }]
+            },
+            "input": {
+                "rich_text": [{
+                    "text": {
+                        "content": text
+                    }
+                }]
+            },
+            "output": {
+                "rich_text": [{
+                    "text": {
+                        "content": ", ".join(str(x) for x in top)
+                    }
+                }]
+            }
+        },
+        "parent": "4a220773ac694851811e87f4571ec41d"
+    }
+    headers = {
+        "Accept": "application/json",
+        "Notion-Version": "2022-02-22",
+        "Content-Type": "application/json",
+        "Authorization": "Bearer secret_lc3Weq9s87EnkNWk2Yibko91hhLuNQIEzSoeAAoriUz"
+    }
+    
+    response = requests.post(url, json=payload, headers=headers)
+
 iface = gr.Interface(
   fn=predict, 
   inputs="text",