Create openai_provider.py

providers/openai_provider.py

@@ -0,0 +1,25 @@
+import os
+from typing import List
+
+from openai import OpenAI
+from .base import ImageProvider
+
+_OPENAI_IMAGE_MODEL = os.getenv("OPENAI_IMAGE_MODEL", "gpt-image-1")
+
+class OpenAIImageProvider(ImageProvider):
+    def __init__(self, api_key: str):
+        # The SDK reads key from env, but we allow explicit injection too.
+        os.environ["OPENAI_API_KEY"] = api_key
+        self.client = OpenAI()
+
+    def generate_image(self, prompt: str, size: str = "1024x1024", n: int = 1) -> List[str]:
+        # OpenAI returns base64 JSON for images
+        resp = self.client.images.generate(
+            model=_OPENAI_IMAGE_MODEL,
+            prompt=prompt,
+            size=size,
+            n=n
+        )
+        # Each item has .b64_json
+        images_b64 = [d.b64_json for d in resp.data]
+        return images_b64