From 00da414b218ff7437305ca83f2cb8129d786f606 Mon Sep 17 00:00:00 2001
From: Ben C <bwc9876@gmail.com>
Date: Tue, 4 Mar 2025 07:46:02 -0500
Subject: [PATCH] Add learning from file (owner only)

---
 src/bingus/cogs/markov.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/src/bingus/cogs/markov.py b/src/bingus/cogs/markov.py
index da4191f..6583453 100644
--- a/src/bingus/cogs/markov.py
+++ b/src/bingus/cogs/markov.py
@@ -94,6 +94,22 @@ class Markov(commands.Cog):
             else:
                 await ctx.respond(f"{head}:\n{msg}")
 
+    @require_owner
+    @commands.slash_command()
+    async def study(
+        self, ctx: discord.ApplicationContext, file: discord.Option(discord.Attachment)
+    ):
+        raw = await file.read()
+        try:
+            text = raw.decode()
+            self.markov.learn(text)
+            await ctx.respond("> Bingus learned from file!", ephemeral=True)
+            await self.update_words()
+        except UnicodeDecodeError:
+            await ctx.respond(
+                "> Bingus only understands UTF-8 text files!", ephemeral=True
+            )
+
     @commands.Cog.listener()
     async def on_ready(self):
         await self.update_words()
-- 
2.51.2