functions.py

import discord
import os
import json
import requests
import random
import praw
import datetime
import urllib3

from urllib.request import urlopen
from dotenv import load_dotenv

load_dotenv()

# Secrets
API_NINJA_KEY = os.getenv("API_NINJA_KEY")  # Used for Fun facts, dad jokes, and jokes
REDDIT_CLIENT_ID = os.getenv("REDDIT_CLIENT_ID")  # Used for Python Reddit API Wrapper
REDDIT_CLIENT_SECRET = os.getenv(
    "REDDIT_CLIENT_SECRET"
)  # Used for Python Reddit API Wrapper
REDDIT_USER_AGENT = os.getenv("REDDIT_USER_AGENT")  # Used for Python Reddit API Wrapper
BOT_TOKEN = os.getenv("BOT_TOKEN")  # Discord Bot Token
DOPPLE_ID = os.getenv("DOPPLE_ID")  # Dopple ID
DOPPLE_CHAT_ID = os.getenv("DOPPLE_CHAT_ID")  # Dopple chat ID
DOPPLE_USERNAME = os.getenv("DOPPLE_USERNAME")  # Dopple registered username
DOPPLE_COOKIE = os.getenv("DOPPLE_COOKIE")  # Dopple cookie

# Customisable values:
LOG_FILE = (
    "exhaust.txt"  # Name of the log file. Keep blank to disable logging to a file
)
embed_colour = [8, 234, 142]  # R,G,B
error_embed_colour = [250, 0, 0]  # R, G, B
memesubs = [  # Subreddits to get a meme from
    "memes",
    "dankmemes",
    "196",
    "surrealmemes",
]

# Initialise PRAW with account information, used for accessing reddit API.
reddit = praw.Reddit(
    client_id=REDDIT_CLIENT_ID,
    client_secret=REDDIT_CLIENT_SECRET,
    user_agent=REDDIT_USER_AGENT,
    check_for_async=False,
)

# Disable warnings for unverified HTTP requests
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)

# <-------ALL THE FUNCTIONS USED BY THE BOT------->


# Function to return an error embed with desired content
def error(title, description):
    erroremb = discord.Embed(
        title=title,
        description=description,
        colour=discord.Color.from_rgb(
            error_embed_colour[0], error_embed_colour[1], error_embed_colour[2]
        ),
    )
    return erroremb


# Function to log stuff, optionally to a text file
def log(text):
    logtext = f"[ {str(datetime.datetime.now())} ] {text}"
    print(logtext)
    if LOG_FILE:
        with open(
            LOG_FILE, "a", encoding="utf-8"
        ) as text_file:  # Exhaust.txt is the name of the log file
            text_file.write(logtext)


def imageGen(prompt: str) -> bytes:
    log(f"[IMAGE GEN]: Prompt: {prompt}\n")

    payload = f'-----011000010111000001101001\r\nContent-Disposition: form-data; name="prompt"\r\n\r\n{prompt}\r\n-----011000010111000001101001\r\nContent-Disposition: form-data; name="output_format"\r\n\r\nbytes\r\n-----011000010111000001101001\r\nContent-Disposition: form-data; name="user_is_subscribed"\r\n\r\ntrue\r\n-----011000010111000001101001--\r\n'

    headers = headers = {
        "accept": "application/json, text/plain, */*",
        "content-type": "multipart/form-data; boundary=---011000010111000001101001",
        "origin": "https://magicstudio.com",
        "referer": "https://magicstudio.com/",
        "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36",
    }

    image = requests.request(
        "POST",
        "https://ai-api.magicstudio.com/api/ai-art-generator",
        data=payload,
        headers=headers,
    )

    if image.status_code != 200:
        log(f"[AI IMAGE]: ERROR {image.status_code} \n")
        log(f"[AI IMAGE]: ERROR {image.text} \n")
        return ""

    return image.content


# Define a function to return ai output on a given input
def gpt(prmpt: str):
    print(f"[PROMPT]: {prmpt}")

    url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent?key={os.getenv('GEMINI_API_KEY')}"
    payload = {
        "contents": [
            {
                "role": "user",
                "parts": [
                    {
                        "text": f'{os.getenv("GEMINI_PROMPT")} In case you need this info, today is {datetime.datetime.now().strftime("%Y-%m-%d")}.'
                    }
                ],
            },
            {"role": "user", "parts": [{"text": prmpt}]},
        ]
    }
    response = requests.post(url, json=payload)

    try:
        response_text = (
            response.json()
            .get("candidates")[0]
            .get("content")
            .get("parts")[0]
            .get("text")
        )
        if response_text == "":
            response_text = "Something went wrong, try again later"
            raise Exception(response_text)
        log(f"[AI]: {response_text}")
        return response_text
    except Exception as e:
        log(f"Error while getting AI response: {response.text}")
        log(e)
        return e


