diff --git a/handyhelper/__init__.py b/handyhelper/__init__.py index a580875..ed77f1f 100644 --- a/handyhelper/__init__.py +++ b/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) @@ -42,6 +45,18 @@ async def getPDF(url): pages = pdftotext.PDF(pdf) 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} + """} + ] + ) + return completion + @bot.event async def on_ready(): print(f'We have logged in as {bot.user}') @@ -61,8 +76,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]} diff --git a/secrets/deploy.nix b/secrets/deploy.nix index 12c6b59..bcf8927 100644 --- a/secrets/deploy.nix +++ b/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"