Update app.py

app.py

@@ -21,10 +21,6 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
 headers = {"Authorization": "Bearer " + HF_TOKEN}
 
-def query(payload):
-    response = requests.post(API_URL, headers=headers, json=payload)
-    return response.content
-
 def ask(input_text):
     api_url = "https://somerandomapifor-disorc-cause-i-need-it.onrender.com/qa"
     params = {'question': input_text}
@@ -73,16 +69,24 @@ async def ai(ctx, *, input_text: str):
     # Reply with the embed
     await ctx.send(result)
 
+def query(prompt):
+    payload = {
+        "prompt": prompt,
+        "num_return_sequences": 1
+    }
+    response = requests.post(API_URL, headers=headers, json=payload)
+    return response.content
+
 @bot.command()
 async def img(ctx, *, prompt: str):
-    """Create an image with description using AI."""
-    # Your payload for the image generation (modify as needed)
-    async with ctx.typing():
-      image_data = query(prompt)
-    
-    # Send the image as a file
-    await ctx.send(file=discord.File(image_data, filename="generated_image.png"))
+    # Query the API to generate the image
+    image_data = query(prompt)
 
+    # Encode image data as base64
+    image_b64 = base64.b64encode(image_data).decode('utf-8')
+
+    # Send the image as a file
+    await ctx.send(file=discord.File(base64.b64decode(image_b64), filename="generated_image.png"))
 
 
 @bot.command()