nutshellai-tech/mobius/research-image-agent

  #!/usr/bin/env python3
  """Minimal gpt-image-2 test (2K, 16:9)."""
  import argparse
  import base64
  import json
  import time
  import urllib.request
  from pathlib import Path
  API_KEY = "sk-xxxxxxxxxxxx"
  GEN_SERVICE_URL = "http://..............."
  PROMPT = "xxxxxxxxxxxxxxxxxxxxxxxxx"
  def main() -> None:
      p = argparse.ArgumentParser()
      p.add_argument("--api-key", default=API_KEY)
      p.add_argument("--gen-service-url", default=GEN_SERVICE_URL)
      p.add_argument("--prompt", default=PROMPT)
      a = p.parse_args()
      req = urllib.request.Request(
          a.gen_service_url,
          data=json.dumps({
              "model": "gpt-image-2-vip",
              "prompt": a.prompt,
              "size": "2048x1152",
              "response_format": "url",
          }).encode(),
          headers={"Authorization": f"Bearer {a.api_key}", "Content-Type": "application/json"},
          method="POST",
      )
      with urllib.request.urlopen(req, timeout=600) as r:
          item = json.load(r)["data"][0]
      out = Path("./outputs") / f"gpt-image-2-{time.strftime('%Y%m%d-%H%M%S')}.png"
      out.parent.mkdir(parents=True, exist_ok=True)
      if item.get("url"):
          with urllib.request.urlopen(item["url"], timeout=600) as r:
              out.write_bytes(r.read())
      else:
          out.write_bytes(base64.b64decode(item["b64_json"]))
      print(out)
  if __name__ == "__main__":
      main()