initial commit

recraft_ai.py

@@ -0,0 +1,78 @@
+import re
+from os import getenv
+from pathlib import Path
+
+import requests
+from openai import OpenAI
+
+API_KEY = getenv("RECRAFT_API_TOKEN")
+client = OpenAI(base_url="https://external.api.recraft.ai/v1", api_key=API_KEY)
+
+styles = {"nature-simplism": "ee2bb119-a790-409f-888e-b4b7b73f0aa0"}
+
+
+def generate_image_v3(prompt: str, file_path: Path, style: str):
+    response = client.images.generate(
+        prompt=prompt,
+        size="1024x1024",
+        model="recraftv3",
+        extra_body={
+            "style_id": styles[style],
+        },
+    )
+    image_url = response.data[0].url
+    if not image_url:
+        raise ValueError("No image URL returned by the API.")
+
+    img_response = requests.get(image_url)
+    if img_response.status_code != 200:
+        raise IOError(f"Failed to download image from {image_url}")
+
+    with open(file_path, "wb") as f:
+        f.write(img_response.content)
+
+
+def generate_image_v2(prompt: str, file_path: Path):
+    response = client.images.generate(
+        prompt=prompt,
+        size="1024x1024",
+        model="recraftv2",
+        style="icon",
+    )
+    image_url = response.data[0].url
+    if not image_url:
+        raise ValueError("No image URL returned by the API.")
+
+    img_response = requests.get(image_url)
+    if img_response.status_code != 200:
+        raise IOError(f"Failed to download image from {image_url}")
+
+    with open(file_path, "wb") as f:
+        f.write(img_response.content)
+
+
+def sanitize_filename(name: str) -> str:
+    # Define illegal characters (union of Windows + Unix)
+    illegal_chars = r'[\\\/:*?"<>|]'
+
+    # Replace spaces and illegal characters with "_"
+    name = re.sub(r"\s", "_", name)
+    name = re.sub(illegal_chars, "_", name)
+
+    # Windows also doesn't allow names ending in . or space
+    name = name.rstrip(" .")
+
+    # Windows reserved device names
+    reserved_names = {
+        "CON",
+        "PRN",
+        "AUX",
+        "NUL",
+        *(f"COM{i}" for i in range(1, 10)),
+        *(f"LPT{i}" for i in range(1, 10)),
+    }
+    name_upper = name.upper().split(".")[0]
+    if name_upper in reserved_names:
+        name = f"_{name}"
+
+    return name