Update app.py

app.py

@@ -269,49 +269,49 @@ css = """
 
 
 
-    @spaces.GPU(duration=400)
-    def generate_all(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps,
-                     progress=gr.Progress()):
-        outputs = [None] * (len(MODEL_CONFIGS) * 2)
-        for idx, model_name in enumerate(MODEL_CONFIGS.keys()):
-            try:
-                # Display progress for the specific model
-                # progress(0, desc=f"Starting generation for {model_name}...")
-                print(f"IMAGE GENERATING {model_name} ")
-                image, used_seed = generate_image(
-                    model_name, prompt, negative_prompt, seed,
-                    randomize_seed, width, height, guidance_scale,
-                    num_inference_steps, progress
-                )
-                print(f"IMAGE GENERATIED {model_name} ")
-                # Update the respective model's tab with the generated image
-                # results[model_name].update(image)
-                # seeds[model_name].update(used_seed)
-                outputs[idx * 2] = image  # Image slot
-                outputs[idx * 2 + 1] = seed  # Seed slot
-                # outputs.extend([image, used_seed])
-                # Add intermediate results to progress * (len(all_outputs) - len(all_outputs))
-                print("YELID")
-                yield outputs + [None]
-
-
-            except Exception as e:
-                print(f"Error generating with {model_name}: {str(e)}")
-                outputs[idx * 2] = None
-                outputs[idx * 2 + 1] = None
-
-        # Update the gallery after generation
-        gallery_images = update_gallery()
-        # file_gallery.update(value=gallery_images)
-        return outputs
-
-    def update_gallery():
-        """Update the file gallery"""
-        files = get_generated_images()
-        return [
-            (f["path"], f"{f['name']}\n{f['date']}")
-            for f in files
-        ]
+@spaces.GPU(duration=400)
+def generate_all(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps,
+                 progress=gr.Progress()):
+    outputs = [None] * (len(MODEL_CONFIGS) * 2)
+    for idx, model_name in enumerate(MODEL_CONFIGS.keys()):
+        try:
+            # Display progress for the specific model
+            # progress(0, desc=f"Starting generation for {model_name}...")
+            print(f"IMAGE GENERATING {model_name} {prompt}")
+            image, used_seed = generate_image(
+                model_name, prompt, negative_prompt, seed,
+                randomize_seed, width, height, guidance_scale,
+                num_inference_steps, progress
+            )
+            print(f"IMAGE GENERATIED {model_name} ")
+            # Update the respective model's tab with the generated image
+            # results[model_name].update(image)
+            # seeds[model_name].update(used_seed)
+            outputs[idx * 2] = image  # Image slot
+            outputs[idx * 2 + 1] = seed  # Seed slot
+            # outputs.extend([image, used_seed])
+            # Add intermediate results to progress * (len(all_outputs) - len(all_outputs))
+            print("YELID")
+            yield outputs + [None]
+
+
+        except Exception as e:
+            print(f"Error generating with {model_name}: {str(e)}")
+            outputs[idx * 2] = None
+            outputs[idx * 2 + 1] = None
+
+    # Update the gallery after generation
+    gallery_images = update_gallery()
+    # file_gallery.update(value=gallery_images)
+    return outputs
+
+def update_gallery():
+    """Update the file gallery"""
+    files = get_generated_images()
+    return [
+        (f["path"], f"{f['name']}\n{f['date']}")
+        for f in files
+    ]
 
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
@@ -397,7 +397,7 @@ with gr.Blocks(css=css) as demo:
                                     height=800,
                                     visible=True
                                 )
-                                refresh_button = gr.Button("Refresh Gallery",visible=True)
+                                refresh_button = gr.Button("Refresh Gallery",visible=True) # not visible. why?
             
                                 refresh_button.click(
                     fn=update_gallery,