jarvis

1 files changed, 103 insertions, 0 deletions

diff --git a/cogs/jarvis.py b/cogs/jarvis.py
new file mode 100644
index 0000000..0f4606b
--- /dev/null
+++ b/cogs/jarvis.py
@@ -0,0 +1,103 @@
+import re
+import discord
+from discord.ext import commands
+import requests
+import json
+import logging
+
+from cogs.compression import Compression
+from cogs.downloader import Downloader
+
+logger = logging.getLogger(__name__)
+
+
+class Jarvis(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+
+    @commands.Cog.listener()
+    async def on_message(self, message):
+        print(message.content)
+        if message.author == self.bot.user:
+            return
+
+        if message.mentions:
+            if message.mentions[0].id == self.bot.user.id:
+                await self.jarvis(message)
+
+        if "jarvis" in message.content.lower():
+            await self.jarvis(message)
+
+    @staticmethod
+    async def jarvis(message):
+        url = "https://openrouter.ai/api/v1/chat/completions"
+        headers = {
+            "Authorization": f"Bearer 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",
+            "Content-Type": "application/json"
+        }
+        payload = {
+            "model": "mistralai/devstral-2512:free",
+            "messages": [
+                {
+                    "role": "system",
+                    "content": "respond in JSON format ONLY.\n"
+                               "your task is to tell the program what to do and how to respond after receiving a message.\n"
+                               "in the json, only fields need to be filled out, message and action\n"
+                               "message is the response that the program will send the user, "
+                               "and action is the action the program will do\n"
+                               "action can either be: compress or download.\n"
+                               "message field has to be filled out in polish.\n"
+                               "act like jarvis, be a personal assistant to user, do not use emojis"
+                },
+                {
+                    "role": "user",
+                    "content": f"{message.content}"
+                }
+            ]
+        }
+
+        response = requests.post(url, headers=headers, json=payload)
+        # Parse the JSON response
+        response_data = response.json()
+        logger.info(response_data)
+        # Extract the content string
+        content = response_data['choices'][0]['message']['content']
+
+        # Remove the code block markers and parse the inner JSON
+        inner_json_str = content.strip('```json\n').strip('```').strip()
+        inner_data = json.loads(inner_json_str)
+
+        # Now you can access the message and action
+        response_message = inner_data['message']  # "Proszę podać plik do kompresji lub link do pobrania."
+        action = inner_data['action']  # "none OR download OR compress"
+        channel = message.channel
+        logger.info(response_message)
+        logger.debug(action)
+
+        await channel.send(response_message)
+
+        if action == "download":
+            # Extract the URL from the message using regex
+            url_match = re.search(r'(https?://\S+)', message.content).group(0)
+            if not url_match:
+                await channel.send("Nie znaleziono poprawnego linku w odpowiedzi.")
+                return
+
+            response = await Downloader.download_standalone(url=url_match)
+            if isinstance(response, dict):
+                if response["status"] == "success":
+                    if response["type"] == "url":
+                        await channel.send(f"pobioron i wysyłon: {response['content']}")
+                    elif response["type"] == "file":
+                        file_data = response["content"]
+                        await channel.send("wysyłon:", file=discord.File(file_data))
+                else:
+                    logger.error(f"Error: {response['content']}")
+                    if 'details' in response:
+                        logger.error(f"Details: {response['details']}")
+
+            elif action == "compress":
+                pass
+
+def setup(bot):
+    bot.add_cog(Jarvis(bot))
\ No newline at end of file