feat: log total GPU time consumed per generate_image call

app.py

@@ -2,6 +2,7 @@ import io
 import os
 import random
 import re
+import time
 
 import numpy as np
 import torch
@@ -325,6 +326,32 @@ def generate_image(
     progress=gr.Progress(track_tqdm=True),
 ):
     """Generate an image from the given prompt using the selected model."""
+    _gpu_start = time.time()
+    try:
+        return _generate_image_inner(
+            model_name, prompt, negative_prompt, use_negative_prompt,
+            height, width, num_inference_steps, guidance_scale, seed, randomize_seed,
+        )
+    finally:
+        print(
+            f"[ImageStudio] GPU time consumed: {time.time() - _gpu_start:.2f}s "
+            f"(model={model_name}, steps={num_inference_steps}, {int(width)}x{int(height)})",
+            flush=True,
+        )
+
+
+def _generate_image_inner(
+    model_name,
+    prompt,
+    negative_prompt,
+    use_negative_prompt,
+    height,
+    width,
+    num_inference_steps,
+    guidance_scale,
+    seed,
+    randomize_seed,
+):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     seed = int(seed)