fixed tab issues via chatGPT

app.py

@@ -327,27 +327,33 @@ with gr.Blocks(css=css) as demo:
 
     @spaces.GPU(duration=600)
     def generate_all(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, progress=gr.Progress()):
-        outputs = []
-        for model_name in MODEL_CONFIGS.keys():
-            print(f"IMAGE GENERATING {model_name} ")
-            try:
-                image, used_seed = generate_image(
-                    model_name, prompt, negative_prompt, seed,
-                    randomize_seed, width, height, guidance_scale,
-                    num_inference_steps, progress
-                )
-                print(f"IMAGE GENERATED {model_name} {update_memory_usage()}")
-                outputs.extend([image, used_seed])
-                
-            except Exception as e:
-                outputs.extend([None, None])
-                print(f"Error generating with {model_name}: {str(e)}")
-        
-        # Update the gallery after generation
-        gallery_images = update_gallery()
-        file_gallery.update(value=gallery_images)
+    outputs = []
+    for model_name in MODEL_CONFIGS.keys():
+        try:
+            # Display progress for the specific model
+            progress(0, desc=f"Starting generation for {model_name}...")
+            
+            image, used_seed = generate_image(
+                model_name, prompt, negative_prompt, seed,
+                randomize_seed, width, height, guidance_scale,
+                num_inference_steps, progress
+            )
+            
+            # Update the respective model's tab with the generated image
+            results[model_name].update(image)
+            seeds[model_name].update(used_seed)
+            
+            outputs.extend([image, used_seed])
         
-        return outputs
+        except Exception as e:
+            print(f"Error generating with {model_name}: {str(e)}")
+            outputs.extend([None, None])
+    
+    # Update the gallery after generation
+    gallery_images = update_gallery()
+    file_gallery.update(value=gallery_images)
+    
+    return outputs
 
     output_components = []
     for model_name in MODEL_CONFIGS.keys():