# Function to make an embed with the best posts of a given subreddit
def meme(posts_lim):
    subreddit = random.choice(memesubs)

    # subreddit - What subreddit
    # posts_lim - How many top posts to choose from

    # Get some top posts from a subreddit to choose from
    submissions = list(reddit.subreddit(subreddit).hot(limit=posts_lim))

    # Get a random post from the chosen top posts and reroll if post is nsfw
    post = submissions[random.randrange(-1, posts_lim)]

    # Read the post JSON
    json_url = "https://www.reddit.com" + post.permalink + ".json"
    json_data = urlopen(json_url).read().decode("utf-8")
    post_json = json.loads(json_data)
    # Get image or gif embed, if any.
    try:
        # Parse the json for the image link
        img_link = post_json[0]["data"]["children"][0]["data"]["url_overridden_by_dest"]
        img_extension = img_link[
            -5:
        ]  # Get last 5 characters of the link which contains the file's extension
    except KeyError:
        log("[REDDIT POST]: No media found, retrying...")
        return meme(subreddit, posts_lim)
    if post.over_18:
        log("[REDDIT POST]: NSFW Post, skipping")
        return meme(subreddit, posts_lim)
    else:
        if any(
            extension in img_extension for extension in ("png", "jpg", "jpeg", "gif")
        ):
            # Create embed with post title + subreddit name
            log("[REDDIT POST]: Title: " + post.title + " on r/" + subreddit)
            embed = discord.Embed(
                title=post.title,
                description="on r/" + subreddit,
                color=discord.Color.from_rgb(
                    embed_colour[0], embed_colour[1], embed_colour[2]
                ),
            )
            embed.set_image(url=img_link)
        else:
            log("[REDDIT POST]: Media not in a suitable format, retrying...")
            return meme(subreddit, posts_lim)

    # Return the embed
    return embed


# Function to return an embed with cat pics
def meow():
    catresponse = requests.get("https://api.thecatapi.com/v1/images/search")
    catdata = catresponse.json()
    catimg = url = catdata[0]["url"]
    catembed = discord.Embed(
        title="Meow 🐈",
        colour=discord.Color.from_rgb(
            embed_colour[0], embed_colour[1], embed_colour[2]
        ),
    )
    catembed.set_image(url=catimg)
    log("[CAT IMAGE]: URL: " + url + "\n")
    return catembed


# Function to return an embed with dog pics
def woof():
    dogresponse = requests.get("https://random.dog/woof.json")
    dogdata = dogresponse.json()
    dogimg = dogdata["url"]
    if any(extension in dogimg[-5:] for extension in ("png", "jpg", "jpeg", "gif")):
        dogembed = discord.Embed(
            title="Woof 🐕",
            colour=discord.Color.from_rgb(
                embed_colour[0], embed_colour[1], embed_colour[2]
            ),
        )
        dogembed.set_image(url=dogimg)
        log("[DOG IMAGE]: URL: " + dogimg + "\n")
    else:
        log(
            "[DOG IMAGE]: Returned media URL not in suitable format. Skipping... URL: "
            + dogimg
            + "\n"
        )
        return woof()
    return dogembed


# Function to return an embed with a joke
def joke():
    api_url = "https://api.api-ninjas.com/v1/jokes"
    jokeresponse = requests.get(api_url, headers={"X-Api-Key": API_NINJA_KEY})
    if jokeresponse.status_code == requests.codes.ok:
        jokedata = jokeresponse.json()
        joke = jokedata[0]["joke"]
        jokeembed = discord.Embed(
            title="Joke 🤣",
            description=joke,
            colour=discord.Color.from_rgb(
                embed_colour[0], embed_colour[1], embed_colour[2]
            ),
        )
        log("[JOKE]: " + joke + "\n")
        return jokeembed
    else:
        jokeembed = discord.Embed(
            title="No joke 😔",
            description="Error getting joke, try again later",
            colour=discord.Color.from_rgb(
                error_embed_colour[0], error_embed_colour[1], error_embed_colour[2]
            ),
        )
        log(f"[JOKE]: ERROR: {jokeresponse.status_code} {jokeresponse.text}\n")
    return jokeembed


# Function to return an embed with a dad joke
def dadjoke():
    api_url = "https://api.api-ninjas.com/v1/dadjokes"
    dadresponse = requests.get(api_url, headers={"X-Api-Key": API_NINJA_KEY})
    if dadresponse.status_code == requests.codes.ok:
        daddata = dadresponse.json()
        dadjoke = daddata[0]["joke"]
        dadembed = discord.Embed(
            title="Dad joke 😁",
            description=dadjoke,
            colour=discord.Color.from_rgb(
                embed_colour[0], embed_colour[1], embed_colour[2]
            ),
        )
        log("[DAD JOKE]: " + dadjoke + "\n")
    else:
        dadembed = discord.Embed(
            title="No dad joke 😔",
            description="Error getting dad joke, try again later",
            colour=discord.Color.from_rgb(
                error_embed_colour[0], error_embed_colour[1], error_embed_colour[2]
            ),
        )
        log(f"[DAD JOKE]: ERROR: {dadresponse.status_code} {dadresponse.text}\n")
    return dadembed


# Function to return an embed with a fact
def fact():
    api_url = "https://api.api-ninjas.com/v1/facts"
    factresponse = requests.get(api_url, headers={"X-Api-Key": API_NINJA_KEY})
    if factresponse.status_code == requests.codes.ok:
        factdata = factresponse.json()
        fact = factdata[0]["fact"]
        factembed = discord.Embed(
            title="Fun fact! 🤯",
            description=fact,
            colour=discord.Color.from_rgb(
                embed_colour[0], embed_colour[1], embed_colour[2]
            ),
        )
        log("[FACT: " + fact)
    else:
        factembed = discord.Embed(
            title="Not so fun fact",
            description="Error getting fun fact, try again later",
            colour=discord.Color.from_rgb(
                error_embed_colour[0], error_embed_colour[1], error_embed_colour[2]
            ),
        )
        log(f"[FACT]: ERROR: {factresponse.status_code} {factresponse.text}\n")
    return factembed