Test Write PDF Text

deploy/flake.nix

@@ -16,6 +16,7 @@
             wantedBy = ["networking-online.target"];
             enable = true;
             serviceConfig = {
+              WorkingDirectory = "handyhelper";
               ExecStart = ''
                 ${handy_py}/bin/handyhelper
               '';

handyhelper/__init__.py

@@ -8,6 +8,9 @@ import requests as req
 from bs4 import BeautifulSoup as soup
 import pdftotext
 import sys
+import openai
+
+openai.api_key = os.getenv("OPENAI_KEY")
 
 def eprint(*args, **kwargs):
     print(*args, file=sys.stderr, **kwargs)
@@ -40,8 +43,28 @@ async def getPDF(url):
     resp = req.get(url, stream=True)
     pdf = io.BytesIO(resp.content)
     pages = pdftotext.PDF(pdf)
+
+    with open("./test.txt","w") as f:
+        f.write("\n\n".join(pages))
+    
     return "\n\n".join(pages)
 
+async def summarize(pdf):
+    completion = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=[
+            {"role": "user", "content": f"""Please summarize the following research paper:
+            ---
+            {pdf}
+            """}
+        ],
+        user="HandyHelper"
+    )
+
+    eprint(completion.usage)
+    
+    return completion.choices[0].message
+
 @bot.event
 async def on_ready():
     print(f'We have logged in as {bot.user}')
@@ -61,8 +84,9 @@ async def summarize(
         return
 
     msg = await interaction.followup.send(f"""Article Found: {resp['ref']}
-    Parsing PDF...""")
+Processing PDF...""")
     pdf = await getPDF(resp['pdf'])
+    
     await msg.edit(f"""Article Found: {resp['ref']}
     ```
     {pdf[:1000]}

secrets/deploy.nix

@@ -1,5 +1,5 @@
 {
-	"data": "ENC[AES256_GCM,data:jLGfwTHJLm0+33OijRRT+l1B4N4mcwxjKj95rcclTf/+qh3fC5COt+PJnkVIfc4rlfraATjEk3OwWeEYabZmIhslpe1T6CP4dgRMQHDh4FXLIWmgu07XVFMWYdiOxxiTXiSy6X6K6C1OQ6EYTQb+werKk4CYBYtuD3xzr+sQswnxiLlpI0BOmxm3xCGAXDK/V/FKb3mA5LSHnXjxNARaLtZBcXDjU4mnJN95yUQyY89kaS6OQzo2TUA7Z04T4xvW5/uGZdtrYTlF4mK5Z9/fDnXNDsHM8rvrfLKvAv0ctN0Pj5Zr/Vex58Ofafb6qcuqwuAlNM8gct7OL7t4tBvBGtJ81LrMS+VeV5h2+0M2hFbPldLu+tcjgUso/FVwwrKD7gaxkF8Ev+CyZBGmPY7ANti8z0+jkpLdkwy758mem9IRIadWz27qaNqz4vdO68bp3C872/knYDPCD6av00p9fg4hPOTDFj2hiO0=,iv:NmaPl/ERtLLabR+65d6UbP7oUXax5R714WgxXolr9Ic=,tag:eS9m77L3K/IsCjrb42MqJQ==,type:str]",
+	"data": "ENC[AES256_GCM,data:wepRFTHhHt4vBz1ZTNUB0m4ECBvHEf7U7sCBNptuxzMX3xpk2Hrk9Cde4OT02UdNnaTSz0h4OANNh/2945CbRl87sFhR2/DY+P21LD240lhbWCgrbSUKgHjI0d+PyC0npOCv2+2cglJn+OsgYe4rrZlA8g6AbRejTJjnc5+jIDkj9VMX2TZCATuIRg2+8aHYr4tNq5Mfd4Y0OnF+VmQH/SDeap1YeUhh1l/6UEiuUrB4RvH+oNtZaDdDLI4Cg6sKXAeFcx8jejtYu59NDv4TiFi6EDZ4/WflWCvTuUqmG7qv8aBcKCK2tIfLPM6POD614I76v/suelZ+pSP0ij4UfC374tIB3CcO3v6MJpj3gwHz9+fomuYq4TC1g5moaVO2OU7oGfEIo5kKCeJpQX8o+qUjwZzpvP0y906Ow7tpBKZr2Se/w1k+IL0ZfzEtRhqN84M/U17pk+Yo0X5DzwLvPRc8rrUMmUNAUqv7W4cN13zakXaWhBBSkVwRnW2raJnW+u5Gd9l4UYz0QY1SpItC49Rm5zOsFWwt92BjV+LLT8pIiRsdDkevTuGuHVAyK3PAixNMPqMJ,iv:5gCuJnHiRVqb8wLgTgY74DbH6CYpmcXIlYBvne7sYF0=,tag:7eVarvHuNk3y3zHc8sDjTQ==,type:str]",
 	"sops": {
 		"kms": null,
 		"gcp_kms": null,
@@ -15,8 +15,8 @@
 				"enc": "-----BEGIN AGE ENCRYPTED FILE-----\nYWdlLWVuY3J5cHRpb24ub3JnL3YxCi0+IFgyNTUxOSBtVWpxVTB3N2xpcXpDWlhI\naHM1TW9LL1FnT2lkRGF2bFM5VEVkNzRCV1djCkFqcmRLeWhYRkE0NzkvU00xcGdG\nYlpFOTlzWm9GS1BERi93QkxzOTJVN0EKLS0tIHpOYnlqMW9zaFpWaTJza2EzV1Jx\nOE1CUENlZlVKbGt6dUJUdUJuL1JmclkKFXf8F6bNDNDRpo+0X2MJKxAFls1vp/tr\n09dMExuNCHwJayKXlIL5+3OMDFTrjz49gZFQpWaOgQUbCVpI2Tff/A==\n-----END AGE ENCRYPTED FILE-----\n"
 			}
 		],
-		"lastmodified": "2023-03-18T01:49:54Z",
-		"mac": "ENC[AES256_GCM,data:iJLr2cw8KQuPSnngHeNy3LU/IEwOq77kNoOyaanQwu6yTz3IyqA7T6HKl94k264sazMjq6Gz5FE+dqM2wwabMEKrLmtoqrtic9GMKZfGyUlY5TcPrZrvS/+3V3To4DN3NIRmw4XFUm7aGHBN0MzXV681NhJZlbh3Qxh6ASwV7xs=,iv:TJD7FzHbmJNpV5gEdwIzmXu7Hqw4bzGQNUxZw+yZ2Vg=,tag:2EbL84mIMtO24oRd1YVEWQ==,type:str]",
+		"lastmodified": "2023-03-18T03:32:23Z",
+		"mac": "ENC[AES256_GCM,data:cajh/+/3udzyWXaUMgcshsfsZ9WnLOCuFu3g/Dbu3riEARGuDf7vWEOFnjJbOvBwyXJ3Vcq7KaoB9x7vKC7vnNs982ffuvhxZTHyNEOs21UZB5IGszj0UCuXnHQJ+qMXVGYbFRU00KXh8e3bMwEe71fPq1iLLLFx82dllNl4Lng=,iv:oAvZNEYWASq+2v0O6ZP56EVI9YrjTeaDhUpcWEGiCE8=,tag:+E0fz422Axr996NBzcm4tg==,type:str]",
 		"pgp": null,
 		"unencrypted_suffix": "_unencrypted",
 		"version": "3.7.3"