add buffer

myapp/app.py

@@ -1,7 +1,6 @@
 import re
 from functools import partial
-from pathlib import Path
-from tempfile import NamedTemporaryFile
+from io import BytesIO
 from typing import Any
 
 import gradio as gr
@@ -63,20 +62,21 @@ with gr.Blocks() as demo:
 
             return f"#{r:x}{g:x}{b:x}".upper()
 
+        image = Image.open(data[background])
         qr_code = segno.make(data[text], error="h")
-        suffix = Path(data[background]).suffix
 
-        with NamedTemporaryFile(suffix=suffix) as temp_file:
+        with BytesIO() as buffer:
             write_artistic(
                 qr_code,
-                target=temp_file,
-                background=data[background],
+                target=buffer,
+                background=image.fp,
+                kind=image.format,
                 scale=data[scale],
                 light=to_hex_format(data[color_light]),
                 dark=to_hex_format(data[color_dark]),
             )
 
-            return Image.open(temp_file.name)
+            return Image.open(buffer)
 
     gr.on(
         [button.click, prompt.submit],