Create ImageGenerator.py

ImageGenerator.py

@@ -0,0 +1,15 @@
+import requests
+from PIL import Image
+from io import BytesIO
+
+HF_API_KEY = os.getenv("HF_API_KEY")
+
+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}"}
+
+    response = requests.post(url, headers=headers, json={"inputs": prompt})
+
+    return Image.open(BytesIO(response.content))