diff --git a/deploy/flake.nix b/deploy/flake.nix index ccacc90..64e7c6d 100644 --- a/deploy/flake.nix +++ b/deploy/flake.nix @@ -16,6 +16,7 @@ wantedBy = ["networking-online.target"]; enable = true; serviceConfig = { + WorkingDirectory = "handyhelper"; ExecStart = '' ${handy_py}/bin/handyhelper ''; diff --git a/handyhelper/__init__.py b/handyhelper/__init__.py index ed77f1f..c45b8f6 100644 --- a/handyhelper/__init__.py +++ b/handyhelper/__init__.py @@ -43,6 +43,10 @@ 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): @@ -53,9 +57,13 @@ async def summarize(pdf): --- {pdf} """} - ] + ], + user="HandyHelper" ) - return completion + + eprint(completion.usage) + + return completion.choices[0].message @bot.event async def on_ready():