diff --git a/main.py b/main.py index b76615c..67d8aca 100644 --- a/main.py +++ b/main.py @@ -92,7 +92,7 @@ def make_wallpaper( size: float = 3, fileformat: str = "svg", orientation: str = "landscape", - local: bool = False, + filename: str = "", markers: str = ".", perlin: bool = True, ): @@ -163,14 +163,13 @@ def make_wallpaper( pad_inches=0, ) buf.seek(0) - if local: - filename = f"speckles.{fileformat}" + if filename: with open(filename, "wb") as f: f.write(buf.getbuffer()) + buf.close() return filename else: return StreamingResponse(content=buf, media_type=MEDIA_TYPES[fileformat]) - buf.close() if __name__ == "__main__":