Add progress tracking to image generation pipeline

app.py

@@ -108,11 +108,12 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_in
         if not prompt or prompt.strip() == "":
             return None, "Please provide a valid prompt."
 
+        progress(0.1, desc="Loading image generation model...")
         pipe = get_image_gen_pipeline()
         if pipe is None:
             return None, "Image generation model is unavailable."
         
-        # Validate width/height dimensions
+        progress(0.2, desc="Validating dimensions...")
         is_valid, error_msg = validate_dimensions(width, height)
         if not is_valid:
             return None, error_msg
@@ -120,7 +121,7 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_in
         if randomize_seed:
             seed = random.randint(0, MAX_SEED)
         
-        progress(0.2, desc="Setting up generator...")
+        progress(0.3, desc="Setting up generator...")
         generator = torch.Generator().manual_seed(seed)
         
         progress(0.4, desc="Generating image...")