From ffd642fd2a139eec042a69e8dedbbda06788a11c Mon Sep 17 00:00:00 2001 From: crate Date: Thu, 22 May 2025 17:09:48 -0500 Subject: [PATCH] add wiki stuff --- garfmain.py | 8 +++++++- garfpy/__init__.py | 1 + garfpy/wiki.py | 8 ++++++++ 3 files changed, 16 insertions(+), 1 deletion(-) diff --git a/garfmain.py b/garfmain.py index 2e23bbd..b445d77 100644 --- a/garfmain.py +++ b/garfmain.py @@ -7,7 +7,7 @@ from garfpy import( logger, is_private, kroger_token, find_store, search_product, picture_time, process_image_requests, generate_chat, - aod_message) + aod_message, wiki) gapikey = config.GIF_TOKEN @@ -39,6 +39,7 @@ async def on_message(message): if message.content.lower().startswith("hey garfield") or isinstance(message.channel, discord.DMChannel): question = message.content[12:] if message.content.lower().startswith("hey garfield") else message.content answer = await generate_chat(question) + logger.info(f"Chat Request - User: {user}, Server: {server}, Prompt: {question}") await message.channel.send(answer) if message.content.lower().startswith('garfpic '): @@ -49,6 +50,11 @@ async def on_message(message): await message.channel.send(f"`Please wait... image generation queued: {prompt}`") await picture_time(message, prompt) + if message.content.lower().startswith('wiki '): + search_term = message.content[5:] + summary = await wikisum(search_term) + await message.channel.send(summary) + if message.content.lower().startswith("garfping "): try: query = message.content.split() diff --git a/garfpy/__init__.py b/garfpy/__init__.py index 292e86a..b9b51aa 100644 --- a/garfpy/__init__.py +++ b/garfpy/__init__.py @@ -10,3 +10,4 @@ from .garfai import( ) from .iputils import is_private from .aod import aod_message +from wiki import wikisum \ No newline at end of file diff --git a/garfpy/wiki.py b/garfpy/wiki.py index bb375a3..4a12d57 100644 --- a/garfpy/wiki.py +++ b/garfpy/wiki.py @@ -1 +1,9 @@ import wikipedia + +async def wikisum(search_term): + try: + summary = wikipedia.summary(search_term) + return summary + + except wikipedia.exceptions.DisambiguationError as e: + return e \ No newline at end of file