Update app.py

app.py

@@ -51,13 +51,25 @@ class MyClient(discord.Client):
             self.is_processing = False
 
 async def generate_image(prompt):
-    """Generate an image using the Stable Diffusion model."""
-    response = hf_client(text=prompt)
-    image_data = response['image'][0]  # Assuming the response contains image data
-    image_bytes = io.BytesIO(image_data)
-    image = Image.open(image_bytes)
-    image.save("output.png")
-    return "output.png"
+    """
+    Generate an image using the Stable Diffusion model from Hugging Face.
+    """
+    try:
+        # Using the Inference API correctly
+        response = hf_client(inputs={"prompt": prompt}, parameters={"num_images": 1}, model_id="stabilityai/stable-diffusion-3-medium")
+        image_data = response['images'][0]  # Assuming the response contains image data
+
+        # Save image data to a file
+        image_bytes = io.BytesIO(base64.b64decode(image_data))
+        image = Image.open(image_bytes)
+        image_path = "output.png"
+        image.save(image_path)
+        return image_path
+
+    except Exception as e:
+        logging.error(f"Failed to generate image: {str(e)}")
+        return None
+
 
 async def send_image(channel, image_path):
     """Send an image to the specified Discord channel."""