Update ImageGenerator.py

ImageGenerator.py

@@ -2,7 +2,7 @@
 # from PIL import Image
 # from io import BytesIO
 
-# HF_API_KEY = os.getenv("HF_API_KEY")
+HF_API_KEY = os.getenv("HF_API_KEY")
 
 # def generate_image(prompt):
 #     """Generate image using HF Inference API."""
@@ -18,13 +18,15 @@ import requests
 from PIL import Image, UnidentifiedImageError, ImageDraw, ImageFont
 from io import BytesIO
 
-def generate_image(prompt, api_url="YOUR_API_URL_HERE"):
+def generate_image(prompt):
     """
     Generate an image from a prompt via an API.
     Returns a PIL.Image object. If the API fails, returns a placeholder image.
     """
     try:
-        response = requests.post(api_url, json={"prompt": prompt})
+        api_url="https://huggingface.co/stable-diffusion-v1-5/stable-diffusion-v1-5
+        headers = {"Authorization": f"Bearer {HF_API_KEY}"}
+        response = requests.post(api_url, headers=headers, json={"inputs": prompt})
 
         # Check if the response is OK
         if response.status_code == 200: