Update ImageGenerator.py

ImageGenerator.py

@@ -1,15 +1,59 @@
-import requests, os
-from PIL import Image
-from io import BytesIO
+# import requests, os
+# 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."""
+# def generate_image(prompt):
+#     """Generate image using HF Inference API."""
     
-    url = "https://huggingface.co/stable-diffusion-v1-5/stable-diffusion-v1-5" #"https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2"
-    headers = {"Authorization": f"Bearer {HF_API_KEY}"}
+#     url = "https://huggingface.co/stable-diffusion-v1-5/stable-diffusion-v1-5" #"https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2"
+#     headers = {"Authorization": f"Bearer {HF_API_KEY}"}
+
+#     response = requests.post(url, headers=headers, json={"inputs": prompt})
+
+#     return Image.open(BytesIO(response.content))
+
+import requests
+from PIL import Image, UnidentifiedImageError, ImageDraw, ImageFont
+from io import BytesIO
 
-    response = requests.post(url, headers=headers, json={"inputs": prompt})
+def generate_image(prompt, api_url="YOUR_API_URL_HERE"):
+    """
+    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})
 
-    return Image.open(BytesIO(response.content))
+        # Check if the response is OK
+        if response.status_code == 200:
+            try:
+                img = Image.open(BytesIO(response.content))
+                return img
+            except UnidentifiedImageError:
+                print("API response is not a valid image.")
+        else:
+            print(f"API returned status {response.status_code}")
+            print("Response content:", response.text[:200])
+
+    except requests.RequestException as e:
+        print(f"Request failed: {e}")
+
+    # Fallback placeholder image
+    print("Using placeholder image.")
+    placeholder = Image.new("RGB", (512, 512), color=(200, 200, 200))
+    draw = ImageDraw.Draw(placeholder)
+    text = "Image\nNot\nAvailable"
+    
+    # Optional: Use a basic font
+    try:
+        font = ImageFont.load_default()
+        w, h = draw.multiline_textsize(text, font=font)
+        draw.multiline_text(
+            ((512 - w) / 2, (512 - h) / 2), text, fill=(50, 50, 50), font=font, align="center"
+        )
+    except Exception:
+        draw.text((50, 200), text, fill=(50, 50, 50))
+    
+    return placeholder