Syzygial/HandyHelper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Basic OpenAI Summarize Function

Browse Source
This commit is contained in:
DavidCrompton1192@gmail.com
2023-03-17 23:42:38 -04:00
parent 06e2b7c641
commit c5ce1629ef
2 changed files with 20 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
handyhelper/__init__.py
View File

@@ -8,6 +8,9 @@ import requests as req
from bs4 import BeautifulSoup as soup from bs4 import BeautifulSoup as soup
import pdftotext import pdftotext
import sys import sys
import openai
openai.api_key = os.getenv("OPENAI_KEY")
def eprint(*args, **kwargs): def eprint(*args, **kwargs):
print(*args, file=sys.stderr, **kwargs) print(*args, file=sys.stderr, **kwargs)
@@ -42,6 +45,18 @@ async def getPDF(url):
pages = pdftotext.PDF(pdf) pages = pdftotext.PDF(pdf)
return "\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}
"""}
]
)
return completion
@bot.event @bot.event
async def on_ready(): async def on_ready():
print(f'We have logged in as {bot.user}') print(f'We have logged in as {bot.user}')
@@ -61,8 +76,9 @@ async def summarize(
return return
msg = await interaction.followup.send(f"""Article Found: {resp['ref']} msg = await interaction.followup.send(f"""Article Found: {resp['ref']}
Parsing PDF...""") Processing PDF...""")
pdf = await getPDF(resp['pdf']) pdf = await getPDF(resp['pdf'])
await msg.edit(f"""Article Found: {resp['ref']} await msg.edit(f"""Article Found: {resp['ref']}
``` ```
{pdf[:1000]} {pdf[:1000]}

6
secrets/deploy.nix
View File

@@ -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": { "sops": {
"kms": null, "kms": null,
"gcp_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" "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", "lastmodified": "2023-03-18T03:32:23Z",
"mac": "ENC[AES256_GCM,data:iJLr2cw8KQuPSnngHeNy3LU/IEwOq77kNoOyaanQwu6yTz3IyqA7T6HKl94k264sazMjq6Gz5FE+dqM2wwabMEKrLmtoqrtic9GMKZfGyUlY5TcPrZrvS/+3V3To4DN3NIRmw4XFUm7aGHBN0MzXV681NhJZlbh3Qxh6ASwV7xs=,iv:TJD7FzHbmJNpV5gEdwIzmXu7Hqw4bzGQNUxZw+yZ2Vg=,tag:2EbL84mIMtO24oRd1YVEWQ==,type:str]", "mac": "ENC[AES256_GCM,data:cajh/+/3udzyWXaUMgcshsfsZ9WnLOCuFu3g/Dbu3riEARGuDf7vWEOFnjJbOvBwyXJ3Vcq7KaoB9x7vKC7vnNs982ffuvhxZTHyNEOs21UZB5IGszj0UCuXnHQJ+qMXVGYbFRU00KXh8e3bMwEe71fPq1iLLLFx82dllNl4Lng=,iv:oAvZNEYWASq+2v0O6ZP56EVI9YrjTeaDhUpcWEGiCE8=,tag:+E0fz422Axr996NBzcm4tg==,type:str]",
"pgp": null, "pgp": null,
"unencrypted_suffix": "_unencrypted", "unencrypted_suffix": "_unencrypted",
"version": "3.7.3" "version": "3.7.3"