Support for horde api

Dockerfile

@@ -12,7 +12,6 @@ RUN apt-get install wget -y
 RUN apt-get install ffmpeg -y
 
 COPY ./requirements.txt /discordbot/requirements.txt
-RUN pip install python-dotenv typing-extensions
 RUN pip install -U -r /discordbot/requirements.txt
 
 COPY ./discord_bot.py /discordbot/discord_bot.py

discord_bot.py

@@ -2,20 +2,60 @@
 # https://discordpy.readthedocs.io/en/stable/quickstart.html#a-minimal-bot
 
 import os
-
 import discord
-
+from discord import app_commands
 from discord.ext import commands
-
 from threading import Thread
-
 import json
+import horde
 
 
 
 intents = discord.Intents.default()
 intents.message_content = True
 bot = commands.Bot(command_prefix='>', intents=intents)
+tree = bot.tree
+
+@tree.command(name="hello", description="Sends a greeting!")
+async def hello(interaction: discord.Interaction):
+    await interaction.response.send_message(f"Hello, {interaction.user.mention}!")
+
+@tree.command(name="greet", description="Greets the specified user")
+async def greet(interaction: discord.Interaction, name: str):
+    await interaction.response.send_message(f"Hello, {name}!")
+
+@tree.command(name="get kudos", description="The amount of Kudos this user has.")
+async def getKudos(interaction: discord.Interaction):
+    details = horde.getUserDetails()
+    if "kudos" not in details:
+        await interaction.response.send_message(f'Error: {details["code"]} {details["reason"]}')
+        return
+    await interaction.response.send_message(f'The amount of Kudos this user has is {details["kudos"]}')
+
+@tree.command(name="generate status", description="Retrieve the status of an Asynchronous generation request.")
+async def generateStatus(interaction: discord.Interaction, id: str):
+    details = horde.generateCheck(id)
+    if "kudos" not in details:
+        await interaction.response.send_message(f'Error: {details["code"]} {details["reason"]}')
+        return
+    if bool(details["is_possible"]) == False:
+        await interaction.response.send_message("This generation is impossible.")
+        return
+    if bool(details["faulted"]) == True:
+        await interaction.response.send_message("This generation is faulted.")
+        return
+    if bool(details["done"]) == True:
+        generationDetail = horde.generateStatus(id)
+        if "generations" not in generationDetail:
+            await interaction.response.send_message(f'Error: {details["code"]} {details["reason"]}')
+        for i in range(len(generationDetail["generations"])):
+            await interaction.response.send_message(generationDetail["generations"][i]["img"])
+        return
+    if int(details["processing"]) > 0:
+        total = int(details["finished"]) + int(details["processing"]) + int(details["queue_position"]) + int(details["restarted"]) + int(details["waiting"])
+        await interaction.response.send_message(f'Processing image: {details["processing"]}/{total}')
+        return
+    await interaction.response.send_message(f'Position in queue: {details["queue_position"]}, wait time: {details["wait_time"]}s')
 
 @bot.command()
 async def ping(ctx):

horde.py

@@ -0,0 +1,46 @@
+import os
+import requests
+
+hordeApiKey = os.environ.get("HORDE_API_KEY")
+baseUrl = "https://aihorde.net/api/v2/"
+
+def getResponseByGet(url):
+    headers = {
+        "Content-Type": "application/json",
+        "apikey": f"{hordeApiKey}"
+    }
+    response = requests.get(url, headers=headers)
+    if response.status_code == 200:
+        return response.json()
+    else:
+        return {
+            "code": response.status_code,
+            "reason": response.reason
+        }
+
+def getResponseByPost(url):
+    headers = {
+        "Content-Type": "application/json",
+        "apikey": f"{hordeApiKey}"
+    }
+    response = requests.post(url, headers=headers)
+    if response.status_code == 200:
+        return response.json()
+    else:
+        return {
+            "code": response.status_code,
+            "reason": response.reason
+        }
+
+def getUserDetails():
+    url = f"{baseUrl}find_user"
+    return getResponseByPost(url)
+
+def generateCheck(id):
+    url = f"{baseUrl}generate/check/{id}"
+    return getResponseByGet(url)
+
+def generateStatus(id):
+    url = f"{baseUrl}generate/status/{id}"
+    return getResponseByGet(url)
+

requirements.txt

@@ -1,3 +1,4 @@
+python-dotenv
 flask[async]
 typing-extensions
 quart
@@ -5,3 +6,6 @@ discord
 discord.py
 urllib3==1.26.5
 requests
+py-cord
+discord-py-interactions
+nextcord