Update Modules/Generate_Image.py

Modules/Generate_Image.py

@@ -1,12 +1,14 @@
 from __future__ import annotations
 
 import os
+import uuid
 import random
 from typing import Annotated
 
 import gradio as gr
 from PIL import Image
 from huggingface_hub import InferenceClient
+from .File_System import ROOT_DIR
 
 from app import _log_call_end, _log_call_start, _truncate_for_log
 from ._docstrings import autodoc
@@ -38,7 +40,7 @@ def Generate_Image(
     width: Annotated[int, "Output width in pixels (64–1216, multiple of 32 recommended)."] = 1024,
     height: Annotated[int, "Output height in pixels (64–1216, multiple of 32 recommended)."] = 1024,
     sampler: Annotated[str, "Sampling method label (UI only). Common options: 'DPM++ 2M Karras', 'DPM++ SDE Karras', 'Euler', 'Euler a', 'Heun', 'DDIM'."] = "DPM++ 2M Karras",
-) -> Image.Image:
+) -> str:
     _log_call_start(
         "Generate_Image",
         prompt=_truncate_for_log(prompt, 200),
@@ -67,8 +69,13 @@ def Generate_Image(
                 guidance_scale=cfg_scale,
                 seed=seed if seed != -1 else random.randint(1, 1_000_000_000),
             )
-            _log_call_end("Generate_Image", f"provider={provider} size={image.size}")
-            return image
+            
+            filename = f"image_{uuid.uuid4().hex[:8]}.png"
+            output_path = os.path.join(ROOT_DIR, filename)
+            image.save(output_path)
+            
+            _log_call_end("Generate_Image", f"provider={provider} size={image.size} saved_to={filename}")
+            return output_path
         except Exception as exc:  # pylint: disable=broad-except
             last_error = exc
             continue