as API

speckles.py

@@ -1,15 +1,44 @@
 import itertools
+import io
+from fastapi.responses import StreamingResponse
 from pathlib import Path
 import pyvips
 import logging
 from perlin import CoordsGenerator
 import svg
-import json
 
 import numpy as np
+import uvicorn
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+
+app = FastAPI(title="Speckles API", root_path="/images")
+origins = [
+    "http://localhost",
+    "http://localhost:3000",
+    "https://localhost",
+    "https://0124816.xyz",
+    "http://0124816.xyz:3001",
+]
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=origins,
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
 
 RGen = np.random.default_rng()
 
+MEDIA_TYPES = {
+    "png": "image/png",
+    "jpeg": "image/jpeg",
+    "jpg": "image/jpeg",
+    "svg": "image/svg+xml",
+    "pdf": "application/pdf",
+}
+
 
 def emoji(marker: str, source="twemoji"):
     if len(marker) != 1:
@@ -181,6 +210,7 @@ def markertoSVG(marker, x, y, c, s):
         )
 
 
+@app.get("/speckles/")
 def make_wallpaper(
     speckle_colours: str,
     density: float = 0.12,
@@ -254,60 +284,19 @@ def make_wallpaper(
         )
 
     content = svg.SVG(width=x, height=y, elements=elements)
-    if Path(filename).suffix == ".svg":
-        with open(filename, "wt") as f:
-            f.write(content.as_str())
-    else:
-        with pyvips.Image.svgload_buffer(content.as_str().encode("utf-8")) as image:
-            image.write_to_file(filename)
+    if filename:
+        if Path(filename).suffix == ".svg":
+            with open(filename, "wt") as f:
+                f.write(content.as_str())
+        else:
+            with pyvips.Image.svgload_buffer(content.as_str().encode("utf-8")) as image:
+                image.write_to_file(filename)
+        return filename
+    elif fileformat:
+        buf = io.BytesIO(content.as_str().encode("utf-8"))
+        buf.seek(0)
+        return StreamingResponse(content=buf, media_type=MEDIA_TYPES[fileformat])
 
 
 if __name__ == "__main__":
-    palettes = [
-        {"colours": ["#DE4A21", "#6F04F2", "#490B1C", "#4a9c9e"], "votes": 1000000},
-        {"colours": ["#eee4ab", "#e5cb9f", "#99c4c8", "#68a7ad"], "votes": 100000},
-        {"colours": ["#e4d192", "#cba0ae", "#af7ab3", "#80558c"], "votes": 100000},
-        {"colours": ["#eeeeee", "#e1d4bb", "#cbb279", "#537188"], "votes": 100000},
-        {"colours": ["#c0dbea", "#ba90c6", "#e8a0bf", "#fdf4f5"], "votes": 100000},
-        {"colours": ["#f5ffc9", "#b3e5be", "#a86464", "#804674"], "votes": 100000},
-        {"colours": ["#ffde7d", "#f6416c", "#f8f3d4", "#00b8a9"], "votes": 100000},
-        {"colours": ["#53354a", "#903749", "#e84545", "#2b2e4a"], "votes": 100000},
-        {"colours": ["#967e76", "#d7c0ae", "#eee3cb", "#b7c4cf"], "votes": 100000},
-        {"colours": ["#fc5185", "#f5f5f5", "#3fc1c9", "#364f6b"], "votes": 100000},
-        {"colours": ["#eaeaea", "#ff2e63", "#252a34", "#08d9d6"], "votes": 100000},
-        {"colours": ["#eeeeee", "#00adb5", "#393e46", "#222831"], "votes": 100000},
-        {"colours": ["#2cd3e1", "#a459d1", "#f266ab", "#ffb84c"], "votes": 100000},
-        {"colours": ["#ffe194", "#e8f6ef", "#1b9c85", "#4c4c6d"], "votes": 100000},
-        {"colours": ["#146c94", "#19a7ce", "#b0daff", "#feff86"], "votes": 100000},
-        {"colours": ["#4c3d3d", "#c07f00", "#ffd95a", "#fff7d4"], "votes": 100000},
-        {"colours": ["#8bacaa", "#b04759", "#e76161", "#f99b7d"], "votes": 100000},
-        {"colours": ["#146c94", "#19a7ce", "#afd3e2", "#f6f1f1"], "votes": 100000},
-        {"colours": ["#9ba4b5", "#212a3e", "#394867", "#f1f6f9"], "votes": 100000},
-        {"colours": ["#00ffca", "#05bfdb", "#088395", "#0a4d68"], "votes": 100000},
-        {"colours": ["#7c9070", "#9ca777", "#fee8b0", "#f97b22"], "votes": 100000},
-        {"colours": ["#002a19", "#000000", "#ffffff", "#e0512f"], "votes": 100000},
-    ]
-    make_wallpaper(
-        ",".join(["#000000"] + palettes[0]["colours"]),
-        orientation="2880x1920",
-        filename="testing.png",
-        markers="tY.",
-        size=3,
-    )
-    exit()
-
-    palette_files = ["colorhunt.json"]
-    for palette_file in palette_files:
-        with open(palette_file, "rt") as f:
-            palettes += json.load(f)
-
-    print(len(palettes))
-    palettes = [
-        palette["colours"]
-        for palette in palettes
-        if palette["votes"] > 30000 or len(palette["colours"]) > 5
-    ]
-    print(len(palettes))
-    for palette in palettes:
-        night_sky(palette)
-        speckles(palette)
+    uvicorn.run("speckles:app", workers=3, port=8099)