Return encrypted generation files

API.md

@@ -70,7 +70,7 @@ Content-Type: application/json
 }
 ```
 
-Returns a PNG with ComfyUI workflow and parameter metadata, scaled after generation by the decimal `return_scale` from `0.01` through `1`, with content type `image/png`. Regional prompts and their mask nodes are included in the metadata. Use `1` for full resolution. `regions` accepts up to three spatial prompts on a 5 by 5 grid. Set `regional_mode` to `attention` for Attention Couple (PPM), or omit it for soft conditioning. Regional prompts establish subjects from the first sampling step. The reduced-strength global prompt starts after the first fifth to complete the environment while regions remain active. Each internal region edge feathers by half a grid cell, so overlap adjacent regions by one cell for a smooth crossfade. `area` accepts `auto`, `full`, `tl`, `tc`, `tr`, `ml`, `mc`, `mr`, `bl`, `bc`, `br`, `th`, `mh`, `bh`, `lh`, `ch`, or `rh`. It also accepts any single cell from `a1` through `e5`, or a rectangular inclusive range such as `a1:c5`. `auto` places one prompt in the center, two left and right, or three left, center, and right. Columns run left to right and rows run top to bottom. Region strength defaults to `1`. Use the global `prompt` for environment, lighting, shared style, and composition outside regions. Omitting `regions` keeps whole-image generation unchanged. Set either scheduler to `AlignYourSteps` to use `AlignYourStepsScheduler` with fixed model type `SDXL` and the corresponding steps and denoise values. Only `prompt` is required for the built-in site defaults. LoRAs and detailers run in array order. A blank detailer `prompt` or `negative` inherits the matching main prompt. Each request creates fresh sampler seeds. When `upscale` is true, the selected upscaler is normalized to the requested `upscale_scale` resolution before the second sampler. Leave `second_model` empty to reuse the first model and LoRA chain; otherwise `second_loras` applies to the selected second model. Width and height must be multiples of 8 from 64 through 2048; steps must be from 1 through 100.
+Returns an encrypted `.epng` with ComfyUI workflow and parameter metadata, scaled after generation by the decimal `return_scale` from `0.01` through `1`, with content type `application/octet-stream`. Regional prompts and their mask nodes are included in the metadata. Use `1` for full resolution. `regions` accepts up to three spatial prompts on a 5 by 5 grid. Set `regional_mode` to `attention` for Attention Couple (PPM), or omit it for soft conditioning. Regional prompts establish subjects from the first sampling step. The reduced-strength global prompt starts after the first fifth to complete the environment while regions remain active. Each internal region edge feathers by half a grid cell, so overlap adjacent regions by one cell for a smooth crossfade. `area` accepts `auto`, `full`, `tl`, `tc`, `tr`, `ml`, `mc`, `mr`, `bl`, `bc`, `br`, `th`, `mh`, `bh`, `lh`, `ch`, or `rh`. It also accepts any single cell from `a1` through `e5`, or a rectangular inclusive range such as `a1:c5`. `auto` places one prompt in the center, two left and right, or three left, center, and right. Columns run left to right and rows run top to bottom. Region strength defaults to `1`. Use the global `prompt` for environment, lighting, shared style, and composition outside regions. Omitting `regions` keeps whole-image generation unchanged. Set either scheduler to `AlignYourSteps` to use `AlignYourStepsScheduler` with fixed model type `SDXL` and the corresponding steps and denoise values. Only `prompt` is required for the built-in site defaults. LoRAs and detailers run in array order. A blank detailer `prompt` or `negative` inherits the matching main prompt. Each request creates fresh sampler seeds. When `upscale` is true, the selected upscaler is normalized to the requested `upscale_scale` resolution before the second sampler. Leave `second_model` empty to reuse the first model and LoRA chain; otherwise `second_loras` applies to the selected second model. Width and height must be multiples of 8 from 64 through 2048; steps must be from 1 through 100.
 
 ## Archive
 

app.py

@@ -2040,7 +2040,11 @@ def api_generate(body, p=""):
     if not valid_pass(p):
         raise gr.Error("Invalid password")
 
-    return generate_image(DirectRequest.model_validate_json(body))
+    image = generate_image(DirectRequest.model_validate_json(body))
+    temp = tempfile.NamedTemporaryFile(suffix=".epng", delete=False)
+    temp.write(encrypted_image_bytes(image))
+    temp.close()
+    return temp.name
 
 
 def api_health(p=""):
@@ -2117,7 +2121,7 @@ def queued_generate(request):
         PASSWORD,
         api_name="/generate",
     )
-    return Image.open(result).copy()
+    return Image.open(BytesIO(stored_bytes(Path(result)))).copy()
 
 
 def queued_matrix_cell(request):
@@ -2643,7 +2647,11 @@ def direct(
             Image.Resampling.LANCZOS,
         )
 
-    return Response(content=image_png_bytes(image), media_type="image/png")
+    return Response(
+        content=encrypted_image_bytes(image),
+        media_type="application/octet-stream",
+        headers={"Content-Disposition": 'attachment; filename="image.epng"'},
+    )
 
 
 @api.post(
@@ -3083,6 +3091,7 @@ with gr.Blocks(title="Image generation") as demo:
     api_pass_input = gr.Textbox(visible=False)
     request_input = gr.Textbox(visible=False)
     api_json_output = gr.JSON(visible=False)
+    api_file_output = gr.File(visible=False)
     matrix_output = gr.Textbox(visible=False)
     login_button.click(
         login,
@@ -3184,7 +3193,7 @@ with gr.Blocks(title="Image generation") as demo:
             request_input,
             api_pass_input,
         ],
-        output,
+        api_file_output,
         api_name="generate",
     )