Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -13,15 +13,11 @@ from discord import Permissions
|
|
13 |
from discord.ext import commands
|
14 |
from discord.utils import oauth_url
|
15 |
from gradio_client.utils import QueueError
|
16 |
-
import base64
|
17 |
-
import io
|
18 |
|
19 |
event = Event()
|
20 |
|
21 |
DISCORD_TOKEN = os.getenv("DISCORD_TOKEN")
|
22 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
23 |
-
API_URL = "https://api-inference.huggingface.co/models/prompthero/openjourney"
|
24 |
-
headers = {"Authorization": "Bearer " + HF_TOKEN}
|
25 |
|
26 |
def ask(input_text):
|
27 |
api_url = "https://somerandomapifor-disorc-cause-i-need-it.onrender.com/qa"
|
@@ -71,23 +67,6 @@ async def ai(ctx, *, input_text: str):
|
|
71 |
# Reply with the embed
|
72 |
await ctx.reply(result)
|
73 |
|
74 |
-
def query(payload):
|
75 |
-
response = requests.post(API_URL, headers=headers, json=payload)
|
76 |
-
return response.content
|
77 |
-
|
78 |
-
@bot.command()
|
79 |
-
async def img(ctx, *, prompt: str):
|
80 |
-
# Query the API to generate the image
|
81 |
-
async with ctx.typing():
|
82 |
-
image_bytes = query({
|
83 |
-
"inputs": prompt,
|
84 |
-
})
|
85 |
-
timestamp = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
|
86 |
-
filename = f"generated_image_{timestamp}.png"
|
87 |
-
image_file = discord.File(io.BytesIO(image_bytes), filename=filename)
|
88 |
-
# Send the image as a file
|
89 |
-
await ctx.reply(content=prompt, file=image_file)
|
90 |
-
|
91 |
|
92 |
@bot.command()
|
93 |
async def verse(ctx):
|
|
|
13 |
from discord.ext import commands
|
14 |
from discord.utils import oauth_url
|
15 |
from gradio_client.utils import QueueError
|
|
|
|
|
16 |
|
17 |
event = Event()
|
18 |
|
19 |
DISCORD_TOKEN = os.getenv("DISCORD_TOKEN")
|
20 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
|
|
|
|
21 |
|
22 |
def ask(input_text):
|
23 |
api_url = "https://somerandomapifor-disorc-cause-i-need-it.onrender.com/qa"
|
|
|
67 |
# Reply with the embed
|
68 |
await ctx.reply(result)
|
69 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
70 |
|
71 |
@bot.command()
|
72 |
async def verse(ctx):
|